Abstract:Medical imaging has witnessed remarkable progress but usually requires a large amount of high-quality annotated data which is time-consuming and costly to obtain. To alleviate this burden, semi-supervised learning has garnered attention as a potential solution. In this paper, we present Meta-Learning for Bootstrapping Medical Image Segmentation (MLB-Seg), a novel method for tackling the challenge of semi-supervised medical image segmentation. Specifically, our approach first involves training a segmentation model on a small set of clean labeled images to generate initial labels for unlabeled data. To further optimize this bootstrapping process, we introduce a per-pixel weight mapping system that dynamically assigns weights to both the initialized labels and the model's own predictions. These weights are determined using a meta-process that prioritizes pixels with loss gradient directions closer to those of clean data, which is based on a small set of precisely annotated images. To facilitate the meta-learning process, we additionally introduce a consistency-based Pseudo Label Enhancement (PLE) scheme that improves the quality of the model's own predictions by ensembling predictions from various augmented versions of the same input. In order to improve the quality of the weight maps obtained through multiple augmentations of a single input, we introduce a mean teacher into the PLE scheme. This method helps to reduce noise in the weight maps and stabilize its generation process. Our extensive experimental results on public atrial and prostate segmentation datasets demonstrate that our proposed method achieves state-of-the-art results under semi-supervision. Our code is available at <a class="link-external link-https" href="https://github.com/aijinrjinr/MLB-Seg" rel="external noopener nofollow">this https URL</a>.

Synergistic Label-Stability Learning for Semi-supervised Left Atrium Segmentation

Towards Expert-Amateur Collaboration: Prototypical Label Isolation Learning for Left Atrium Segmentation with Mixed-Quality Labels

Segmentation of Cardiac Tissues and Organs for CCTA Images Based on a Deep Learning Model

Semi-Supervised Learning for Semantic Segmentation of Emphysema With Partial Annotations

MLC: Multi-level consistency learning for semi-supervised left atrium segmentation

DC-net: Dual-Consistency semi-supervised learning for 3D left atrium segmentation from MRI

Leveraging CORAL-Correlation Consistency Network for Semi-Supervised Left Atrium MRI Segmentation

Competitive Ensembling Teacher-Student Framework for Semi-Supervised Left Atrium MRI Segmentation

Anti-Interference From Noisy Labels: Mean-Teacher-Assisted Confident Learning for Medical Image Segmentation

Hierarchical Consistency Regularized Mean Teacher for Semi-supervised 3D Left Atrium Segmentation

Automatic Delineation of the 3D Left Atrium from LGE-MRI: Actor-Critic based Detection and Semi-Supervised Segmentation

Ambiguity-selective consistency regularization for mean-teacher semi-supervised medical image segmentation

Uncertainty-Guided Voxel-Level Supervised Contrastive Learning for Semi-Supervised Medical Image Segmentation

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

Complementary consistency semi-supervised learning for 3D left atrial image segmentation

An improved contrastive learning network for semi-supervised multi-structure segmentation in echocardiography

Consistency-guided Meta-Learning for Bootstrapping Semi-Supervised Medical Image Segmentation

Inherent Consistent Learning for Accurate Semi-supervised Medical Image Segmentation

Semi-supervised Cardiac Image Segmentation via Label Propagation and Style Transfer

Semi-supervised Medical Image Segmentation via Geometry-aware Consistency Training

Label Filling via Mixed Supervision for Medical Image Segmentation from Noisy Annotations