Mutual consistency learning for semi-supervised medical image segmentation. 2022

Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
Department of Data Science & AI, Faculty of Information Technology, Monash University, Melbourne, VIC 3800, Australia. Electronic address: yicheng.wu@monash.edu.

In this paper, we propose a novel mutual consistency network (MC-Net+) to effectively exploit the unlabeled data for semi-supervised medical image segmentation. The MC-Net+ model is motivated by the observation that deep models trained with limited annotations are prone to output highly uncertain and easily mis-classified predictions in the ambiguous regions (e.g., adhesive edges or thin branches) for medical image segmentation. Leveraging these challenging samples can make the semi-supervised segmentation model training more effective. Therefore, our proposed MC-Net+ model consists of two new designs. First, the model contains one shared encoder and multiple slightly different decoders (i.e., using different up-sampling strategies). The statistical discrepancy of multiple decoders' outputs is computed to denote the model's uncertainty, which indicates the unlabeled hard regions. Second, we apply a novel mutual consistency constraint between one decoder's probability output and other decoders' soft pseudo labels. In this way, we minimize the discrepancy of multiple outputs (i.e., the model uncertainty) during training and force the model to generate invariant results in such challenging regions, aiming at regularizing the model training. We compared the segmentation results of our MC-Net+ model with five state-of-the-art semi-supervised approaches on three public medical datasets. Extension experiments with two standard semi-supervised settings demonstrate the superior performance of our model over other methods, which sets a new state of the art for semi-supervised medical image segmentation. Our code is released publicly at https://github.com/ycwu1997/MC-Net.

UI MeSH Term Description Entries
D007091 Image Processing, Computer-Assisted A technique of inputting two-dimensional or three-dimensional images into a computer and then enhancing or analyzing the imagery into a form that is more useful to the human observer. Biomedical Image Processing,Computer-Assisted Image Processing,Digital Image Processing,Image Analysis, Computer-Assisted,Image Reconstruction,Medical Image Processing,Analysis, Computer-Assisted Image,Computer-Assisted Image Analysis,Computer Assisted Image Analysis,Computer Assisted Image Processing,Computer-Assisted Image Analyses,Image Analyses, Computer-Assisted,Image Analysis, Computer Assisted,Image Processing, Biomedical,Image Processing, Computer Assisted,Image Processing, Digital,Image Processing, Medical,Image Processings, Medical,Image Reconstructions,Medical Image Processings,Processing, Biomedical Image,Processing, Digital Image,Processing, Medical Image,Processings, Digital Image,Processings, Medical Image,Reconstruction, Image,Reconstructions, Image
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
D000077321 Deep Learning Supervised or unsupervised machine learning methods that use multiple layers of data representations generated by nonlinear transformations, instead of individual task-specific ALGORITHMS, to build and train neural network models. Hierarchical Learning,Learning, Deep,Learning, Hierarchical

Related Publications

Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
April 2023, Artificial intelligence in medicine,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
April 2024, Computers in biology and medicine,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
July 2023, Computers in biology and medicine,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
May 2024, Medical image analysis,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
August 2023, IEEE journal of biomedical and health informatics,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
October 2023, Computers in biology and medicine,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
May 2023, IEEE transactions on medical imaging,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
January 2023, Medical image analysis,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
March 2023, IEEE transactions on medical imaging,
Yicheng Wu, and Zongyuan Ge, and Donghao Zhang, and Minfeng Xu, and Lei Zhang, and Yong Xia, and Jianfei Cai
August 2022, Medical image analysis,
Copied contents to your clipboard!