arXiv:2506.18070cs.CV2025-06中稿 · MICCAI 2025被引 2

无需训练,仅用少量图像标签即可提升医学图像解释模型的泛化能力。

Training-free Test-time Improvement for Explainable Medical Image Classification

  • 利用少量新数据中的图像标签,识别并修正混淆概念
  • 在皮肤和白细胞图像上实现跨域性能提升,源域准确率不变
  • 适合资源有限、难以获取专家标注的医疗场景

基于深度学习的医学图像分类技术快速发展,亟需高效部署且可信的模型。概念瓶颈模型(CBMs)通过先预测可解释概念再进行分类,提升了模型可解释性,但在新环境部署时面临概念分布偏移问题,如成像协议或染色方法差异导致的颜色和尺度变化。此外,由于CBM训练依赖显式概念标注,仅用图像级标签微调会损害概念预测的准确性与忠实性,而医学领域专家标注成本高昂。为此,本文提出一种无需训练的概念混淆识别策略:仅需每类4张新图像及图像级标签,通过掩蔽误激活的混淆概念、增强弱激活的判别性概念,提升跨域性能,同时保持源域准确率。该方法在皮肤和白血病细胞图像上验证有效。代码已开源。

原文摘要 · Abstract (English)

Deep learning-based medical image classification techniques are rapidly advancing in medical image analysis, making it crucial to develop accurate and trustworthy models that can be efficiently deployed across diverse clinical scenarios. Concept Bottleneck Models (CBMs), which first predict a set of explainable concepts from images and then perform classification based on these concepts, are increasingly being adopted for explainable medical image classification. However, the inherent explainability of CBMs introduces new challenges when deploying trained models to new environments. Variations in imaging protocols and staining methods may induce concept-level shifts, such as alterations in color distribution and scale. Furthermore, since CBM training requires explicit concept annotations, fine-tuning models solely with image-level labels could compromise concept prediction accuracy and faithfulness - a critical limitation given the high cost of acquiring expert-annotated concept labels in medical domains. To address these challenges, we propose a training-free confusion concept identification strategy. By leveraging minimal new data (e.g., 4 images per class) with only image-level labels, our approach enhances out-of-domain performance without sacrificing source domain accuracy through two key operations: masking misactivated confounding concepts and amplifying under-activated discriminative concepts. The efficacy of our method is validated on both skin and white blood cell images. Our code is available at: https://github.com/riverback/TF-TTI-XMed.

医学图像可解释性零样本迁移概念瓶颈

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。