Adaptive Learning of High-Value Regions for Semi-Supervised Medical Image Segmentation
Tao Lei, Ziyao Yang, Xingwu Wang, Yi Wang, Xuan Wang, Feiman Sun, Asoke K. Nandi
Abstract
Existing semi-supervised learning methods typically mitigate the impact of unreliable predictions by suppressing low-confidence regions. However, these methods fail to explore which regions hold higher learning value and how to design adaptive learning strategies for these regions. To address these issues, we propose a novel adaptive learning of high-value regions (ALHVR) framework. By exploiting the diversity of predictions from dual-branch networks, the prediction regions are classified into three groups: reliable stable region, reliable unstable region, and unreliable stable region. For high-value regions (reliable unstable region and unreliable stable region), different training strategies are designed. Specifically, for reliable unstable region, we propose a confidence-guided cross-prototype consistency learning (CG-CPCL) module, which enforces prototype consistency constraints in the feature space. By leveraging confidence information, the high-confidence predictions from one network selectively supervise the low-confidence predictions from the other, thus helping the model learn inter-class discrimination more stably. Additionally, for unreliable stable region, we design a dynamic teacher competition teaching (DTCT) module, which dynamically selects the most reliable pixels as teachers by evaluating the unperturbed predictions from both networks. These selected pixels are then used to supervise perturbed predictions, thereby enhancing the model's learning capability in unreliable region. Experimental results show that our method outperforms state-of-the-art approaches on three public datasets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 71e8f72f-dfc0-4064-9ecb-4531d1024882Cited by top-tier papers1
Ask how each one uses itBuilds on10
- CutMix: Regularization Strategy to Train Strong Classifiers With Localizable FeaturesSangdoo Yun, Dongyoon Han, Sanghyuk Chun, Seong Joon Oh et al.ICCV 2019 · 5,843 citations
- FixMatch: Simplifying Semi-Supervised Learning with Consistency and ConfidenceKihyuk Sohn, David Berthelot, Nicholas Carlini, Zizhao Zhang et al.NeurIPS 2020 · 5,129 citations
- Unsupervised Data Augmentation for Consistency TrainingQizhe Xie, Zihang Dai, Eduard H. Hovy, Thang Luong et al.NeurIPS 2020 · 2,774 citations
- ST++: Make Self-trainingWork Better for Semi-supervised Semantic SegmentationLihe Yang, Wei Zhuo, Lei Qi, Yinghuan Shi et al.CVPR 2022 · 467 citations
- Semi-Supervised Semantic Segmentation With Cross-Consistency TrainingYassine Ouali, Céline Hudelot, Myriam TamiCVPR 2020
Related papers
- C3-SemiSeg: Contrastive Semi-supervised Segmentation via Cross-set Learning and Dynamic Class-balancingYanning Zhou, Hang Xu, Wei Zhang, Bin Gao et al.ICCV 2021 · 87 citations
- Cross-View Mutual Learning for Semi-Supervised Medical Image SegmentationSong Wu, Xiaoyu Wei, Xinyue Chen, Yazhou Ren et al.ACM MM 2024 · 16 citations
- UCC: Uncertainty guided Cross-head Cotraining for Semi-Supervised Semantic SegmentationJiashuo Fan, Bin Gao, Huan Jin, Lihui JiangCVPR 2022 · 77 citations
- Conflict-Based Cross-View Consistency for Semi-Supervised Semantic SegmentationZicheng Wang, Zhen Zhao, Xiaoxia Xing, Dong Xu et al.CVPR 2023
- CFCG: Semi-Supervised Semantic Segmentation via Cross-Fusion and Contour Guidance SupervisionShuo Li, Yue He, Weiming Zhang, Wei Zhang et al.ICCV 2023 · 20 citations
