arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.04741eess.IV
收藏

重新思考医学图像分割中的领域泛化:一张图像即一个领域

Rethinking domain generalization in medical image segmentation: One image as one domain

Nanchang University(南昌大学)

Jin Hong, Bo Liu

AI总结:

本文提出'一张图像即一个领域'假设,通过统一解耦框架实现多源和单源领域泛化,提升分割鲁棒性,实验显示在视盘和视杯分割中Dice分数达84.43%和88.91%。

摘要:

医学图像分割中的领域偏移,特别是在数据来自不同中心时,带来重大挑战。中心内变异性,如不同扫描仪型号或成像协议差异,可导致领域偏移,大小甚至超过中心间差异。为此,我们提出'一张图像即一个领域'(OIOD)假设,将每张图像视为独特领域,实现灵活且稳健的领域泛化。基于此假设,我们开发了统一解耦基于领域泛化的(UniDDG)框架,同时处理多源和单源领域泛化,无需显式领域标签。此方法简化训练,架构固定,不依赖源领域数量,降低复杂度并提升可扩展性。我们将每张输入图像解耦为内容表示和风格代码,然后在批量内交换和组合这些表示用于分割、重建和进一步解耦。通过为每张图像保持独特的风格代码,我们的模型确保内容表示和风格代码的彻底解耦,提高内容表示的领域不变性。此外,我们通过扩展掩码注意力(EMA)保持边界并用风格增强(SA)模拟多样图像风格,提升对领域偏移的鲁棒性。大量实验显示,我们的方法在视盘和视杯分割中实现多源到单中心泛化Dice分数84.43%和88.91%,在前列腺分割中实现86.96%和88.56%,优于当前最先进的领域泛化方法,提供更优的性能和适应性。

英文摘要:

Domain shifts in medical image segmentation, particularly when data comes from different centers, pose significant challenges. Intra-center variability, such as differences in scanner models or imaging protocols, can cause domain shifts as large as, or even larger than, those between centers. To address this, we propose the "one image as one domain" (OIOD) hypothesis, which treats each image as a unique domain, enabling flexible and robust domain generalization. Based on this hypothesis, we develop a unified disentanglement-based domain generalization (UniDDG) framework, which simultaneously handles both multi-source and single-source domain generalization without requiring explicit domain labels. This approach simplifies training with a fixed architecture, independent of the number of source domains, reducing complexity and enhancing scalability. We decouple each input image into content representation and style code, then exchange and combine these within the batch for segmentation, reconstruction, and further disentanglement. By maintaining distinct style codes for each image, our model ensures thorough decoupling of content representations and style codes, improving domain invariance of the content representations. Additionally, we enhance generalization with expansion mask attention (EMA) for boundary preservation and style augmentation (SA) to simulate diverse image styles, improving robustness to domain shifts. Extensive experiments show that our method achieves Dice scores of 84.43% and 88.91% for multi-source to single-center and single-center generalization in optic disc and optic cup segmentation, respectively, and 86.96% and 88.56% for prostate segmentation, outperforming current state-of-the-art domain generalization methods, offering superior performance and adaptability across clinical settings.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑