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

在皮肤病诊断中实现无人口统计信息的公平性

Achieve Fairness without Demographics for Dermatological Disease Diagnosis

National Tsing Hua University(国立清华大学);University of Notre Dame(圣母大学);The Chinese University of Hong Kong(香港中文大学)

Ching-Hao Chiu, Yu-Jen Chen, Yawen Wu, Yiyu Shi, Tsung-Yi Ho

AI总结:

本文提出一种无需训练期敏感属性信息即可在测试阶段实现公平皮肤病诊断的方法,通过特征解纠缠正则化和SAM疾病掩码增强特征,在两个皮肤病数据集上验证了其公平性与准确性的提升。

摘要:

在医学图像诊断中,公平性已变得日益重要。若不进行偏置缓解,部署不公平的AI将损害弱势群体的利益,并可能撕裂社会。近期研究通过在训练期间利用人口统计(敏感属性)信息,来解决深度学习模型中关于人口统计群体(如性别、年龄和种族)的预测偏置。然而,许多敏感属性天然存在于皮肤病图像中。如果训练后的模型仅针对某一特定属性实现公平,那么它对于其他属性仍然是不公平的。此外,由于隐私问题,训练一个能够容纳多个敏感属性的模型是不切实际的。为克服这一问题,我们提出了一种方法,在测试阶段实现对敏感属性的公平预测,而无需在训练期间使用此类信息。受先前工作强调特征纠缠对公平性影响的启发,我们通过捕获与敏感属性和目标属性相关的特征,并正则化相应类别之间的特征纠缠,来增强模型特征。这确保了模型只能基于与目标属性相关的特征进行分类,而不依赖于与敏感属性相关的特征,从而同时提高公平性和准确性。此外,我们使用来自Segment Anything Model(SAM)的疾病掩码来提升所学特征的质量。实验结果表明,在两个皮肤病数据集上,与最先进的方法相比,所提出的方法能够提高分类中的公平性。

英文摘要:

In medical image diagnosis, fairness has become increasingly crucial. Without bias mitigation, deploying unfair AI would harm the interests of the underprivileged population and potentially tear society apart. Recent research addresses prediction biases in deep learning models concerning demographic groups (e.g., gender, age, and race) by utilizing demographic (sensitive attribute) information during training. However, many sensitive attributes naturally exist in dermatological disease images. If the trained model only targets fairness for a specific attribute, it remains unfair for other attributes. Moreover, training a model that can accommodate multiple sensitive attributes is impractical due to privacy concerns. To overcome this, we propose a method enabling fair predictions for sensitive attributes during the testing phase without using such information during training. Inspired by prior work highlighting the impact of feature entanglement on fairness, we enhance the model features by capturing the features related to the sensitive and target attributes and regularizing the feature entanglement between corresponding classes. This ensures that the model can only classify based on the features related to the target attribute without relying on features associated with sensitive attributes, thereby improving fairness and accuracy. Additionally, we use disease masks from the Segment Anything Model (SAM) to enhance the quality of the learned feature. Experimental results demonstrate that the proposed method can improve fairness in classification compared to state-of-the-art methods in two dermatological disease datasets.

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

↑