Semi-supervised Medical Image Segmentation via Geometry-aware Consistency Training
Not on the map. Why?
Abstract (arXiv)
The performance of supervised deep learning methods for medical image segmentation is often limited by the scarcity of labeled data. As a promising research direction, semi-supervised learning addresses this dilemma by leveraging unlabeled data information to assist the learning process. In this paper, a novel geometry-aware semi-supervised learning framework is proposed for medical image segmentation, which is a consistency-based method. Considering that the hard-to-segment regions are mainly located around the object boundary, we introduce an auxiliary prediction task to learn the global geometric information. Based on the geometric constraint, the ambiguous boundary regions are emphasized through an exponentially weighted strategy for the model training to better exploit both labeled and unlabeled data. In addition, a dual-view network is designed to perform segmentation from different perspectives and reduce the prediction uncertainty. The proposed method is evaluated on the public left atrium benchmark dataset and improves fully supervised method by 8.7% in Dice with 10% labeled images, while 4.3% with 20% labeled images. Meanwhile, our framework outperforms six state-of-the-art semi-supervised segmentation methods.
Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 27 of its 51 references that are in the map. First posted on arXiv 12 Feb 2022.
Builds on
27 of 51 references.
References · 27
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,845
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,284
- 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,471
- 2017A Survey on Deep Learning in Medical Image Analysiscited by 1,182
- 2016Temporal Ensembling for Semi-Supervised Learningcited by 1,054
- 2019UNet++: Redesigning Skip Connections to Exploit Multiscale Features in Image Segmentationcited by 583
- 2019Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentationcited by 378
- 2020Semi-supervised Medical Image Segmentation through Dual-task Consistencycited by 215
- 2020Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Imagescited by 152
- 2019Transformation Consistent Self-ensembling Model for Semi-supervised Medical Image Segmentationcited by 138
- 2016Unsupervised domain adaptation in brain lesion segmentation with adversarial networkscited by 125
15 more
- 2020Efficient Semi-Supervised Gross Target Volume of Nasopharyngeal Carcinoma Segmentation via Uncertainty Rectified Pyramid Consistencycited by 102
- 2018Weakly-Supervised Convolutional Neural Networks for Multimodal Image Registrationcited by 90
- 2020Uncertainty-aware multi-view co-training for semi-supervised medical image segmentation and domain adaptationcited by 81
- 2020CA-Net: Comprehensive Attention Convolutional Neural Networks for Explainable Medical Image Segmentationcited by 79
- 2017Unsupervised Reverse Domain Adaptation for Synthetic Medical Images via Adversarial Trainingcited by 57
- 20183D Semi-Supervised Learning with Uncertainty-Aware Multi-View Co-Trainingcited by 57
- 2020Double-Uncertainty Weighted Method for Semi-supervised Learningcited by 44
- 2019Distance Map Loss Penalty Term for Semantic Segmentationcited by 27
- 2021Dual-Consistency Semi-Supervised Learning with Uncertainty Quantification for COVID-19 Lesion Segmentation from CT Imagescited by 22
- 2017Discriminative Localization in CNNs for Weakly-Supervised Segmentation of Pulmonary Nodulescited by 20
- 2021Focus U-Net: A novel dual attention-gated CNN for polyp segmentation during colonoscopycited by 20
- 2017Weakly-supervised localization of diabetic retinopathy lesions in retinal fundus imagescited by 18
- 2020Self-paced and self-consistent co-training for semi-supervised image segmentationcited by 18
- 2018Unsupervised Domain Adaptation for Automatic Estimation of Cardiothoracic Ratiocited by 16
- 2021Duo-SegNet: Adversarial Dual-Views for Semi-Supervised Medical Image Segmentationcited by 14
Built on by
Citing papers per year: 2023: 2.
2 papers.