Out-of-distribution detection for long-tailed and fine-grained skin lesion images

Deval Mehta, Yaniv Gal, Adrian Bowling, Paul Bonnington, Zongyuan Ge

Research output: Chapter in Book/Report/Conference proceedingConference PaperResearchpeer-review

6 Citations (Scopus)

Abstract

Recent years have witnessed a rapid development of automated methods for skin lesion diagnosis and classification. Due to an increasing deployment of such systems in clinics, it has become important to develop a more robust system towards various Out-of-Distribution (OOD) samples (unknown skin lesions and conditions). However, the current deep learning models trained for skin lesion classification tend to classify these OOD samples incorrectly into one of their learned skin lesion categories. To address this issue, we propose a simple yet strategic approach that improves the OOD detection performance while maintaining the multi-class classification accuracy for the known categories of skin lesion. To specify, this approach is built upon a realistic scenario of a long-tailed and fine-grained OOD detection task for skin lesion images. Through this approach, 1) First, we target the mixup amongst middle and tail classes to address the long-tail problem. 2) Later, we combine the above mixup strategy with prototype learning to address the fine-grained nature of the dataset. The unique contribution of this paper is two-fold, justified by extensive experiments. First, we present a realistic problem setting of OOD task for skin lesion. Second, we propose an approach to target the long-tailed and fine-grained aspects of the problem setting simultaneously to increase the OOD performance.

Original languageEnglish
Title of host publicationMedical Image Computing and Computer Assisted Intervention – MICCAI 2022
Subtitle of host publication25th International Conference, Singapore, September 18–22, 2022, Proceedings, Part I
EditorsLinwei Wang, Qi Dou, P. Thomas Fletcher, Stefanie Speidel, Shuo Li
Place of PublicationCham Switzerland
PublisherSpringer
Pages732-742
Number of pages11
Edition1st
ISBN (Electronic)9783031164316
ISBN (Print)9783031164309
DOIs
Publication statusPublished - 2022
EventMedical Image Computing and Computer-Assisted Intervention 2022 - Singapore, Singapore
Duration: 18 Sept 202222 Sept 2022
Conference number: 25th
https://link.springer.com/book/10.1007/978-3-031-16434-7 (Proceedings - Part 2)
https://conferences.miccai.org/2022/en/ (Website)

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13431
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

ConferenceMedical Image Computing and Computer-Assisted Intervention 2022
Abbreviated titleMICCAI 2022
Country/TerritorySingapore
CitySingapore
Period18/09/2222/09/22
Internet address

Keywords

  • Mixup
  • Openset
  • Out-of-distribution
  • Prototype
  • Skin lesion

Cite this