Dual-Decoder Consistency via Pseudo-Labels Guided Data Augmentation for Semi-Supervised Medical Image Segmentation
Not on the map. Why?
Abstract (arXiv)
While supervised learning has achieved remarkable success, obtaining large-scale labeled datasets in biomedical imaging is often impractical due to high costs and the time-consuming annotations required from radiologists. Semi-supervised learning emerges as an effective strategy to overcome this limitation by leveraging useful information from unlabeled datasets. In this paper, we present a novel semi-supervised learning method, Dual-Decoder Consistency via Pseudo-Labels Guided Data Augmentation (DCPA), for medical image segmentation. We devise a consistency regularization to promote consistent representations during the training process. Specifically, we use distinct decoders for student and teacher networks while maintain the same encoder. Moreover, to learn from unlabeled data, we create pseudo-labels generated by the teacher networks and augment the training data with the pseudo-labels. Both techniques contribute to enhancing the performance of the proposed method. The method is evaluated on three representative medical image segmentation datasets. Comprehensive comparisons with state-of-the-art semi-supervised medical image segmentation methods were conducted under typical scenarios, utilizing 10% and 20% labeled data, as well as in the extreme scenario of only 5% labeled data. The experimental results consistently demonstrate the superior performance of our method compared to other methods across the three semi-supervised settings. The source code is publicly available at https://github.com/BinYCn/DCPA.git.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 30 of its 48 references that are in the map. First posted on arXiv 31 Aug 2023.
Builds on
30 of 48 references.
References · 30
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2022Generative Adversarial Networkscited by 5,627
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,859
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
- 2019MixMatch: A Holistic Approach to Semi-Supervised Learningcited by 1,391
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
- 2016Temporal Ensembling for Semi-Supervised Learningcited by 1,057
- 2020A Review of Uncertainty Quantification in Deep Learning: Techniques, Applications and Challengescited by 819
18 more
- 2018Meta-Learning for Semi-Supervised Few-Shot Classificationcited by 456
- 2016Regularization With Stochastic Transformations and Perturbations for Deep Semi-Supervised Learningcited by 441
- 2019Uncertainty-aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentationcited by 379
- 2021In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised Learningcited by 301
- 2019CE-Net: Context Encoder Network for 2D Medical Image Segmentationcited by 220
- 2020Semi-supervised Medical Image Segmentation through Dual-task Consistencycited by 216
- 2018Boundary loss for highly unbalanced segmentationcited by 190
- 2020MixText: Linguistically-Informed Interpolation of Hidden Space for Semi-Supervised Text Classificationcited by 187
- 2020Shape-aware Semi-supervised 3D Semantic Segmentation for Medical Imagescited by 152
- 2019Transformation Consistent Self-ensembling Model for Semi-supervised Medical Image Segmentationcited by 140
- 2021Mutual Consistency Learning for Semi-supervised Medical Image Segmentationcited by 108
- 2022Exploring Smoothness and Class-Separation for Semi-supervised Medical Image Segmentationcited by 93
- 2021SimCVD: Simple Contrastive Voxel-Wise Representation Distillation for Semi-Supervised Medical Image Segmentationcited by 86
- 2020Regularizing Deep Networks with Semantic Data Augmentationcited by 73
- 2019Attention Guided Network for Retinal Image Segmentationcited by 32
- 2022DAE-Former: Dual Attention-guided Efficient Transformer for Medical Image Segmentationcited by 27
- 2023RCPS: Rectified Contrastive Pseudo Supervision for Semi-Supervised Medical Image Segmentationcited by 27
- 2020Automatic Data Augmentation for 3D Medical Image Segmentationcited by 22
Built on by
Citing papers per year: 2024: 1.
1 paper.
- 2024A Survey on Mixup Augmentations and Beyondcited by 20
- 2024A Survey on Mixup Augmentations and Beyond8 Sep 2024