arXiv:2605.03098cs.CV2026-05

针对CT/MRI脊柱分割,提出高效数据增强方案提升跨模态泛化能力。

One Sequence to Segment Them All: Efficient Data Augmentation for CT and MRI Cross-Domain 3D Spine Segmentation

论文配图:One Sequence to Segment Them All: Efficient Data Augmentation for CT and MRI Cross-Domain 3D Spine Segmentation
图 1 · 摘自论文原文
  • 设计专用于跨模态的增强策略,提升模型对未知序列/对比度的适应性。
  • 在7个跨域测试集上平均Dice提升155%,训练域性能几乎无损(下降0.008%)。
  • 开源工具包支持nnUNet/MONAI,增强效率约10%,适合临床部署场景。

基于深度学习的医学图像分割日益用于辅助临床诊断和新疗法开发。然而,模型性能受限于高质量标注数据稀缺及跨成像协议泛化不足。这一问题在CT与MRI中尤为突出:模型通常仅在单一序列上训练,面对未见序列或对比度时鲁棒性下降。尽管数据增强被广泛用于提升医学图像模型鲁棒性,其在跨模态泛化中的作用尚未量化评估。本文研究一组针对性数据增强技术以改善跨模态迁移能力。我们在三个单模态/序列数据集上分别训练脊柱分割模型,并在七个分布外数据集(涵盖CT与MRI)上评估性能,模拟真实场景中单序列训练、多序列/对比度/模态部署。结果表明,模型在未见域上获得显著性能提升(平均Dice增益155%),同时保持训练域精度(平均下降0.008%),包括成功实现从CT到MRI的跨模态迁移。为降低强增强带来的计算开销,我们实现GPU优化增强,使训练效率提升约10%。代码已开源,可无缝集成至nnUNet和MONAI等主流框架。该方法显著增强对异质临床影像场景的鲁棒性,且不牺牲训练速度。

原文摘要 · Abstract (English)

Deep learning-based medical image segmentation is increasingly used to support clinical diagnosis and develop new treatment strategies. However, model performance remains limited by the scarcity of high-quality annotated data and insufficient generalization across imaging protocols. This limitation is particularly evident in MRI and CT, where models are typically trained on a single acquisition sequence and exhibit reduced robustness when applied to unseen sequences or contrasts. Although data augmentation is widely used to improve general robustness on medical images, its impact on cross-modality generalization has not been quantitatively explored. In this work, we study a targeted set of data augmentation techniques designed to improve cross-modality transfer. We train three spine segmentation models, each on a single-modality/sequence dataset, and evaluate them across seven out-of-distribution datasets (spanning CT and MRI), reflecting a realistic single-sequence training and multi-sequence/contrast/modality deployment scenario. Our results demonstrate substantial performance gains on unseen domains (average Dice gain of 155 %) while preserving in-domain accuracy (average Dice decrease of 0.008 %), including effective transfer between CT and MRI. To mitigate the computational cost typically associated with strong data augmentation, we implement GPU-optimized augmentations that maintain, and even improve, training efficiency by approximately 10 %. We release our approach as an open-source toolbox, enabling seamless integration into commonly used frameworks such as nnUNet and MONAI. These augmentations significantly enhance robustness to heterogeneous clinical imaging scenarios without compromising training speed.

医学图像数据增强跨模态脊柱分割

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