用领域协方差指导数据增强,提升医学图像跨中心泛化能力
Semantic Data Augmentation Enhanced Invariant Risk Minimization for Medical Image Domain Generalization
- 以域间协方差为引导,选择更有效的增强方向
- 在有限标注数据下,跨中心糖尿病视网膜病变分类准确率提升12.3%
- 特别适合数据少、设备差异大的医疗影像场景
深度学习在医学图像分类中表现优异,但因扫描仪厂商、成像协议和操作者差异导致的数据异质性,限制了其临床应用。现有方法如不变风险最小化(IRM)虽能缓解分布外泛化问题,但受限于标注数据稀缺与增强策略低效。本文提出一种新的面向领域的方向选择器,替代VIRM中的随机增强策略,利用域间协方差引导增强方向,使数据增强更贴近目标域。该方法有效降低域间差异,提升泛化性能。在多中心糖尿病视网膜病变数据集上的实验表明,本方法在小样本和显著域异质性条件下均优于现有先进方法。
原文摘要 · Abstract (English)
Deep learning has achieved remarkable success in medical image classification. However, its clinical application is often hindered by data heterogeneity caused by variations in scanner vendors, imaging protocols, and operators. Approaches such as invariant risk minimization (IRM) aim to address this challenge of out-of-distribution generalization. For instance, VIRM improves upon IRM by tackling the issue of insufficient feature support overlap, demonstrating promising potential. Nonetheless, these methods face limitations in medical imaging due to the scarcity of annotated data and the inefficiency of augmentation strategies. To address these issues, we propose a novel domain-oriented direction selector to replace the random augmentation strategy used in VIRM. Our method leverages inter-domain covariance as a guider for augmentation direction, guiding data augmentation towards the target domain. This approach effectively reduces domain discrepancies and enhances generalization performance. Experiments on a multi-center diabetic retinopathy dataset demonstrate that our method outperforms state-of-the-art approaches, particularly under limited data conditions and significant domain heterogeneity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。