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.

LSRML: A latent space regularization based meta-learning framework for MR image segmentation

MsVRL: Self-Supervised Multiscale Visual Representation Learning Via Cross-Level Consistency for Medical Image Segmentation

A feature regularization based meta-learning framework for generalizing prostate mri segmentation

Domain Adaptation Meets Zero-Shot Learning: an Annotation-Efficient Approach to Multi-Modality Medical Image Segmentation

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

ACL-Net: Adaptive and Collaborative Learning Network for Multi-Site Prostate MRI Segmentation.

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

Shape-aware Meta-learning for Generalizing Prostate MRI Segmentation to Unseen Domains

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

Semi-supervised Meta-learning with Disentanglement for Domain-generalised Medical Image Segmentation

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

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

MMSeg: A novel multi-task learning framework for class imbalance and label scarcity in medical image segmentation

Consistency Label-Activated Region Generating Network for Weakly Supervised Medical Image Segmentation

Multi-step Segmentation for Prostate MR Image Based on Reinforcement Learning

MAPSeg: Unified Unsupervised Domain Adaptation for Heterogeneous Medical Image Segmentation Based on 3D Masked Autoencoding and Pseudo-Labeling

Linear semantic transformation for semi-supervised medical image segmentation

Learning Better Registration to Learn Better Few-Shot Medical Image Segmentation: Authenticity, Diversity, and Robustness

Advancing MRI Segmentation with CLIP-driven Semi-Supervised Learning and Semantic Alignment

Collaborative Learning for Annotation-Efficient Volumetric MR Image Segmentation

MRL-Seg: Overcoming Imbalance in Medical Image Segmentation With Multi-Step Reinforcement Learning