论文标题

EDGEMIXUP:改善皮肤疾病分类和细分的公平性

EdgeMixup: Improving Fairness for Skin Disease Classification and Segmentation

论文作者

Yuan, Haolin, Hadzic, Armin, Paul, William, de Flores, Daniella Villegas, Mathew, Philip, Aucott, John, Cao, Yinzhi, Burlina, Philippe

论文摘要

皮肤病变可能是各种传染病和其他疾病的早期指标。使用深度学习(DL)模型诊断皮肤病变具有帮助临床医生对预筛查患者的巨大潜力。但是,这些模型通常会学习训练数据中固有的偏见,这可能会导致诊断浅色和/或深色皮肤的人的性能差距。据我们所知,在识别皮肤疾病分类和分割中识别模型偏见的方面已经完成了有限的工作。在本文中,我们研究了DL公平性,并证明了与肤色较轻的个体相比,针对皮肤色调较深的肤色的分类和分割模型中存在偏见,包括莱姆,锡那科和疱疹带状疱疹在内的特定疾病。然后,我们提出了一种新型的预处理,数据改变方法,称为Edgemixup,以通过输入皮肤病变图像的线性组合和相应的预测边缘检测掩码结合颜色饱和度改变来提高模型公平性。对于皮肤疾病分类的任务,Edgemixup的表现要优于更复杂的竞争方法,例如对抗方法,在光和深色肤色样本之间的准确性差距降低了10.99%,导致了8.4%的绩效提高了代表性不足的亚属性。

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.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源