English

Generalization by Adaptation: Diffusion-Based Domain Extension for Domain-Generalized Semantic Segmentation

Computer Vision and Pattern Recognition 2023-12-05 v1 Machine Learning

Abstract

When models, e.g., for semantic segmentation, are applied to images that are vastly different from training data, the performance will drop significantly. Domain adaptation methods try to overcome this issue, but need samples from the target domain. However, this might not always be feasible for various reasons and therefore domain generalization methods are useful as they do not require any target data. We present a new diffusion-based domain extension (DIDEX) method and employ a diffusion model to generate a pseudo-target domain with diverse text prompts. In contrast to existing methods, this allows to control the style and content of the generated images and to introduce a high diversity. In a second step, we train a generalizing model by adapting towards this pseudo-target domain. We outperform previous approaches by a large margin across various datasets and architectures without using any real data. For the generalization from GTA5, we improve state-of-the-art mIoU performance by 3.8% absolute on average and for SYNTHIA by 11.8% absolute, marking a big step for the generalization performance on these benchmarks. Code is available at https://github.com/JNiemeijer/DIDEX

Keywords

Cite

@article{arxiv.2312.01850,
  title  = {Generalization by Adaptation: Diffusion-Based Domain Extension for Domain-Generalized Semantic Segmentation},
  author = {Joshua Niemeijer and Manuel Schwonberg and Jan-Aike Termöhlen and Nico M. Schmidt and Tim Fingscheidt},
  journal= {arXiv preprint arXiv:2312.01850},
  year   = {2023}
}

Comments

Accepted to WACV 2024

R2 v1 2026-06-28T13:40:17.184Z