Difficulty-aware Meta-Learning for Rare Disease Diagnosis

By Xiaomeng Li, Lequan Yu, Chi-Wing Fu, P. Heng. 2019

75 citations recorded by the source corpus.

Rare diseases have extremely low-data regimes, unlike common diseases with large amount of available labeled data. Hence, to train a neural network to classify rare diseases with a few per-class data samples is very challenging, and so far, catches very little attention. In this paper, we present a difficulty-aware meta-learning method to address rare disease classifications and demonstrate its capability to classify dermoscopy images. Our key approach is to first train and construct a meta-learning model from data of common diseases, then adapt the model to perform rare disease this http URL achieve this, we develop the difficulty-aware meta-learning method that dynamically monitors the importance of learning tasks during the meta-optimization stage. To evaluate our method, we use the recent ISIC 2018 skin lesion classification dataset, and show that with only five samples per class, our model can quickly adapt to classify unseen classes by a high AUC of 83.3%. Also, we evaluated several rare disease classification results in the public Dermofit Image Library to demonstrate the potential of our method for real clinical practice.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Oncology, Meta & Continual Learning, Medical & Health AI

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Katja Hofmann
  2. Daniela Massiceti
  3. Edward Cutrell
  4. Simone Stumpf
  5. Luisa M. Zintgraf
  6. John F. Bronskill
  7. Chelsea Finn
  8. Sergey Levine

In Computer Vision and Pattern Recognition

  1. Cecily Morrison
  2. Katja Hofmann
  3. Edward Cutrell
  4. Simone Stumpf
  5. Lida Theodorou
  6. Luisa M. Zintgraf
  7. Daniela Massiceti
  8. John F. Bronskill

In Meta & Continual Learning

  1. Katja Hofmann
  2. Luisa M. Zintgraf
  3. Daniela Massiceti
  4. John F. Bronskill
  5. Chelsea Finn
  6. Sergey Levine

Citations in the corpus