Diverse Cotraining Makes Strong Semi-Supervised Segmentor
Yijiang Li, Xinjiang Wang, Lihe Yang, Litong Feng, Wayne Zhang, Ying Gao
Abstract
Deep co-training has been introduced to semisupervised segmentation and achieves impressive results, yet few studies have explored the working mechanism behind it. In this work, we revisit the core assumption that supports co-training: multiple compatible and conditionally independent views. By theoretically deriving the generalization upper bound, we prove the prediction similarity between two models negatively impacts the model's generalization ability. However, most current co-training models are tightly coupled together and violate this assumption. Such coupling leads to the homogenization of networks and confirmation bias which consequently limits the performance. To this end, we explore different dimensions of co-training and systematically increase the diversity from the aspects of input domains, different augmentations and model architectures to counteract homogenization. Our Diverse Co-training outperforms the state-of-theart (SOTA) methods by a large margin across different evaluation protocols on the Pascal and Cityscapes. For example. we achieve the best mIoU of 76.2%, 77.7% and 80.2% on Pascal with only 92, 183 and 366 labeled images, surpassing the previous best results by more than 5%. Code will be available at https://github.com/ williamium3000/diverse-cotraining .
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 5248c72e-6600-4fe5-afac-ebaf0603e1e9Cited by top-tier papers15
- Diffusion Model as Representation LearnerXingyi Yang, Xinchao WangICCV 2023 · 100 citations
- Multi-metrics adaptively identifies backdoors in Federated learningSiquan Huang, Yijiang Li, Chong Chen, Leyu Shi et al.ICCV 2023 · 56 citations
- RankMatch: Exploring the Better Consistency Regularization for Semi-Supervised Semantic SegmentationHuayu Mai, Rui Sun, Tianzhu Zhang, Feng WuCVPR 2024 · 48 citations
- AllSpark: Reborn Labeled Features from Unlabeled in Transformer for Semi-Supervised Semantic SegmentationHaonan Wang, Qixiang Zhang, Yi Li, Xiaomeng LiCVPR 2024 · 39 citations
- ScaleMatch: Multi-scale Consistency Enhancement for Semi-supervised Semantic SegmentationLiang Lv, Lefei ZhangAAAI 2025 · 5 citations
Builds on44
- Swin Transformer: Hierarchical Vision Transformer using Shifted WindowsZe Liu, Yutong Lin, Yue Cao, Han Hu et al.ICCV 2021 · 31,683 citations
- A Simple Framework for Contrastive Learning of Visual RepresentationsTing Chen, Simon Kornblith, Mohammad Norouzi, Geoffrey E. HintonICML 2020 · 24,064 citations
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- SegFormer: Simple and Efficient Design for Semantic Segmentation with TransformersEnze Xie, Wenhai Wang, Zhiding Yu, Anima Anandkumar et al.NeurIPS 2021 · 9,661 citations
- Bootstrap Your Own Latent - A New Approach to Self-Supervised LearningJean-Bastien Grill, Florian Strub, Florent Altché, Corentin Tallec et al.NeurIPS 2020 · 9,171 citations
Related papers
- 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
- Cross-View Mutual Learning for Semi-Supervised Medical Image SegmentationSong Wu, Xiaoyu Wei, Xinyue Chen, Yazhou Ren et al.ACM MM 2024 · 16 citations
- All Labels Are Not Created Equal: Enhancing Semi-Supervision via Label Grouping and Co-TrainingIslam Nassar, Samitha Herath, Ehsan Abbasnejad, Wray L. Buntine et al.CVPR 2021
- Three Ways To Improve Semantic Segmentation With Self-Supervised Depth EstimationLukas Hoyer, Dengxin Dai, Yuhua Chen, Adrian Köring et al.CVPR 2021
