提出新方法解决脑肿瘤分割中多模态数据缺失问题
DC-Seg: Disentangled Contrastive Learning for Brain Tumor Segmentation with Missing Modalities
- 分离解耦图像为解剖结构与模态特异性特征
- 在BraTS 2020上显著优于现有方法,处理缺失模态更鲁棒
- 适合临床数据不全场景下的医学图像分割任务
准确的脑部图像分割通常需要融合多模态影像的互补信息,但临床数据中各模态并非总能获取,带来挑战。现有方法将多模态编码至共享潜在空间,效果有限,因各模态包含独特有价值信息。本文提出DC-Seg(解耦对比学习用于分割),通过解剖对比学习和模态对比学习,显式分离图像为模态不变的解剖表示与模态特定表示,提升解剖与模态特征的分离能力,增强对模态缺失的鲁棒性。进一步引入基于分割的正则化项,强化模型泛化能力。在BraTS 2020和一个私有白质高信号(WMH)分割数据集上的实验表明,DC-Seg在不同缺失模态设置下均优于当前最优方法,且在WMH分割中表现优异。代码已公开于https://github.com/CuCl-2/DC-Seg。
原文摘要 · Abstract (English)
Accurate segmentation of brain images typically requires the integration of complementary information from multiple image modalities. However, clinical data for all modalities may not be available for every patient, creating a significant challenge. To address this, previous studies encode multiple modalities into a shared latent space. While somewhat effective, it remains suboptimal, as each modality contains distinct and valuable information. In this study, we propose DC-Seg (Disentangled Contrastive Learning for Segmentation), a new method that explicitly disentangles images into modality-invariant anatomical representation and modality-specific representation, by using anatomical contrastive learning and modality contrastive learning respectively. This solution improves the separation of anatomical and modality-specific features by considering the modality gaps, leading to more robust representations. Furthermore, we introduce a segmentation-based regularizer that enhances the model's robustness to missing modalities. Extensive experiments on the BraTS 2020 and a private white matter hyperintensity(WMH) segmentation dataset demonstrate that DC-Seg outperforms state-of-the-art methods in handling incomplete multimodal brain tumor segmentation tasks with varying missing modalities, while also demonstrate strong generalizability in WMH segmentation. The code is available at https://github.com/CuCl-2/DC-Seg.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。