Source-linked AI summary
PixMatch: Unsupervised Domain Adaptation via Pixelwise Consistency Training
Luke Melas-Kyriazi, Arjun K. Manrai
TL;DR
Pixel-level annotation is costly, motivating semantic segmentation models trained on labeled simulated images for use on real domains. PixMatch enforces pixelwise consistency between target-image predictions and predictions for perturbed versions, and its augmentation baseline achieves strong results on two synthetic-to-real benchmarks. The authors report that combining PixMatch with self-training reaches a new state of the art on GTA5-to-Cityscapes.
Problem
Semantic segmentation depends on costly pixel-level annotations, motivating adaptation from labeled simulated domains to unlabeled real domains.
Source: §1 · p. 1Method
PixMatch adds a target-domain loss that enforces pixelwise consistency between predictions on an image and a perturbed version of that image.
Source: §1 · pp. 1–2Results
The heavy-augmentation baseline performs strongly on GTA5-to-Cityscapes and SYNTHIA-to-Cityscapes, and combining PixMatch with self-training achieves a new GTA5-to-Cityscapes state of the art.
Source: §1 · p. 2Takeaways & Limitations
PixMatch is presented as a simple, stable, memory-efficient alternative to adversarial approaches that can also be combined with self-training.
Source: §7 · p. 8The study does not test different weights or strengths for perturbation functions or stochastic application of those functions.
Source: §5 · p. 7
Abstract
from arXiv · showhide
Unsupervised domain adaptation is a promising technique for semantic segmentation and other computer vision tasks for which large-scale data annotation is costly and time-consuming. In semantic segmentation, it is attractive to train models on annotated images from a simulated (source) domain and deploy them on real (target) domains. In this work, we present a novel framework for unsupervised domain adaptation based on the notion of target-domain consistency training. Intuitively, our work is based on the idea that in order to perform well on the target domain, a model's output should be consistent with respect to small perturbations of inputs in the target domain. Specifically, we introduce a new loss term to enforce pixelwise consistency between the model's predictions on a target image and a perturbed version of the same image. In comparison to popular adversarial adaptation methods, our approach is simpler, easier to implement, and more memory-efficient during training. Experiments and extensive ablation studies demonstrate that our simple approach achieves remarkably strong results on two challenging synthetic-to-real benchmarks, GTA5-to-Cityscapes and SYNTHIA-to-Cityscapes. Code is available at: https://github.com/lukemelas/pixmatch
1. Introduction
PixelMatch addresses the cost of pixel-level annotation by adapting from labeled simulated images to unlabeled real images. It enforces target-domain consistency, and its augmentation baseline performs strongly on two benchmarks while combining with self-training to reach a new GTA5-to-Cityscapes state of the art.
1. Introduction
- Pixel-level annotation is laborious and expensive; Cityscapes urban-scene annotation takes an estimated 90 minutes per image.The introduction motivates using labeled simulated-source images and unlabeled real-target images for semantic segmentation.Source: §1 · p. 1
- PixMatch adds a target-domain consistency loss that encourages matching predictions for an image and a perturbed version of that image.The motivating premise is that smooth predictions under small target-image perturbations are likely to indicate better adaptation.Source: §1 · p. 1
- The augmentation-based baseline achieves extremely strong results on GTA5-to-Cityscapes and SYNTHIA-to-Cityscapes, outperforming complex prior methods that combine source, adversarial, and self-training losses.The paper also reports a new state of the art on GTA5-to-Cityscapes when PixMatch is combined with self-training.Source: §1 · p. 2
- PixMatch is described as easier to implement, more stable, and less memory-intensive than adversarial approaches, and it introduces only one hyperparameter.That hyperparameter weights the source supervised loss against the target consistency loss.Source: §1 · p. 2
- Among the tested perturbation functions, the simple baseline using heavy data augmentation performs best.The paper investigates data augmentation, CutMix, style transfer, and Fourier-transform perturbations.Source: §1 · p. 2
- The authors report that their baseline ablations show domain adaptation differs markedly from semi-supervised image classification.They also report that combining PixMatch with self-training can further improve performance.Source: §1 · p. 2
2. Background
The background frames semantic segmentation UDA as learning from labeled source images and unlabeled target images, often across simulated-to-real domains. It reviews adversarial and self-training methods before distinguishing PixMatch's direct target consistency objective from prior consistency approaches and semi-supervised learning.
2. Background
- UDA for semantic segmentation uses labeled source data and unlabeled target data, commonly adapting from simulated images to real ones.The paper gives GTA V-to-Cityscapes and SYNTHIA-to-Cityscapes as synthetic-to-real examples.Source: §2 · p. 2
2.1.1 Unsupervised Domain Adaptation for Semantic Segmentation
- Adversarial methods align source and target feature distributions but are computationally expensive, unstable, and hyperparameter-sensitive; self-training instead uses target pseudolabels or confidence-oriented losses.The reviewed self-training examples include class-balanced pseudolabeling, entropy minimization, and a linear alternative to entropy minimization.Source: §2.1.1 · p. 2
- Unlike prior cycle-consistency and other consistency approaches, PixMatch directly applies a general non-adversarial consistency loss to the segmentation network on target-domain perturbations.It uses a single segmentation network alongside the standard source cross-entropy loss and can be combined with prior approaches.Source: §2.1.1 · pp. 2–3
2.1.2 Consistency Training
- Modern consistency training creates multiple augmented views of an input for one model, but PixMatch targets UDA rather than semi-supervised image classification.The paper asks which forms of consistency loss are effective specifically for UDA.Source: §2.1.2 · p. 3
2.1.3 Contrastive Learning and Data Augmentation
- PixMatch's baseline uses SimCLR's proposed data augmentations as its perturbation function for consistency training.Source: §2.1.3 · p. 3
3. PixMatch
PixMatch combines supervised source-domain training with target-domain consistency training, comparing predictions on target images and perturbed versions. Its perturbation functions connect the framework to both self-supervised learning and domain adaptation, while its relationship to self-training depends partly on the perturbation choice.
3.1. Overview
- PixMatch combines supervised source cross-entropy with a target consistency loss based on pseudolabels from target images and perturbed image–pseudolabel pairs.The final loss weights the target consistency loss relative to the source loss with λ_T.Source: §3.1 · p. 4
3.2. Perturbation Functions
- Four perturbation functions cover augmentation, CutMix, style transfer, and Fourier-based target-image modification.Geometric augmentation is also applied to the pseudolabel; CutMix mixes source regions and labels, style transfer applies source style to target images, and Fourier consistency replaces target low frequencies with source frequencies.Source: §3.2 · p. 4
3.3. Relationship between PixMatch and Self-Training
- With identity perturbation, PixMatch pseudolabels become standard self-training pseudolabels; the baseline’s hard approach performs better empirically than its soft variant.Unlike pure self-training, PixMatch can incorporate source information into perturbations, as in its use of source-domain Fourier frequencies.Source: §3.3 · p. 4
4. Experiments
Experiments evaluate PixMatch on GTA5-to-Cityscapes and SYNTHIA-to-Cityscapes, comparing it with 27 other methods. The paper reports strong relative performance and shows qualitative segmentation gains over prior techniques.
Datasets
- Evaluation uses GTA5-to-Cityscapes and SYNTHIA-to-Cityscapes, measuring per-class and mean IoU across 19 GTA5 classes and 16 or 13 SYNTHIA classes.SYNTHIA shares only 16 classes with Cityscapes.Source: §4 · pp. 4–5
Model
- Experiments use DeepLab-v2 with a ResNet-101 backbone and run on a single NVIDIA RTX 2080TI GPU with 11GB of VRAM.Training uses SGD with learning rate 1.0·10−4, momentum 0.9, weight decay 5·10−4, and polynomial learning-rate decay.Source: §4 · p. 5
Results
- PixMatch performs extremely well relative to 27 other methods, and its augmentation-based baseline visually outsegments prior techniques in the qualitative examples.On the third example, it alone properly segments the bicycle and avoids mistaking much of the building for sky.Source: §4 · p. 5, Figure 2 · p. 5
5. Analysis
Across GTA5-to-Cityscapes and SYNTHIA-to-Cityscapes, PixMatch’s augmentation-based consistency model is highly competitive, and combining it with another self-training loss can improve results. The analysis also reports differences among perturbations and notes untested ways of combining them.
Fourier
Overall
- 50.3 mIoU on GTA5-to-Cityscapes with max-square loss outperforms all but a multiple-model distillation method at 50.4; on SYNTHIA, CutMix leads and all but style are within 0.4 mIoU.The augmentation-based model ranks first among the tested perturbations on GTA5-to-Cityscapes; on SYNTHIA-to-Cityscapes, CutMix ranks first.Source: §5 · p. 5, Table 1 · p. 6, Table 2 · p. 7
CutMix
- The CutMix model uses no domain-adaptation-specific components and substantially outperforms pure self-training methods, including MinEnt [52], MaxSquare [7], CRST [62], and PyCDA [34].The cited examples of domain-adaptation-specific components include adversarial training [53], domain-specific priors [26], style transfer [29], and distribution alignment [30].Source: §5 · p. 6
Style
- The style-based variant performs poorly relative to PixMatch’s other models and direct style-transfer approaches; the authors hypothesize this may reflect using AvatarNet rather than CycleGAN or CycleGAN + StyleSwap.Source: §5 · pp. 6–7
Combining Perturbation Functions
- Combining two perturbation functions often improves results, whereas combining three or more introduces too much perturbation.The paper gives Fourier and CutMix together on SYNTHIA as an example of a two-function combination.Source: §5 · p. 7
Limitations
- The study did not test different strengths or weights per perturbation function, or stochastic application of perturbations.The authors identify studying interactions among perturbation functions as a possible future direction.Source: §5 · p. 7
6. Ablation Studies and Further Experiments
The ablations find strong results at moderate consistency-loss weights and with no pseudolabel threshold. They also show that PixMatch can be combined with max-square loss, with the combined variant reaching state-of-the-art performance on GTA5-to-Cityscapes.
Ablation: Varying λT
- λT = 0.10 and 0.15 both yield strong results, contrasting with semi-supervised image classification, where consistency loss weights are usually greater than 1.λT trades off the source supervised loss and target consistency loss; the tested values were 0.00, 0.05, 0.10, 0.15, and 0.20.Source: §6 · p. 8
Ablation: Adding a Pseudolabel Threshold
- τ = 0.0, with no pseudolabel threshold, performs best; the authors hypothesize larger thresholds may make the loss dominated by easy classes.A threshold applies consistency loss only to pixels whose target output probability exceeds τ.Source: §6 · p. 8
Combining PixMatch with Other Methods
- The λMSL = 0.05 variant combining max-square loss and consistency training achieves state-of-the-art performance on GTA5-to-Cityscapes.Across-the-board performance improves for the tested max-square loss weightings.Source: §6 · p. 8
7. Conclusion
PixMatch uses target-domain consistency training for semantic segmentation, with extensive ablations examining its framework. The authors report strong results and practical advantages over adversarial approaches.
- PixMatch adapts semantic segmentation through target-domain consistency training, evaluated with extensive ablation studies.The ablations examine which aspects of the framework matter most to final performance.Source: §7 · p. 8
- Compared with adversarial approaches, PixMatch is reported to be faster, more stable, and to have superior performance.The authors also say it may be readily combined with other methods.Source: §7 · p. 8
- Future work may explore new perturbation functions, combinations of functions, and applying the framework to tasks such as object detection.The authors state that code and pretrained models will be released to facilitate future research.Source: §7 · p. 8
Appendix
The appendix specifies the perturbation functions used in PixMatch and presents additional function combinations alongside visual examples and a comparison with state of the art.
7.1. Perturbation Function Details
- The baseline uses SimCLR-based augmentations, while style, Fourier, and CutMix consistency use Avatar-Net, Fourier swapping, and CutMix functions, respectively.The style-transfer strength is 0.1, and the Fourier low-frequency window parameter β is 0.01.Source: §7.1 · p. 12
7.2. Additional Combinations of Perturbation Functions
- Figure 4 shows target-image augmentations; random cropping changes their size and aspect ratio, while style and Fourier images appear locally similar to the source domain despite global unnaturalness.The passage characterizes the similarity as local/low-frequency.Source: Figure 4 · p. 13
- Table 6 compares additional perturbation-function combinations with state of the art, and Figure 5 presents qualitative examples on SYNTHIA-to-Cityscapes.The supplied passages identify the comparison and examples but do not provide table values or qualitative outcomes.Source: §7.2 · p. 12, Figure 5 · p. 14, Table 6 · p. 15