无需配对数据,用文本嵌入提升多模态医学图像分割效果
MulModSeg: Enhancing Unpaired Multi-Modal Medical Image Segmentation with Modality-Conditioned Text Embedding and Alternating Training
- 用冻结的文本编码器生成带模态信息的嵌入,不改主干结构
- 交替训练让未配对的CT和MR图像共享关键特征,提升分割精度
- 适用于需要跨模态泛化的医疗图像分割任务
在医学影像领域,自动分割需应对多种输入模态,如不同类型的计算机断层扫描(CT)和磁共振(MR)图像。由于缺乏空间对齐的配对数据,现有算法难以在不同模态间保持一致性能。传统分割模型通常仅基于单一模态训练,限制了跨模态泛化能力。为利用多模态互补信息,常需复杂修改主流编码-解码结构。本文提出一种简洁的多模态分割方法(MulModSeg),针对CT与MR图像,在不显著增加结构或计算开销的前提下,引入模态条件文本嵌入框架,通过冻结文本编码器增强模态感知;同时设计交替训练策略,有效融合未配对的CT与MR输入特征。在全卷积网络与基于Transformer的骨干网络上均验证其有效性,显著优于现有方法,在腹部多器官及心脏亚结构分割任务中表现优异。
原文摘要 · Abstract (English)
In the diverse field of medical imaging, automatic segmentation has numerous applications and must handle a wide variety of input domains, such as different types of Computed Tomography (CT) scans and Magnetic Resonance (MR) images. This heterogeneity challenges automatic segmentation algorithms to maintain consistent performance across different modalities due to the requirement for spatially aligned and paired images. Typically, segmentation models are trained using a single modality, which limits their ability to generalize to other types of input data without employing transfer learning techniques. Additionally, leveraging complementary information from different modalities to enhance segmentation precision often necessitates substantial modifications to popular encoder-decoder designs, such as introducing multiple branched encoding or decoding paths for each modality. In this work, we propose a simple Multi-Modal Segmentation (MulModSeg) strategy to enhance medical image segmentation across multiple modalities, specifically CT and MR. It incorporates two key designs: a modality-conditioned text embedding framework via a frozen text encoder that adds modality awareness to existing segmentation frameworks without significant structural modifications or computational overhead, and an alternating training procedure that facilitates the integration of essential features from unpaired CT and MR inputs. Through extensive experiments with both Fully Convolutional Network and Transformer-based backbones, MulModSeg consistently outperforms previous methods in segmenting abdominal multi-organ and cardiac substructures for both CT and MR modalities. The code is available in this {\href{https://github.com/ChengyinLee/MulModSeg_2024}{link}}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。