Hard-aware Instance Adaptive Self-training for Unsupervised Cross-domain Semantic Segmentation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhu, Chuang, Liu, Kebin, Tang, Wenqi, Mei, Ke, Zou, Jiaqi, Huang, Tiejun
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910890438164480
author Zhu, Chuang
Liu, Kebin
Tang, Wenqi
Mei, Ke
Zou, Jiaqi
Huang, Tiejun
author_facet Zhu, Chuang
Liu, Kebin
Tang, Wenqi
Mei, Ke
Zou, Jiaqi
Huang, Tiejun
contents 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 to Cityscapes, SYNTHIA to Cityscapes, and Cityscapes to Oxford RobotCar demonstrate the superior performance of our approach compared with the state-of-the-art methods. Our codes are available at https://github.com/bupt-ai-cz/HIAST.
format Preprint
id arxiv_https___arxiv_org_abs_2302_06992
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Hard-aware Instance Adaptive Self-training for Unsupervised Cross-domain Semantic Segmentation
Zhu, Chuang
Liu, Kebin
Tang, Wenqi
Mei, Ke
Zou, Jiaqi
Huang, Tiejun
Computer Vision and Pattern Recognition
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 to Cityscapes, SYNTHIA to Cityscapes, and Cityscapes to Oxford RobotCar demonstrate the superior performance of our approach compared with the state-of-the-art methods. Our codes are available at https://github.com/bupt-ai-cz/HIAST.
title Hard-aware Instance Adaptive Self-training for Unsupervised Cross-domain Semantic Segmentation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2302.06992