Class imbalance is a common problem in medical diagnosis, causing a standard classifier to be biased towards the common classes and perform poorly on the rare classes. This is especially true for dermatology, a specialty with thousands of skin conditions but many of which have low prevalence in the real world. Motivated by recent advances, we explore few-shot learning methods as well as conventional class imbalance techniques for the skin condition recognition problem and propose an evaluation setup to fairly assess the real-world utility of such approaches. We find the performance of few-show learning methods does not reach that of conventional class imbalance techniques, but combining the two approaches using a novel ensemble improves model performance, especially for rare classes. We conclude that ensembling can be useful to address the class imbalance problem, yet progress can further be accelerated by real-world evaluation setups for benchmarking new methods.
@article{arxiv.2010.04308,
title = {Addressing the Real-world Class Imbalance Problem in Dermatology},
author = {Wei-Hung Weng and Jonathan Deaton and Vivek Natarajan and Gamaleldin F. Elsayed and Yuan Liu},
journal= {arXiv preprint arXiv:2010.04308},
year = {2020}
}
Comments
Machine Learning for Health Workshop at NeurIPS 2020; 14 pages + 4 pages appendix, 8 figures, 6 appendix tables