LLM2D
Towards Fair Medical AI: adversarial 去偏 3D CT 基础嵌入
Towards Fair Medical AI: Adversarial Debiasing of 3D CT Foundation Embeddings
作者: Guangyao Zheng, Michael A. Jacobs, Vladimir Braverman, Vishwa S. Parekh
发布日期: 2/10/2025
arXiv ID: oai:arXiv.org:2502.04386v1

摘要

arXiv:2502.04386v1 类型: cross 摘要: 自监督学习通过从大规模未标记数据集中高效且通用地提取特征,彻底改变了医学成像领域。最近,自监督基础模型已扩展到三维(3D)计算机断层扫描(CT)数据,生成包含1408个特征的紧凑且信息丰富的嵌入,这些嵌入在颅内出血检测和肺癌风险预测等下游任务中达到了最先进的性能。然而,这些嵌入已被证明编码了年龄、性别和种族等人口统计数据,这对临床应用的公平性构成了重大风险。 在此项工作中,我们提出了一种基于变分自编码器(VAE)的对抗性去偏置框架,以将这些嵌入转换到一个新的潜在空间,在该空间中不再编码人口统计数据,同时保持关键下游任务的性能。我们在NLST肺癌筛查数据集上验证了我们的方法,结果显示,去偏置嵌入有效地消除了多种编码的人口统计数据,并在1年和2年间隔内的肺癌风险预测准确性方面未损害性能。此外,我们的方法确保嵌入能够抵御对抗性偏置攻击。这些结果突显了对抗性去偏置技术确保自监督3D CT嵌入在临床应用中的公平性和平等性的潜力,为其在无偏医学决策中的更广泛采用铺平了道路。