Abstract:Multimodal magnetic resonance imaging (MRI) provides complementary information about targets, and the segmentation of multimodal MRI is widely used as an essential preprocessing step for initial diagnosis, stage differentiation, and post-treatment efficacy evaluation in clinical situations. For the main modality or each of the modalities, it is important to enhance the visual information by modeling the connection and effectively fusing the features among them. However, the existing methods for multimodal segmentation have a drawback; they coincidentally drop information of individual modality during the fusion process. Recently, graph learning-based methods have been applied in segmentation, and these methods have achieved considerable improvements by modeling the relationships across feature regions and reasoning using global information. In this paper, we propose a graph learning-based approach to efficiently extract modality-specific features and establish regional correspondence effectively among all modalities. In detail, after projecting features into a graph domain and employing graph convolution to propagate information across all regions for learning global modality-specific features, we propose a mutual information-based graph co-attention module to learn the weight coefficients of one bipartite graph constructed by the fully connected graphs having different modalities in the graph domain and by selectively fusing the node features. Based on the deformation diagram between the spatial-graph space and our proposed graph co-attention module, we present a multimodal prior-guided segmentation framework, which uses two strategies for two clinical situations: Modality-Specific Learning Strategy and Co-Modality Learning Strategy. Besides, the improved Co-Modality Learning Strategy is used with trainable weights in the multi-task loss for the optimization of the proposed framework. We validated our proposed modules and frameworks on two multimodal MRI datasets: our private liver lesion dataset and a public prostate zone dataset. Our experimental results on both datasets prove the superiority of our proposed approaches.

Coco-Attention for Tumor Segmentation in Weakly Paired Multimodal MRI Images

Mutual Information-Based Graph Co-Attention Networks for Multimodal Prior-Guided Magnetic Resonance Imaging Segmentation

Multimodal Priors Guided Segmentation of Liver Lesions in MRI Using Mutual Information Based Graph Co-Attention Networks.

Tumor Segmentation in Weakly Paired Anatomical and Functional MRI Images with Multimodal Information Fusion

Multimodal Spatial Attention Module for Targeting Multimodal PET-CT Lung Tumor Segmentation

Modality-Pairing Learning for Brain Tumor Segmentation

Cross-modality Synthesis Aiding Lung Tumor Segmentation on Multi-Modal MRI Images

Multi-modal Brain Tumor Segmentation via Missing Modality Synthesis and Modality-level Attention Fusion

Modality-Specific Information Disentanglement from Multi-parametric MRI for Breast Tumor Segmentation and Computer-aided Diagnosis

Unpaired Multi-Modal Tumor Segmentation with Structure Adaptation

Segmentation of Low-Grade Brain Tumors Using Mutual Attention Multimodal MRI

Brain Tumor Segmentation via Multi-Modalities Interactive Feature Learning

Breast tumor segmentation via deep correlation analysis of multi-sequence MRI

Modality-level cross-connection and attentional feature fusion based deep neural network for multi-modal brain tumor segmentation

Modality-Adaptive Feature Interaction for Brain Tumor Segmentation with Missing Modalities

Cross-modality deep feature learning for brain tumor segmentation

Feature-enhanced generation and multi-modality fusion based deep neural network for brain tumor segmentation with missing MR modalities

Attention in Attention for PET-CT Modality Consensus Lung Tumor Segmentation

Learning intra-inter-modality complementary for brain tumor segmentation

Multimodality-Assisted Semi-Supervised Brain Tumor Segmentation in Nondominant Modality Based on Consistency Learning

Joint-phase attention network for breast cancer segmentation in DCE-MRI