Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentation

Abstract

Training deep convolutional neural networks usually requires a large amount of labeled data. However, it is expensive and time-consuming to annotate data for medical image segmentation tasks. In this paper, we present a novel uncertainty-aware semi-supervised framework for left atrium segmentation from 3D MR images. Our framework can effectively leverage the unlabeled data by encouraging consistent predictions of the same input under different perturbations. Concretely, the framework consists of a student model and a teacher model, and the student model learns from the teacher model by minimizing a segmentation loss and a consistency loss with respect to the targets of the teacher model. We design a novel uncertainty-aware scheme to enable the student model to gradually learn from the meaningful and reliable targets by exploiting the uncertainty information. Experiments show that our method achieves high performance gains by incorporating the unlabeled data. Our method outperforms the state-of-the-art semi-supervised methods, demonstrating the potential of our framework for the challenging semi-supervised problems.

Accepted by MICCAI2019; Code is available in https://github.com/yulequan/UA-MT

References in corpus (4)

• What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?
• Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results
• Semi-Supervised Deep Learning for Fully Convolutional Networks
• Semi-supervised Skin Lesion Segmentation via Transformation Consistent Self-ensembling Model

Cited by in corpus (8)

• A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challenges
• Embracing Imperfect Datasets: A Review of Deep Learning Solutions for Medical Image Segmentation
• Active, Continual Fine Tuning of Convolutional Neural Networks for Reducing Annotation Efforts
• Federated Semi-supervised Medical Image Classification via Inter-client Relation Matching
• Double-Uncertainty Weighted Method for Semi-supervised Learning
• Federated Semi-Supervised Learning for COVID Region Segmentation in Chest CT using Multi-National Data from China, Italy, Japan
• Interactive Medical Image Segmentation with Self-Adaptive Confidence Calibration
• A Macro-Micro Weakly-supervised Framework for AS-OCT Tissue Segmentation