HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Unsupervised Domain Adaptation for Semantic Segmentation via Class-Balanced Self-Training

{B. V. K. Vijaya Kumar Yang Zou Zhiding Yu Jinsong Wang}

Unsupervised Domain Adaptation for Semantic Segmentation via Class-Balanced Self-Training

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

BenchmarkMethodologyMetrics
image-to-image-translation-on-gtav-toCBST
mIoU: 47.0
semi-supervised-semantic-segmentation-on-23CBST (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-24CBST (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-25CBST (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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp