研究标签偏差如何影响乳腺密度分类模型的公平性。
Exploring the interplay of label bias with subgroup size and separability: A case study in mammographic density classification
- 通过模拟不同大小和可分性的子群体偏差,分析对模型特征的影响。
- 当偏差影响主要可分子群时,真阳性率从0.898降至0.518。
- 验证集是否含清洁标签显著影响模型在子群体上的表现。
医学影像数据集中特定子群体的系统性误标注(即标签偏差)是医疗AI公平性中被忽视的重要问题。本文以EMory BrEast影像数据集(EMBED)为基础,研究标签偏差作用下子群体的相对大小与可分性如何影响深度学习模型的特征表示与性能。我们训练了用于二分类组织密度的深度模型,其中偏差影响可区分子群(按成像设备厂商划分)或不可区分的“伪子群”。结果发现,模拟的标签偏差导致模型特征表示发生显著偏移,且这种偏移同时依赖于受影响子群的相对大小和可分性。此外,使用带清洁标签的验证集定义分类阈值时,子群表现有明显差异:当偏差影响主要可分子群时,该子群真阳性率从0.898下降至0.518(验证集含偏差标签)。本工作为理解标签偏差对医疗影像AI子群体公平性的后果提供了关键洞见。
原文摘要 · Abstract (English)
Systematic mislabelling affecting specific subgroups (i.e., label bias) in medical imaging datasets represents an understudied issue concerning the fairness of medical AI systems. In this work, we investigated how size and separability of subgroups affected by label bias influence the learned features and performance of a deep learning model. Therefore, we trained deep learning models for binary tissue density classification using the EMory BrEast imaging Dataset (EMBED), where label bias affected separable subgroups (based on imaging manufacturer) or non-separable "pseudo-subgroups". We found that simulated subgroup label bias led to prominent shifts in the learned feature representations of the models. Importantly, these shifts within the feature space were dependent on both the relative size and the separability of the subgroup affected by label bias. We also observed notable differences in subgroup performance depending on whether a validation set with clean labels was used to define the classification threshold for the model. For instance, with label bias affecting the majority separable subgroup, the true positive rate for that subgroup fell from 0.898, when the validation set had clean labels, to 0.518, when the validation set had biased labels. Our work represents a key contribution toward understanding the consequences of label bias on subgroup fairness in medical imaging AI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。