CARE: Class Attention to Regions of Lesion for Classification on Imbalanced DataDownload PDF

17 Dec 2018 (modified: 24 Jun 2019)MIDL 2019 Conference Full SubmissionReaders: Everyone
  • Keywords: Attention Mechanism, Imbalanced Data, Small Samples, Skin Lesion, Pneumonia Chest X-ray
  • Abstract: To date, it is still an open and challenging problem for intelligent diagnosis systems to effectively learn from imbalanced data, especially with large samples of common diseases and much smaller samples of rare ones. Inspired by the process of human learning, this paper proposes a novel and effective way to embed attention into the machine learning process, particularly for learning characteristics of rare diseases. This approach does not change architectures of the original CNN classifiers and therefore can directly plug and play for any existing CNN architecture. Comprehensive experiments on a skin lesion dataset and a pneumonia chest X-ray dataset showed that paying attention to lesion regions of rare diseases during learning not only improved the classification performance on rare diseases, but also on the mean class accuracy.
  • Code Of Conduct: I have read and accept the code of conduct.
  • Remove If Rejected: (optional) Remove submission if paper is rejected.
8 Replies