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

来源至关重要:医学影像中源数据集对模型鲁棒性的影响

Source Matters: Source Dataset Impact on Model Robustness in Medical Imaging

IT University of Copenhagen(哥本哈根信息技术大学);Universidad Nacional de San Martín(圣马丁国立大学);CONICET - Universidad Nacional del Litoral(CONICET-洛马国立大学)

Dovile Juodelyte, Yucheng Lu, Amelia Jiménez-Sánchez, Sabrina Bottazzi, Enzo Ferrante, Veronika Cheplygina

AI总结:

该研究针对医学影像迁移学习中性能提升来源不明的问题,提出MICCAT分类法,通过胸部X光和CT数据集实验发现ImageNet与RadImageNet分类性能相当但前者更易过拟合混杂因素,建议重检ImageNet预训练模型鲁棒性并开源了相关代码。

摘要:

迁移学习已成为医学影像分类算法的核心组成部分,通常会利用ImageNet权重。从自然图像到医学图像的域偏移催生了RadImageNet这类替代方案,其分类性能往往与前者相当。然而,目前尚不清楚迁移学习带来的性能提升是源于泛化能力增强还是捷径学习。为解决这一问题,我们提出医学影像情境化混杂因素分类法(Medical Imaging Contextualized Confounder Taxonomy, MICCAT)对混杂因素进行概念化,并使用两个公开的胸部X光和CT数据集,研究该分类体系下的一系列混杂因素——无论是合成的还是从数据中采样得到的。我们发现,ImageNet和RadImageNet的分类性能相当,但ImageNet更容易对混杂因素产生过拟合。我们建议使用ImageNet预训练模型的研究人员通过开展类似实验重新检验模型的鲁棒性。我们的代码和实验可在https://github.com/DovileDo/source-matters获取。

英文摘要:

Transfer learning has become an essential part of medical imaging classification algorithms, often leveraging ImageNet weights. The domain shift from natural to medical images has prompted alternatives such as RadImageNet, often showing comparable classification performance. However, it remains unclear whether the performance gains from transfer learning stem from improved generalization or shortcut learning. To address this, we conceptualize confounders by introducing the Medical Imaging Contextualized Confounder Taxonomy (MICCAT) and investigate a range of confounders across it -- whether synthetic or sampled from the data -- using two public chest X-ray and CT datasets. We show that ImageNet and RadImageNet achieve comparable classification performance, yet ImageNet is much more prone to overfitting to confounders. We recommend that researchers using ImageNet-pretrained models reexamine their model robustness by conducting similar experiments. Our code and experiments are available at https://github.com/DovileDo/source-matters.

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

↑