用核空间差异度量提升医学图像分类的跨中心泛化能力
A Robust Unsupervised Domain Adaptation Framework for Medical Image Classification Using RKHS-MMD

- 基于再生核希尔伯特空间的MMD损失实现源域与目标域对齐
- 在两个不同医院的胸部X光数据集上准确率显著提升
- 无需标注即可适应新设备数据,适合医疗AI部署场景
医学图像标注因需专业领域知识而成为瓶颈,且不同医疗中心和设备间的差异导致域偏移与模态不一致,限制模型泛化。为此,本文提出一种无监督域自适应框架,结合迁移学习与基于再生核希尔伯特空间的最大均值差异(RKHS-MMD)损失,实现源域与目标域的特征对齐。通过联合优化分类损失与RKHS-MMD损失,该方法在不依赖人工标注的前提下显著提升模型对未标注医学数据的泛化能力。在来自不同医疗机构的两个胸部X光数据集上的实验表明,该方法相比未进行域自适应的模型有显著性能提升。对比研究进一步证明,RKHS-MMD在缩小模态差距方面优于标准MMD,展现出在医学图像分类及先进AI辅助诊断中的有效性。
原文摘要 · Abstract (English)
Labeling medical images is a major bottleneck in the field of medical imaging, as it requires domain-specific expertise, and it gets further complicated due to variability across different medical centers and different imaging devices. Such heterogeneity introduces domain shifts and modality discrepancies, which limits the generalization of trained models. To address this important challenge, we propose an unsupervised domain adaptation framework that combines transfer learning with a Reproducing Kernel Hilbert Space based Maximum Mean Discrepancy loss for the alignment of source and target domains. By jointly optimizing classification and RKHS-MMD losses, the methodology enhances generalization to unannotated medical datasets while diminishing reliance on manual annotation. Experimental evaluations presented on two chest X-ray datasets, which are obtained from different medical centers, show outstanding improvements over models trained without adaptation. Furthermore, we perform a comparative study to see that RKHS-MMD performs better than the standard Maximum Mean Discrepancy in reducing modality gap, emphasizing its effectiveness for medical image classification and also its strong capability in advanced AI-driven medical diagnostics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。