Abstract:Automated brain lesion segmentation provides valuable information for the analysis and intervention of patients. In particular, methods that are based on convolutional neural networks (CNNs) have achieved state-of-the-art segmentation performance. However, CNNs usually require a decent amount of annotated data, which may be costly and time-consuming to obtain. Since unannotated data is generally abundant, it is desirable to use unannotated data to improve the segmentation performance for CNNs when limited annotated data is available. In this work, we propose a semi-supervised learning (SSL) approach to brain lesion segmentation, where unannotated data is incorporated into the training of CNNs. We adapt the mean teacher model, which is originally developed for SSL-based image classification, for brain lesion segmentation. Assuming that the network should produce consistent outputs for similar inputs, a loss of segmentation consistency is designed and integrated into a self-ensembling framework. Self-ensembling exploits the information in the intermediate training steps, and the ensemble prediction based on the information can be closer to the correct result than the single latest model. To exploit such information, we build a student model and a teacher model, which share the same CNN architecture for segmentation. The student and teacher models are updated alternately. At each step, the student model learns from the teacher model by minimizing the weighted sum of the segmentation loss computed from annotated data and the segmentation consistency loss between the teacher and student models computed from unannotated data. Then, the teacher model is updated by combining the updated student model with the historical information of teacher models using an exponential moving average strategy. For demonstration, the proposed approach was evaluated on ischemic stroke lesion segmentation. Results indicate that the proposed method improves stroke lesion segmentation with the incorporation of unannotated data and outperforms competing SSL-based methods.

InterTeach: A Novel Approach for Semi-Supervised Medical Image Segmentation Using Cooperative Teacher-Student Networks

MedFCT: A Frequency Domain Joint CNN-Transformer Network for Semi-supervised Medical Image Segmentation

Robust Semi-supervised 3D Medical Image Segmentation with Diverse Joint-task Learning and Decoupled Inter-student Learning

ACT-Net: Asymmetric Co-Teacher Network for Semi-supervised Memory-efficient Medical Image Segmentation

Dual-Teacher Ensemble Models with Double-Copy-Paste for 3D Semi-Supervised Medical Image Segmentation

Specific Instance and Cross-Prompt-Based Robust 3-D Semi-Supervised Medical Image Segmentation

Diversity matters: Cross-head mutual mean-teaching for semi-supervised medical image segmentation

Dual-Task Mutual Learning for Semi-Supervised Medical Image Segmentation

MSKD: Structured knowledge distillation for efficient medical image segmentation

A Medical Image Segmentation Method Combining Knowledge Distillation and Contrastive Learning

Role Exchange-Based Self-Training Semi-Supervision Framework for Complex Medical Image Segmentation

A Teacher-Student Framework for Semi-supervised Medical Image Segmentation From Mixed Supervision

Cross co-teaching for semi-supervised medical image segmentation

MTANS: Multi-Scale Mean Teacher Combined Adversarial Network with Shape-Aware Embedding for Semi-Supervised Brain Lesion Segmentation

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

SC-SSL: Self-correcting Collaborative and Contrastive Co-training Model for Semi-Supervised Medical Image Segmentation

Semi-supervised medical image segmentation via a tripled-uncertainty guided mean teacher model with contrastive learning

Hybrid dual mean-teacher network with double-uncertainty guidance for semi-supervised segmentation of magnetic resonance images

Alternate Diverse Teaching for Semi-supervised Medical Image Segmentation

3D Medical Image Segmentation with Sparse Annotation via Cross-Teaching between 3D and 2D Networks

Semi-Supervised Brain Lesion Segmentation With An Adapted Mean Teacher Model