通过中间层知识蒸馏,让模型避开医学图像中的虚假关联特征。
Preventing Shortcut Learning in Medical Image Analysis through Intermediate Layer Knowledge Distillation from Specialist Teachers
- 用专业教师网络在小样本数据上微调,指导学生网络学习中间层特征。
- 在多个数据集上性能接近无偏基线模型,且对分布外数据仍有效。
- 特别适合标注少、偏差难识别的真实医疗场景。
深度学习模型容易依赖训练数据中虚假相关但无关的特征形成捷径学习,这在医疗图像分析等高风险场景中可能导致模型忽视临床有意义的特征,影响鲁棒性并危及患者安全。我们发现不同类型的捷径(全局弥散型与局部聚焦型)在神经网络各层表现各异,因此可通过针对中间层的策略更有效地抑制。提出一种新型知识蒸馏框架:利用在少量任务相关数据上微调的教师网络,缓解学生网络在大规模含偏数据上训练时的捷径学习问题。在CheXpert、ISIC 2017和SimBA数据集上,使用ResNet-18、AlexNet、DenseNet-121和3D CNN等多种架构进行大量实验,结果表明该方法显著优于传统经验风险最小化、基于增强的偏见缓解以及基于分组的偏见缓解方法。许多情况下,性能达到与在无偏数据上训练的基线模型相当水平,即使在分布外测试数据上也保持稳定。结果证明该方法在真实医疗影像场景中具有实用性,尤其适用于缺乏偏差标注且捷径特征难以事先识别的情形。
原文摘要 · Abstract (English)
Deep learning models are prone to learning shortcut solutions to problems using spuriously correlated yet irrelevant features of their training data. In high-risk applications such as medical image analysis, this phenomenon may prevent models from using clinically meaningful features when making predictions, potentially leading to poor robustness and harm to patients. We demonstrate that different types of shortcuts (those that are diffuse and spread throughout the image, as well as those that are localized to specific areas) manifest distinctly across network layers and can, therefore, be more effectively targeted through mitigation strategies that target the intermediate layers. We propose a novel knowledge distillation framework that leverages a teacher network fine-tuned on a small subset of task-relevant data to mitigate shortcut learning in a student network trained on a large dataset corrupted with a bias feature. Through extensive experiments on CheXpert, ISIC 2017, and SimBA datasets using various architectures (ResNet-18, AlexNet, DenseNet-121, and 3D CNNs), we demonstrate consistent improvements over traditional Empirical Risk Minimization, augmentation-based bias-mitigation, and group-based bias-mitigation approaches. In many cases, we achieve comparable performance with a baseline model trained on bias-free data, even on out-of-distribution test data. Our results demonstrate the practical applicability of our approach to real-world medical imaging scenarios where bias annotations are limited and shortcut features are difficult to identify a priori.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。