Abstract:Semi-supervised learning has garnered significant interest as a method to alleviate the burden of data annotation. Recently, semi-supervised medical image segmentation has garnered significant interest that can alleviate the burden of densely annotated data. Substantial advancements have been achieved by integrating consistency-regularization and pseudo-labeling techniques. The quality of the pseudo-labels is crucial in this regard. Unreliable pseudo-labeling can result in the introduction of noise, leading the model to converge to suboptimal solutions. To address this issue, we propose learning from reliable pseudo-labels. In this paper, we tackle two critical questions in learning from reliable pseudo-labels: which pseudo-labels are reliable and how reliable are they? Specifically, we conduct a comparative analysis of two subnetworks to address both challenges. Initially, we compare the prediction confidence of the two subnetworks. A higher confidence score indicates a more reliable pseudo-label. Subsequently, we utilize intra-class similarity to assess the reliability of the pseudo-labels to address the second challenge. The greater the intra-class similarity of the predicted classes, the more reliable the pseudo-label. The subnetwork selectively incorporates knowledge imparted by the other subnetwork model, contingent on the reliability of the pseudo labels. By reducing the introduction of noise from unreliable pseudo-labels, we are able to improve the performance of segmentation. To demonstrate the superiority of our approach, we conducted an extensive set of experiments on three datasets: Left Atrium, Pancreas-CT and Brats-2019. The experimental results demonstrate that our approach achieves state-of-the-art performance. Code is available at: https://github.com/Jiawei0o0/mutual-learning-with-reliable-pseudo-labels .

Effective Semi-Supervised Medical Image Segmentation with Probabilistic Representations and Prototype Learning

Boundary-Aware Prototype in Semi-Supervised Medical Image Segmentation

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

EPL: Evidential Prototype Learning for Semi-supervised Medical Image Segmentation

Uncertainty-aware pseudo-label and consistency for semi-supervised medical image segmentation

Mutually aided uncertainty incorporated Dual Consistency Regularization with Pseudo Label for Semi-Supervised Medical Image Segmentation

Enhancing Weakly Supervised 3D Medical Image Segmentation through Probabilistic-aware Learning

Semi-supervised Medical Image Segmentation via Feature-perturbed Consistency.

Semisupervised Medical Image Segmentation through Prototype-Based Mutual Consistency Learning

Prototype-oriented Contrastive Learning for Semi-Supervised Medical Image Segmentation

PixelSeg: Pixel-by-Pixel Stochastic Semantic Segmentation for Ambiguous Medical Images

Dual uncertainty-guided multi-model pseudo-label learning for semi-supervised medical image segmentation

Uncertainty-aware and Dynamically-Mixed Pseudo-Labels for Semi-Supervised Defect Segmentation

Bayesian Pseudo Labels: Expectation Maximization for Robust and Efficient Semi-Supervised Segmentation

Advancing Medical Image Segmentation Via Self-supervised Instance-adaptive Prototype Learning

Semi-Supervised Medical Image Segmentation Via Dynamic Pseudo-Label Refinement

Shape and boundary-aware multi-branch model for semi-supervised medical image segmentation

An Implicit-Explicit Prototypical Alignment Framework for Semi-Supervised Medical Image Segmentation

Semi-Supervised Semantic Segmentation Using Unreliable Pseudo-Labels

Mutual learning with reliable pseudo label for semi-supervised medical image segmentation

Semi-supervised Defect Segmentation with Uncertainty-aware Pseudo-labels from Multi-branch Network.