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>.

Inherent Consistent Learning for Accurate Semi-supervised Medical Image Segmentation

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

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

Semi-Supervised Medical Image Segmentation Based on Deep Consistent Collaborative Learning

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

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

An Efficient Semi-Supervised Framework with Multi-Task and Curriculum Learning for Medical Image Segmentation

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

Semisupervised Medical Image Segmentation through Prototype-Based Mutual Consistency Learning

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

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

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

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

Affinity-Graph-Guided Contractive Learning for Pretext-Free Medical Image Segmentation with Minimal Annotation

Self-aware and Cross-Sample Prototypical Learning for Semi-supervised Medical Image Segmentation

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

Decoupled Consistency for Semi-supervised Medical Image Segmentation

Transformation Consistent Self-ensembling Model for Semi-supervised Medical Image Segmentation

Combining Contrastive Learning and Shape Awareness for Semi-Supervised Medical Image Segmentation

Exploring Inherent Consistency for Semi-supervised Anatomical Structure Segmentation in Medical Imaging

Deep Compatible Learning for Partially-Supervised Medical Image Segmentation