The ISC is part of the Johns Hopkins Applied Physics Laboratory and will follow all current policies. Please visit the JHU/APL page for more information on the Lab's visitor guidance.

2022

EDGEMIXUP: Improving Fairness for Skin Disease Classification and Segmentation


Abstract

Skin lesions can be an early indicator of a wide range of infectious and other diseases. The use of deep learning (DL) models to diagnose skin lesions has great potential in assisting clinicians with prescreening patients. However, these models often learn biases inherent in training data, which can lead to a performance gap in the diagnosis of people with light and/or dark skin tones. To the best of our knowledge, limited work has been done on identifying, let alone reducing, model bias in skin disease classification and segmentation. In this paper, we examine DL fairness and demonstrate the existence of bias in classification and segmentation models for subpopulations with darker skin tones compared to individuals with lighter skin tones, for specific diseases including Lyme, Tinea Corporis and Herpes Zoster. Then, we propose a novel preprocessing, data alteration method, called EDGEMIXUP, to improve model fairness with a linear combination of an input skin lesion image and a corresponding a predicted edge detection mask combined with color saturation alteration. For the task of skin disease classification, EDGEMIXUP outperforms much more complex competing methods such as adversarial approaches, achieving a 10.99% reduction in accuracy gap between light and dark skin tone samples, and resulting in 8.4% improved performance for an underrepresented subpopulation.


Citation

@misc2202.13883 Author: Haolin Yuan and Armin Hadzic and William Paul and Daniella Villegas de Flores and Philip Mathew and John Aucott and Yinzhi Cao and Philippe Burlina Title: EdgeMixup: Improving Fairness for Skin Disease Classification and Segmentation Year: 2022 Eprint: arXiv:2202.13883

Citation

@misc2202.13883 Author: Haolin Yuan and Armin Hadzic and William Paul and Daniella Villegas de Flores and Philip Mathew and John Aucott and Yinzhi Cao and Philippe Burlina Title: EdgeMixup: Improving Fairness for Skin Disease Classification and Segmentation Year: 2022 Eprint: arXiv:2202.13883