将每张医学图像视为独立域,提升模型跨中心分割泛化能力
Rethinking domain generalization in medical image segmentation: One image as one domain
- 每张图像独立建模为一个域,实现灵活泛化
- 在视盘和前列腺分割上分别达84.43%和86.96% Dice分数
- 无需标签即可处理多源与单源场景,适合临床部署
医学图像分割中的域偏移问题,尤其在多中心数据中尤为显著。中心内差异(如扫描仪型号或成像协议)可能导致的域偏移,甚至超过中心间差异。为此,我们提出“一张图像即一个域”(OIOD)假设,将每张图像视为独立域,从而实现灵活且鲁棒的域泛化。基于此,我们构建统一解耦式域泛化框架(UniDDG),无需显式域标签即可同时处理多源与单源域泛化,采用固定架构训练,不随源域数量变化,降低复杂度并增强可扩展性。该方法将输入图像解耦为内容表征与风格码,在批内交换组合以完成分割、重建及进一步解耦。通过为每张图像保留独立风格码,确保内容与风格充分解耦,提升内容表征的域不变性。此外,引入扩展掩码注意力(EMA)以保持边界,结合风格增强(SA)模拟多样化图像风格,提升对域偏移的鲁棒性。大量实验表明,本方法在视盘与视杯分割上分别达到84.43%与88.91%的Dice分数(多源→单中心),在前列腺分割上达86.96%与88.56%,优于现有最先进方法,展现出卓越的临床适应能力。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。