Rectifying Pseudo Label Learning via Uncertainty Estimation for Domain Adaptive Semantic Segmentation

被引:354
作者
Zheng, Zhedong [1 ]
Yang, Yi [1 ]
机构
[1] Univ Technol Sydney, Australian Artificial Intelligence Inst AAII, Ultimo, NSW, Australia
关键词
Unsupervised domain adaptation; Domain adaptive semantic segmentation; Image segmentation; Uncertainty estimation;
D O I
10.1007/s11263-020-01395-y
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
This paper focuses on the unsupervised domain adaptation of transferring the knowledge from the source domain to the target domain in the context of semantic segmentation. Existing approaches usually regard the pseudo label as the ground truth to fully exploit the unlabeled target-domain data. Yet the pseudo labels of the target-domain data are usually predicted by the model trained on the source domain. Thus, the generated labels inevitably contain the incorrect prediction due to the discrepancy between the training domain and the test domain, which could be transferred to the final adapted model and largely compromises the training process. To overcome the problem, this paper proposes to explicitly estimate the prediction uncertainty during training to rectify the pseudo label learning for unsupervised semantic segmentation adaptation. Given the input image, the model outputs the semantic segmentation prediction as well as the uncertainty of the prediction. Specifically, we model the uncertainty via the prediction variance and involve the uncertainty into the optimization objective. To verify the effectiveness of the proposed method, we evaluate the proposed method on two prevalent synthetic-to-real semantic segmentation benchmarks, i.e., GTA5 -> Cityscapes and SYNTHIA -> Cityscapes, as well as one cross-city benchmark, i.e., Cityscapes -> Oxford RobotCar. We demonstrate through extensive experiments that the proposed approach (1) dynamically sets different confidence thresholds according to the prediction variance, (2) rectifies the learning from noisy pseudo labels, and (3) achieves significant improvements over the conventional pseudo label learning and yields competitive performance on all three benchmarks.
引用
收藏
页码:1106 / 1120
页数:15
相关论文
共 52 条
  • [11] He K, P IEEE C COMP VIS PA, P770, DOI [DOI 10.1109/CVPR.2016.90, 10.1109/CVPR.2016.90]
  • [12] Identity Mappings in Deep Residual Networks
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. COMPUTER VISION - ECCV 2016, PT IV, 2016, 9908 : 630 - 645
  • [13] Hendrycks D, 2019, P INT C LEARN REPR
  • [14] Hoffman J, 2018, PR MACH LEARN RES, V80
  • [15] Domain Transfer Through Deep Activation Matching
    Huang, Haoshuo
    Huang, Qixing
    Krahenbuhl, Philipp
    [J]. COMPUTER VISION - ECCV 2018, PT XVI, 2018, 11220 : 611 - 626
  • [16] Kang Guoliang, 2020, NeurIPS, V33, P2
  • [17] Kendall A, 2017, 31 ANN C NEURAL INFO, V30
  • [18] Lee D-H, 2013, WORKSH CHALL REPR LE, P896
  • [19] LI P., 2020, NeurIPS
  • [20] Li Peike, 2020, ACM MULTIMEDIA