arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.13883cs.CVcs.AI
收藏

EdgeMixup:提高皮肤病分类和分割的公平性

EdgeMixup: Improving Fairness for Skin Disease Classification and Segmentation

Johns Hopkins University(约翰斯·霍普金斯大学)

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

摘要:

皮肤损伤可以是广泛的传染病和其他疾病的早期指标。使用深度学习(DL)模型诊断皮肤病变在帮助临床医生对患者进行预筛选方面具有很大的潜力。然而,这些模型经常学习训练数据中固有的偏差,这可能导致在诊断具有浅色和/或深色肤色的人时的性能差距。据我们所知,在识别(更不用说减少)皮肤病分类和分割中的模型偏差方面所做的工作有限。在本文中,我们研究DL公平性,并证明存在偏见的分类和分割模型的亚群与肤色较深的个人相比,肤色较浅,特定的疾病,包括莱姆病,Tumbles Corporis和带状疱疹。然后,我们提出了一种新的预处理,数据更改方法,称为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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑