Hard-Aware Instance Adaptive Self-Training for Unsupervised Cross-Domain Semantic Segmentation.

Journal: IEEE transactions on pattern analysis and machine intelligence
Published Date:

Abstract

The divergence between labeled training data and unlabeled testing data is a significant challenge for recent deep learning models. Unsupervised domain adaptation (UDA) attempts to solve such problem. Recent works show that self-training is a powerful approach to UDA. However, existing methods have difficulty in balancing the scalability and performance. In this paper, we propose a hard-aware instance adaptive self-training framework for UDA on the task of semantic segmentation. To effectively improve the quality and diversity of pseudo-labels, we develop a novel pseudo-label generation strategy with an instance adaptive selector. We further enrich the hard class pseudo-labels with inter-image information through a skillfully designed hard-aware pseudo-label augmentation. Besides, we propose the region-adaptive regularization to smooth the pseudo-label region and sharpen the non-pseudo-label region. For the non-pseudo-label region, consistency constraint is also constructed to introduce stronger supervision signals during model optimization. Our method is so concise and efficient that it is easy to be generalized to other UDA methods. Experiments on GTA5 $\rightarrow$→ Cityscapes, SYNTHIA $\rightarrow$→ Cityscapes, and Cityscapes $\rightarrow$→ Oxford RobotCar demonstrate the superior performance of our approach compared with the state-of-the-art methods.

Authors

  • Chuang Zhu
    The Center for Data Science, the Beijing Key Laboratory of Network System Architecture and Convergence, the School of Information and Communication Engineering, Beijing University of Posts and Telecommunications, Xitucheng Road, Beijing, China. czhu@bupt.edu.cn.
  • Kebin Liu
  • Wenqi Tang
  • Ke Mei
    School of Information and Communication Engineering, Beijing University of Posts and Telecommunications, Beijing, China.
  • Jiaqi Zou
  • Tiejun Huang
    National Engineering Laboratory for Video Technology, School of Electronics Engineering and Computer Science, Peking University, Beijing, China; Peng Cheng Laboratory, Shenzhen, China.

Keywords

No keywords available for this article.