Uncertainty-aware Pseudo Label Refinery for Domain Adaptive Semantic Segmentation
Yuxi Wang, Junran Peng, Zhaoxiang Zhang
Abstract
Unsupervised domain adaptation for semantic segmentation aims to assign the pixel-level labels for unlabeled target domain by transferring knowledge from the labeled source domain. A typical self-supervised learning approach generates pseudo labels from the source model and then re-trains the model to fit the target distribution. However, it suffers from noisy pseudo labels due to the existence of domain shift. Related works alleviate this problem by selecting high-confidence predictions, but uncertain classes with low confidence scores have rarely been considered. This informative uncertainty is essential to enhance feature representation and align source and target domains. In this paper, we propose a novel uncertainty-aware pseudo label refinery framework considering two crucial factors simultaneously. First, we progressively enhance the feature alignment model via the target-guided uncertainty rectifying framework. Second, we provide an uncertainty-aware pseudo label assignment strategy without any manually de-signed threshold to reduce the noisy labels. Extensive experiments demonstrate the effectiveness of our proposed approach and achieve state-of-the-art performance on two standard synthetic-2-real tasks.
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.
Cited by top-tier papers28
- Towards Fewer Annotations: Active Learning via Region Impurity and Prediction Uncertainty for Domain Adaptive Semantic SegmentationBinhui Xie, Longhui Yuan, Shuang Li, Chi Harold Liu et al.CVPR 2022 · 89 citations
- SPA: A Graph Spectral Alignment Perspective for Domain AdaptationZhiqing Xiao, Haobo Wang, Ying Jin, Lei Feng et al.NeurIPS 2023 · 60 citations
- Diffusion-based Image Translation with Label Guidance for Domain Adaptive Semantic SegmentationDuo Peng, Ping Hu, Qiuhong Ke, Jun LiuICCV 2023 · 42 citations
- Learning from Future: A Novel Self-Training Framework for Semantic SegmentationYe Du, Yujun Shen, Haochen Wang, Jingjing Fei et al.NeurIPS 2022 · 40 citations
- Remember the Difference: Cross-Domain Few-Shot Semantic Segmentation via Meta-Memory TransferWenjian Wang, Lijuan Duan, Yuxi Wang, Qing En et al.CVPR 2022 · 32 citations
Builds on12
- Confidence Regularized Self-TrainingYang Zou, Zhiding Yu, Xiaofeng Liu, B. V. K. Vijaya Kumar et al.ICCV 2019 · 901 citations
- In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised LearningMamshad Nayeem Rizve, Kevin Duarte, Yogesh S. Rawat, Mubarak ShahICLR 2021 · 630 citations
- Domain Adaptation for Structured Output via Discriminative Patch RepresentationsYi-Hsuan Tsai, Kihyuk Sohn, Samuel Schulter, Manmohan ChandrakerICCV 2019 · 333 citations
- Self-Ensembling With GAN-Based Data Augmentation for Domain Adaptation in Semantic SegmentationJaehoon Choi, Taekyung Kim, Changick KimICCV 2019 · 264 citations
- Constructing Self-Motivated Pyramid Curriculums for Cross-Domain Semantic Segmentation: A Non-Adversarial ApproachQing Lian, Lixin Duan, Fengmao Lv, Boqing GongICCV 2019 · 238 citations
Related papers
- SSAL: Synergizing between Self-Training and Adversarial Learning for Domain Adaptive Object DetectionMuhammad Akhtar Munir, Muhammad Haris Khan, M. Saquib Sarfraz, Mohsen AliNeurIPS 2021 · 53 citations
- Delving into Probabilistic Uncertainty for Unsupervised Domain Adaptive Person Re-identificationJian Han, Ya-Li Li, Shengjin WangAAAI 2022 · 70 citations
- Prototypical Pseudo Label Denoising and Target Structure Learning for Domain Adaptive Semantic SegmentationPan Zhang, Bo Zhang, Ting Zhang, Dong Chen et al.CVPR 2021
- Continuous Pseudo-Label Rectified Domain Adaptive Semantic Segmentation with Implicit Neural RepresentationsRui Gong, Qin Wang, Martin Danelljan, Dengxin Dai et al.CVPR 2023
- Source Data-free Unsupervised Domain Adaptation for Semantic SegmentationMucong Ye, Jing Zhang, Jinpeng Ouyang, Ding YuanACM MM 2021 · 41 citations
