Multi-modal contrastive mutual learning and pseudo-label re-learning for semi-supervised medical image segmentation. 2023

Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
State Key Laboratory of Reliability and Intelligence of Electrical Equipment, School of Health Sciences and Biomedical Engineering, Hebei University of Technology, China; Tianjin Key Laboratory of Bioelectromagnetic Technology and Intelligent Health, School of Health Sciences and Biomedical Engineering, Hebei University of Technology, China.

Semi-supervised learning has a great potential in medical image segmentation tasks with a few labeled data, but most of them only consider single-modal data. The excellent characteristics of multi-modal data can improve the performance of semi-supervised segmentation for each image modality. However, a shortcoming for most existing multi-modal solutions is that as the corresponding processing models of the multi-modal data are highly coupled, multi-modal data are required not only in the training but also in the inference stages, which thus limits its usage in clinical practice. Consequently, we propose a semi-supervised contrastive mutual learning (Semi-CML) segmentation framework, where a novel area-similarity contrastive (ASC) loss leverages the cross-modal information and prediction consistency between different modalities to conduct contrastive mutual learning. Although Semi-CML can improve the segmentation performance of both modalities simultaneously, there is a performance gap between two modalities, i.e., there exists a modality whose segmentation performance is usually better than that of the other. Therefore, we further develop a soft pseudo-label re-learning (PReL) scheme to remedy this gap. We conducted experiments on two public multi-modal datasets. The results show that Semi-CML with PReL greatly outperforms the state-of-the-art semi-supervised segmentation methods and achieves a similar (and sometimes even better) performance as fully supervised segmentation methods with 100% labeled data, while reducing the cost of data annotation by 90%. We also conducted ablation studies to evaluate the effectiveness of the ASC loss and the PReL module.

UI MeSH Term Description Entries
D006801 Humans Members of the species Homo sapiens. Homo sapiens,Man (Taxonomy),Human,Man, Modern,Modern Man
D000069553 Supervised Machine Learning A MACHINE LEARNING paradigm used to make predictions about future instances based on a given set of labeled paired input-output training (sample) data. Active Machine Learning,Inductive Machine Learning,Learning from Labeled Data,Machine Learning with a Teacher,Semi-supervised Learning,Learning, Active Machine,Learning, Inductive Machine,Learning, Semi-supervised,Learning, Supervised Machine,Machine Learning, Active,Machine Learning, Inductive,Machine Learning, Supervised,Semi supervised Learning

Related Publications

Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
May 2024, Medical image analysis,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
January 2024, Mathematical biosciences and engineering : MBE,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
October 2023, IEEE journal of biomedical and health informatics,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
July 2023, Medical image analysis,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
September 2023, Physics in medicine and biology,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
December 2023, Bioengineering (Basel, Switzerland),
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
June 2024, Computers in biology and medicine,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
September 2022, Medical image computing and computer-assisted intervention : MICCAI ... International Conference on Medical Image Computing and Computer-Assisted Intervention,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
July 2016, IEEE transactions on image processing : a publication of the IEEE Signal Processing Society,
Shuo Zhang, and Jiaojiao Zhang, and Biao Tian, and Thomas Lukasiewicz, and Zhenghua Xu
July 2022, Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference,
Copied contents to your clipboard!