Abstract:Medical image segmentation is a fundamental and critical step in many clinical approaches. Semi-supervised learning has been widely applied to medical image segmentation tasks since it alleviates the heavy burden of acquiring expert-examined annotations and takes the advantage of unlabeled data which is much easier to acquire. Although consistency learning has been proven to be an effective approach by enforcing an invariance of predictions under different distributions, existing approaches cannot make full use of region-level shape constraint and boundary-level distance information from unlabeled data. In this paper, we propose a novel uncertainty-guided mutual consistency learning framework to effectively exploit unlabeled data by integrating intra-task consistency learning from up-to-date predictions for self-ensembling and cross-task consistency learning from task-level regularization to exploit geometric shape information. The framework is guided by the estimated segmentation uncertainty of models to select out relatively certain predictions for consistency learning, so as to effectively exploit more reliable information from unlabeled data. Experiments on two publicly available benchmark datasets showed that: 1) Our proposed method can achieve significant performance improvement by leveraging unlabeled data, with up to 4.13% and 9.82% in Dice coefficient compared to supervised baseline on left atrium segmentation and brain tumor segmentation, respectively. 2) Compared with other semi-supervised segmentation methods, our proposed method achieve better segmentation performance under the same backbone network and task settings on both datasets, demonstrating the effectiveness and robustness of our method and potential transferability for other medical image segmentation tasks.

Beyond Low-Dimensional Features: Enhancing Semi-Supervised Medical Image Semantic Segmentation with Advanced Consistency Learning Techniques

Dual-consistency guidance semi-supervised medical image segmentation with low-level detail feature augmentation

Unified Feature Consistency of Under-Performing Pixels and Valid Regions for Semi-Supervised Medical Image Segmentation

Inherent Consistent Learning for Accurate Semi-supervised Medical Image Segmentation

Uncertainty-Guided Mutual Consistency Learning for Semi-Supervised Medical Image Segmentation

Consistency learning with dynamic weighting and class-agnostic regularization for semi-supervised medical image segmentation

Multi-Consistency Training for Semi-Supervised Medical Image Segmentation

Dual-Decoder Consistency via Pseudo-Labels Guided Data Augmentation for Semi-Supervised Medical Image Segmentation

Contour-aware Consistency for Semi-Supervised Medical Image Segmentation

Decoupled Consistency for Semi-supervised Medical Image Segmentation

Curriculum Consistency Learning and Multi-Scale Contrastive Constraint in Semi-Supervised Medical Image Segmentation

Cross-level Contrastive Learning and Consistency Constraint for Semi-supervised Medical Image Segmentation

Transformation-Consistent Self-Ensembling Model for Semisupervised Medical Image Segmentation

Semi-Supervised 3D Medical Image Segmentation Based on Dual-Task Consistent Joint Learning and Task-Level Regularization.

Semi-Supervised 3D Medical Image Segmentation Using Shape-Guided Dual Consistency Learning

Dual-stream-based dense local features contrastive learning for semi-supervised medical image segmentation

SDMI-Net

Towards Better Semantic Consistency of 2D Medical Image Segmentation

Dual consistency regularization with subjective logic for semi-supervised medical image segmentation

Low-Quality Sensor Data-Based Semi-Supervised Learning for Medical Image Segmentation

Semi-Supervised Medical Image Segmentation Based on Local Consistency and Volume Constancy