Abstract:Background and objective: Training convolutional neural networks based on large amount of labeled data has made great progress in the field of image segmentation. However, in medical image segmentation tasks, annotating the data is expensive and time-consuming because pixel-level annotation requires experts in the relevant field. Currently, the combination of consistent regularization and pseudo labeling-based semi-supervised methods has shown good performance in image segmentation. However, in the training process, a portion of low-confidence pseudo labels are generated by the model. And the semi-supervised segmentation method still has the problem of distribution bias between labeled and unlabeled data. The objective of this study is to address the challenges of semi-supervised learning and improve the segmentation accuracy of semi-supervised models on medical images. Methods: To address these issues, we propose an Uncertainty-based Region Clipping Algorithm for semi-supervised medical image segmentation, which consists of two main modules. A module is introduced to compute the uncertainty of two sub-networks predictions with diversity using Monte Carlo Dropout, allowing the model to gradually learn from more reliable targets. To retain model diversity, we use different loss functions for different branches and use Non-Maximum Suppression in one of the branches. The other module is proposed to generate new samples by masking the low-confidence pixels in the original image based on uncertainty information. New samples are fed into the model to facilitate the model to generate pseudo labels with high confidence and enlarge the training data distribution. Results: Comprehensive experiments on the combination of two benchmarks ACDC and BraTS2019 show that our proposed model outperforms state-of-the-art methods in terms of Dice, HD95 and ASD. The results reach an average Dice score of 87.86 % and a HD95 score of 4.214 mm on ACDC dataset. For the brain tumor segmentation, the results reach an average Dice score of 84.79 % and a HD score of 10.13 mm. Conclusions: Our proposed method improves the accuracy of semi-supervised medical image segmentation. Extensive experiments on two public medical image datasets including 2D and 3D modalities demonstrate the superiority of our model. The code is available at: https://github.com/QuintinDong/URCA.

ASC-Net: Unsupervised Medical Anomaly Segmentation Using an Adversarial-based Selective Cutting Network

ASC-Net : Adversarial-based Selective Network for Unsupervised Anomaly Segmentation

Many Birds, One Stone: Medical Image Segmentation with Multiple Partially Labeled Datasets

Generative Adversarial Semi-Supervised Network for Medical Image Segmentation

WNET: an End-to-End Atlas-Guided and Boundary-Enhanced Network for Medical Image Segmentation

Unsupervised Brain Tumor Segmentation Using a Symmetric-Driven Adversarial Network

Unsupervised Medical Image Segmentation with Adversarial Networks: From Edge Diagrams to Segmentation Maps

ACA U-Net: Asymmetric Channel Attention Mechanism for Semantic Segmentation of Breast Ultrasound Images

Multi-scale adversarial learning with difficult region supervision learning models for primary tumor segmentation

Self-supervised Pseudo Multi-class Pre-training for Unsupervised Anomaly Detection and Segmentation in Medical Images

Spatial-Frequency Dual Progressive Attention Network For Medical Image Segmentation

URCA: Uncertainty-based region clipping algorithm for semi-supervised medical image segmentation

UGAN: Semi-supervised Medical Image Segmentation Using Generative Adversarial Network

MSE-Nets: Multi-annotated Semi-supervised Ensemble Networks for Improving Segmentation of Medical Image with Ambiguous Boundaries

Unsupervised Anomaly Segmentation for Brain Lesions Using Dual Semantic-Manifold Reconstruction.

Unsupervised Region-based Anomaly Detection in Brain MRI with Adversarial Image Inpainting

Scribble-Supervised Medical Image Segmentation Via Dual-Branch Network and Dynamically Mixed Pseudo Labels Supervision

Manifold-Aware Local Feature Modeling for Semi-Supervised Medical Image Segmentation

Unsupervised Model Adaptation for Source-free Segmentation of Medical Images

IterMask2: Iterative Unsupervised Anomaly Segmentation via Spatial and Frequency Masking for Brain Lesions in MRI

AMSUnet: A neural network using atrous multi-scale convolution for medical image segmentation