Command Palette
Search for a command to run...
Unsupervised Domain Adaptation for Semantic Segmentation via Class-Balanced Self-Training
{B. V. K. Vijaya Kumar Yang Zou Zhiding Yu Jinsong Wang}

Abstract
Recent deep networks achieved state of the art performanceon a variety of semantic segmentation tasks. Despite such progress, thesemodels often face challenges in real world âwild tasksâ where large differ-ence between labeled training/source data and unseen test/target dataexists. In particular, such difference is often referred to as âdomain gapâ,and could cause significantly decreased performance which cannot beeasily remedied by further increasing the representation power. Unsuper-vised domain adaptation (UDA) seeks to overcome such problem withouttarget domain labels. In this paper, we propose a novel UDA frameworkbased on an iterative self-training (ST) procedure, where the problemis formulated as latent variable loss minimization, and can be solved byalternatively generating pseudo labels on target data and re-training themodel with these labels. On top of ST, we also propose a novel class-balanced self-training (CBST) framework to avoid the gradual domi-nance of large classes on pseudo-label generation, and introduce spatialpriors to refine generated labels. Comprehensive experiments show thatthe proposed methods achieve state of the art semantic segmentationperformance under multiple major UDA settings.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| image-to-image-translation-on-gtav-to | CBST | mIoU: 47.0 |
| semi-supervised-semantic-segmentation-on-23 | CBST (Range View) | mIoU (1% Labels): 35.7 mIoU (10% Labels): 50.7 mIoU (20% Labels): 52.7 mIoU (50% Labels): 54.6 |
| semi-supervised-semantic-segmentation-on-24 | CBST (Range View) | mIoU (1% Labels): 39.9 mIoU (10% Labels): 53.4 mIoU (20% Labels): 56.1 mIoU (50% Labels): 56.9 |
| semi-supervised-semantic-segmentation-on-25 | CBST (Range View) | mIoU (1% Labels): 40.9 mIoU (10% Labels): 60.5 mIoU (20% Labels): 64.3 mIoU (50% Labels): 69.3 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.