Abstract:Segmenting prostate from magnetic resonance imaging (MRI) is a critical procedure in prostate cancer staging and treatment planning. Considering the nature of labeled data scarcity for medical images, semi-supervised learning (SSL) becomes an appealing solution since it can simultaneously exploit limited labeled data and a large amount of unlabeled data. However, SSL relies on the assumption that the unlabeled images are abundant, which may not be satisfied when the local institute has limited image collection capabilities. An intuitive solution is to seek support from other centers to enrich the unlabeled image pool. However, this further introduces data heterogeneity, which can impede SSL that works under identical data distribution with certain model assumptions. Aiming at this under-explored yet valuable scenario, in this work, we propose a separated collaborative learning (SCL) framework for semi-supervised prostate segmentation with multi-site unlabeled MRI data. Specifically, on top of the teacher-student framework, SCL exploits multi-site unlabeled data by: (i) Local learning, which advocates local distribution fitting, including the pseudo label learning that reinforces confirmation of low-entropy easy regions and the cyclic propagated real label learning that leverages class prototypes to regularize the distribution of intra-class features; (ii) External multi-site learning, which aims to robustly mine informative clues from external data, mainly including the local-support category mutual dependence learning, which takes the spirit that mutual information can effectively measure the amount of information shared by two variables even from different domains, and the stability learning under strong adversarial perturbations to enhance robustness to heterogeneity. Extensive experiments on prostate MRI data from six different clinical centers show that our method can effectively generalize SSL on multi-site unlabeled data and significantly outperform other semi-supervised segmentation methods. Besides, we validate the extensibility of our method on the multi-class cardiac MRI segmentation task with data from four different clinical centers.

Category-Level Regularized Unlabeled-to-Labeled Learning for Semi-supervised Prostate Segmentation with Multi-site Unlabeled Data

Separated collaborative learning for semi-supervised prostate segmentation with multi-site heterogeneous unlabeled MRI data

Semi-Supervised Convolutional Vision Transformer with Bi-Level Uncertainty Estimation for Medical Image Segmentation

Uncertainty-Aware Semi-Supervised Learning for Prostate MRI Zonal Segmentation

Location-based Radiology Report-Guided Semi-supervised Learning for Prostate Cancer Detection

Semi-Supervised and Self-Supervised Collaborative Learning for Prostate 3D MR Image Segmentation

All-Around Real Label Supervision: Cyclic Prototype Consistency Learning for Semi-supervised Medical Image Segmentation

SSL^2: Self-Supervised Learning meets Semi-Supervised Learning: Multiple Sclerosis Segmentation in 7T-MRI from large-scale 3T-MRI

An Automatic Ultrasonic Segmentation Method by Two-stage Semi-supervised Learning Strategy

Semi-Automatic Segmentation of Prostate in CT Images Via Coupled Feature Representation and Spatial-Constrained Transductive Lasso

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

Automatic Prostate MR Image Segmentation with Sparse Label Propagation and Domain-Specific Manifold Regularization

Self-Supervised Learning with Limited Labeled Data for Prostate Cancer Detection in High Frequency Ultrasound

Attention-guided multi-scale learning network for automatic prostate and tumor segmentation on MRI

MCICSAM: Monte Carlo-guided Interpolation Consistency Segment Anything Model for Semi-Supervised Prostate Zone Segmentation

Exploring Unlabeled Data in Multiple Aspects for Semi-Supervised MRI Segmentation.

Self-supervised learning via inter-modal reconstruction and feature projection networks for label-efficient 3D-to-2D segmentation

Self-supervised dual-head attentional bootstrap learning network for prostate cancer screening in transrectal ultrasound images

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

Learning Semi-Supervised Medical Image Segmentation from Spatial Registration

Universal Semi-Supervised Learning for Medical Image Classification