当前位置: X-MOL 学术Int. J. Comput. Vis. › 论文详情
Our official English website, www.x-mol.net, welcomes your feedback! (Note: you will need to create a separate account there.)
Rectifying Pseudo Label Learning via Uncertainty Estimation for Domain Adaptive Semantic Segmentation
International Journal of Computer Vision ( IF 19.5 ) Pub Date : 2021-01-06 , DOI: 10.1007/s11263-020-01395-y
Zhedong Zheng , Yi Yang

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 $$\rightarrow $$ → Cityscapes and SYNTHIA $$\rightarrow $$ → Cityscapes, as well as one cross-city benchmark, i.e., Cityscapes $$\rightarrow $$ → 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.

中文翻译:

通过域自适应语义分割的不确定性估计来纠正伪标签学习

本文重点研究在语义分割的上下文中将知识从源域转移到目标域的无监督域适应。现有方法通常将伪标签视为充分利用未标记目标域数据的基本事实。然而,目标域数据的伪标签通常由在源域上训练的模型预测。因此,由于训练域和测试域之间的差异,生成的标签不可避免地包含不正确的预测,这可能会转移到最终的适应模型并在很大程度上损害训练过程。为了克服这个问题,本文提出在训练期间明确估计预测不确定性,以纠正无监督语义分割适应的伪标签学习。给定输入图像,模型输出语义分割预测以及预测的不确定性。具体来说,我们通过预测方差对不确定性进行建模,并将不确定性纳入优化目标。为了验证所提出方法的有效性,我们在两个流行的合成到真实语义分割基准上评估了所提出的方法,即 GTA5 $$\rightarrow $$ → Cityscapes 和 SYNTHIA $$\rightarrow $$ → Cityscapes,以及作为一种跨城市基准,即 Cityscapes $$\rightarrow $$ → Oxford RobotCar。我们通过大量实验证明所提出的方法(1)根据预测方差动态设置不同的置信阈值,(2)纠正从嘈杂伪标签中的学习,
更新日期:2021-01-06
down
wechat
bug