LLM2D
基于反事实对比学习的鲁棒图像表示
Robust image representations with counterfactual contrastive learning
作者: M\'elanie Roschewitz, Fabio De Sousa Ribeiro, Tian Xia, Galvin Khara, Ben Glocker
发布日期: 4/11/2025
arXiv ID: oai:arXiv.org:2409.10365v2

摘要

arXiv:2409.10365v2 宣告类型: replace-cross 摘要:对比预训练可以显著提高模型的泛化能力和下游性能。然而,学到的表示的质量高度依赖于用于生成正样本对的数据增强策略。正对比样本应该保留语义意义,同时丢弃与数据采集领域相关的不需要的变化。传统对比预训练管道试图通过预定义的通用图像变换来模拟领域转换。然而,这些方法并不总是模拟医疗成像中真实且相关的领域变化,例如扫描仪差异。为了应对这一问题,我们在此引入了反事实对比学习,这是一种新颖的框架,利用了最近在因果图像合成方面取得的进展,以创建能够准确捕捉相关领域变化的对比正样本对。我们的方法在涵盖了胸部X光摄影和乳腺摄影数据的五个数据集中进行了评估,并针对两种已建立的对比目标(SimCLR和DINO-v2)表现出了对采集变换的鲁棒性。值得注意的是,反事实对比学习在内部和外部数据集上的下游性能表现更优,尤其是在训练集中代表性不足的扫描仪采集的图像上。进一步的实验表明,所提出的方法超越了采集变换,使用反事实对比学习训练的模型减少了不同性别亚组之间的差异。