用对抗方法去除3D CT影像嵌入中的年龄性别种族偏见,提升医疗AI公平性
Towards Fair Medical AI: Adversarial Debiasing of 3D CT Foundation Embeddings
- 设计基于变分自编码器的对抗去偏框架,将嵌入映射到不携带人口统计信息的新空间
- 在NLST数据集上验证,去偏后仍保持1年和2年肺癌风险预测准确率
- 可抵御对抗性偏见攻击,适合关注医疗AI公平性的研究者与临床开发者
自监督学习已革新医学影像领域,通过大规模无标签数据高效提取通用特征。近期,自监督基础模型被扩展至三维(3D)CT数据,生成包含1408个特征的紧凑、信息丰富的嵌入,已在颅内出血检测和肺癌风险预测等下游任务中达到领先性能。然而,这些嵌入被证实编码了年龄、性别和种族等人口统计信息,对临床应用的公平性构成重大风险。本文提出一种基于变分自编码器(VAE)的对抗去偏框架,将嵌入转换至不再编码人口统计信息的新潜在空间,同时保持关键下游任务的性能。我们在NLST肺癌筛查数据集上验证该方法,结果表明去偏嵌入能有效消除多种人口统计信息编码,并在不牺牲1年和2年肺癌风险预测准确率的前提下提升公平性。此外,该方法还增强了嵌入对对抗性偏见攻击的鲁棒性。这些结果凸显了对抗去偏技术在保障自监督3D CT嵌入临床应用公平性方面的潜力,为实现无偏医疗决策提供了可行路径。
原文摘要 · Abstract (English)
Self-supervised learning has revolutionized medical imaging by enabling efficient and generalizable feature extraction from large-scale unlabeled datasets. Recently, self-supervised foundation models have been extended to three-dimensional (3D) computed tomography (CT) data, generating compact, information-rich embeddings with 1408 features that achieve state-of-the-art performance on downstream tasks such as intracranial hemorrhage detection and lung cancer risk forecasting. However, these embeddings have been shown to encode demographic information, such as age, sex, and race, which poses a significant risk to the fairness of clinical applications. In this work, we propose a Variation Autoencoder (VAE) based adversarial debiasing framework to transform these embeddings into a new latent space where demographic information is no longer encoded, while maintaining the performance of critical downstream tasks. We validated our approach on the NLST lung cancer screening dataset, demonstrating that the debiased embeddings effectively eliminate multiple encoded demographic information and improve fairness without compromising predictive accuracy for lung cancer risk at 1-year and 2-year intervals. Additionally, our approach ensures the embeddings are robust against adversarial bias attacks. These results highlight the potential of adversarial debiasing techniques to ensure fairness and equity in clinical applications of self-supervised 3D CT embeddings, paving the way for their broader adoption in unbiased medical decision-making.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。