Unpaired Multi-Modal Tumor Segmentation with Structure Adaptation

Zhou Pei,Chen Houjin,Li Yanfeng,Peng Yahui
DOI: https://doi.org/10.1007/s10489-022-03610-4
IF: 5.3
2022-01-01
Applied Intelligence
Abstract:Multi-modal learning of unpaired anatomical and functional modalities is challenging because the large discrepancies between feature representations, and the absence of registration, hinder deeper cross-modality information extraction. To address this issue, we propose a novel unpaired multi-modal learning framework for lung tumor segmentation in anatomical and functional images (i.e., T2w-MRI and DWI-MRI). The framework consists of two models: a multi-modal segmentation network (Segmentor) and an adversarial learning-based structure adaptation (SA) model. The Segmentor contains a modality-specific encoder for intensity offset reduction and a shared decoder for cross-modality information fusion. The SA model helps the Segmentor obtain high-level features with similar distributions from different modalities through adversarial training. To address the modality imbalance problem caused by the large gap in the number of training samples and image characteristics, we introduce a harmonic constraint term derived from two weighted dice losses to maintain the balance of multi-modal training. Experimental results based on a clinical dataset of 326 T2w-MRI and 112 DWI-MRI scans show that the proposed framework improves segmentation performance compared with several state-of-the-art unpaired multi-modal segmentation methods.
What problem does this paper attempt to address?