A modality‐collaborative convolution and transformer hybrid network for unpaired multi‐modal medical image segmentation with limited annotations

Abstract

AbstractBackgroundMulti‐modal learning is widely adopted to learn the latent complementary information between different modalities in multi‐modal medical image segmentation tasks. Nevertheless, the traditional multi‐modal learning methods require spatially well‐aligned and paired multi‐modal images for supervised training, which cannot leverage unpaired multi‐modal images with spatial misalignment and modality discrepancy. For training accurate multi‐modal segmentation networks using easily accessible and low‐cost unpaired multi‐modal images in clinical practice, unpaired multi‐modal learning has received comprehensive attention recently.PurposeExisting unpaired multi‐modal learning methods usually focus on the intensity distribution gap but ignore the scale variation problem between different modalities. Besides, within existing methods, shared convolutional kernels are frequently employed to capture common patterns in all modalities, but they are typically inefficient at learning global contextual information. On the other hand, existing methods highly rely on a large number of labeled unpaired multi‐modal scans for training, which ignores the practical scenario when labeled data is limited. To solve the above problems, we propose a modality‐collaborative convolution and transformer hybrid network (MCTHNet) using semi‐supervised learning for unpaired multi‐modal segmentation with limited annotations, which not only collaboratively learns modality‐specific and modality‐invariant representations, but also could automatically leverage extensive unlabeled scans for improving performance.MethodsWe make three main contributions to the proposed method. First, to alleviate the intensity distribution gap and scale variation problems across modalities, we develop a modality‐specific scale‐aware convolution (MSSC) module that can adaptively adjust the receptive field sizes and feature normalization parameters according to the input. Secondly, we propose a modality‐invariant vision transformer (MIViT) module as the shared bottleneck layer for all modalities, which implicitly incorporates convolution‐like local operations with the global processing of transformers for learning generalizable modality‐invariant representations. Third, we design a multi‐modal cross pseudo supervision (MCPS) method for semi‐supervised learning, which enforces the consistency between the pseudo segmentation maps generated by two perturbed networks to acquire abundant annotation information from unlabeled unpaired multi‐modal scans.ResultsExtensive experiments are performed on two unpaired CT and MR segmentation datasets, including a cardiac substructure dataset derived from the MMWHS‐2017 dataset and an abdominal multi‐organ dataset consisting of the BTCV and CHAOS datasets. Experiment results show that our proposed method significantly outperforms other existing state‐of‐the‐art methods under various labeling ratios, and achieves a comparable segmentation performance close to single‐modal methods with fully labeled data by only leveraging a small portion of labeled data. Specifically, when the labeling ratio is 25%, our proposed method achieves overall mean DSC values of 78.56% and 76.18% in cardiac and abdominal segmentation, respectively, which significantly improves the average DSC value of two tasks by 12.84% compared to single‐modal U‐Net models.ConclusionsOur proposed method is beneficial for reducing the annotation burden of unpaired multi‐modal medical images in clinical applications.