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.

Cross-Modality Multi-Atlas Segmentation Using Deep Neural Networks

Cross-Modality Multi-Atlas Segmentation via Deep Registration and Label Fusion

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

Neural Multi-Atlas Label Fusion: Application to Cardiac MR Images

A New Multi-Atlas Based Deep Learning Segmentation Framework with Differentiable Atlas Feature Warping

Learning Cross-Modal Deep Representations for Multi-Modal MR Image Segmentation

Deep Fusion Net for Multi-atlas Segmentation: Application to Cardiac MR Images

Robust Multi-Atlas Label Propagation by Deep Sparse Representation.

MvMM-RegNet: A new image registration framework based on multivariate mixture model and neural network estimation

Multi-Atlas Segmentation of Anatomical Brain Structures Using Hierarchical Hypergraph Learning

DeepAtlas: Joint Semi-Supervised Learning of Image Registration and Segmentation

Multi-Modal Pulmonary Mass Segmentation Network Based on Cross-Modal Spatial Alignment

Novel Multi-Atlas Segmentation by Matrix Completion

AtlasNet: Multi-atlas Non-linear Deep Networks for Medical Image Segmentation

Medical Image Segmentation Based on Multi-Modal Convolutional Neural Network: Study on Image Fusion Schemes

Automatic Brain Labeling Via Multi-Atlas Guided Fully Convolutional Networks.

Cross-modality deep feature learning for brain tumor segmentation

Multi-Atlas Based Segmentation of Brainstem Nuclei from MR Images by Deep Hyper-Graph Learning

Deep coupled registration and segmentation of multimodal whole-brain images

MSMHSA-DeepLab V3+: An Effective Multi-Scale, Multi-Head Self-Attention Network for Dual-Modality Cardiac Medical Image Segmentation