arXiv:2505.19659cs.CV2025-05ICML被引 2

用朗之万动力学生成跨域医学图像,提升分割模型泛化能力。

LangDAug: Langevin Data Augmentation for Multi-Source Domain Generalization in Medical Image Segmentation

  • 基于能量模型和朗之万动力学,在源域间生成中间样本。
  • 在视网膜和前列腺MRI数据集上超越现有最先进方法。
  • 理论证明具正则化效果,适合医疗图像领域泛化任务。

医学图像分割模型常因域间差异难以泛化。领域泛化(DG)方法通过表征学习或数据增强(DAug)来解决。表征学习依赖启发式技术且缺乏理论保障;而数据增强通过合成样本丰富模型表征,表现可媲美甚至优于表征学习。本文提出LangDAug,一种用于二维医学图像分割的多源域泛化的新型朗之万数据增强方法。该方法利用对比散度训练的能量模型(EBM),通过朗之万动力学在源域间迁移,生成中间样本。理论分析表明,LangDAug具有正则化效应,对广义线性模型(GLMs),其雷电纳赫复杂度被数据流形的内在维度所上界。在视网膜分割与2D MRI前列腺分割基准测试中,实验结果表明,LangDAug显著优于现有主流领域泛化方法,并能有效补充现有的域随机化策略。代码已开源:https://github.com/backpropagator/LangDAug。

原文摘要 · Abstract (English)

Medical image segmentation models often struggle to generalize across different domains due to various reasons. Domain Generalization (DG) methods overcome this either through representation learning or data augmentation (DAug). While representation learning methods seek domain-invariant features, they often rely on ad-hoc techniques and lack formal guarantees. DAug methods, which enrich model representations through synthetic samples, have shown comparable or superior performance to representation learning approaches. We propose LangDAug, a novel $\textbf{Lang}$evin $\textbf{D}$ata $\textbf{Aug}$mentation for multi-source domain generalization in 2D medical image segmentation. LangDAug leverages Energy-Based Models (EBMs) trained via contrastive divergence to traverse between source domains, generating intermediate samples through Langevin dynamics. Theoretical analysis shows that LangDAug induces a regularization effect, and for GLMs, it upper-bounds the Rademacher complexity by the intrinsic dimensionality of the data manifold. Through extensive experiments on Fundus segmentation and 2D MRI prostate segmentation benchmarks, we show that LangDAug outperforms state-of-the-art domain generalization methods and effectively complements existing domain-randomization approaches. The codebase for our method is available at https://github.com/backpropagator/LangDAug.

医学图像数据增强域泛化朗之万

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