用无数据依赖的增强方法提升医学影像分割模型在真实场景下的泛化能力
Data-Agnostic Augmentations for Unknown Variations: Out-of-Distribution Generalisation in MRI Segmentation
- 采用MixUp和傅里叶辅助增强,不依赖具体分布变化源
- 在心脏和前列腺MRI上显著提升跨分布泛化性能
- 可直接接入nnU-Net,适合临床部署的模型优化
医学图像分割模型通常在精心筛选的数据集上训练,导致在真实临床环境中因训练与测试分布不匹配而性能下降。尽管数据增强被广泛使用,但传统视觉一致的增强策略难以应对多样化的现实场景。本文系统评估了MixUp和辅助傅里叶增强等替代策略,这些方法无需明确针对特定分布偏移来源,即可有效缓解多种图像变化的影响。实验表明,这些技术显著提升了心脏动态MRI和前列腺MRI分割任务中的分布外泛化能力与鲁棒性。定量分析显示,它们通过增强特征表示的可分性和紧凑性来改善模型表现。此外,将这些方法集成到nnU-Net训练流程中,提供了一种简单有效的方案,显著提高医学分割模型在真实应用中的可靠性。
原文摘要 · Abstract (English)
Medical image segmentation models are often trained on curated datasets, leading to performance degradation when deployed in real-world clinical settings due to mismatches between training and test distributions. While data augmentation techniques are widely used to address these challenges, traditional visually consistent augmentation strategies lack the robustness needed for diverse real-world scenarios. In this work, we systematically evaluate alternative augmentation strategies, focusing on MixUp and Auxiliary Fourier Augmentation. These methods mitigate the effects of multiple variations without explicitly targeting specific sources of distribution shifts. We demonstrate how these techniques significantly improve out-of-distribution generalization and robustness to imaging variations across a wide range of transformations in cardiac cine MRI and prostate MRI segmentation. We quantitatively find that these augmentation methods enhance learned feature representations by promoting separability and compactness. Additionally, we highlight how their integration into nnU-Net training pipelines provides an easy-to-implement, effective solution for enhancing the reliability of medical segmentation models in real-world applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。