Style Mixing and Patchwise Prototypical Matching for One-Shot Unsupervised Domain Adaptive Semantic Segmentation

Authors

  • Xinyi Wu University of South Carolina
  • Zhenyao Wu University of South Carolina
  • Yuhang Lu University of South Carolina
  • Lili Ju University of South Carolina
  • Song Wang University of South Carolina

DOI:

https://doi.org/10.1609/aaai.v36i3.20177

Keywords:

Computer Vision (CV)

Abstract

In this paper, we tackle the problem of one-shot unsupervised domain adaptation (OSUDA) for semantic segmentation where the segmentors only see one unlabeled target image during training. In this case, traditional unsupervised domain adaptation models usually fail since they cannot adapt to the target domain with over-fitting to one (or few) target samples. To address this problem, existing OSUDA methods usually integrate a style-transfer module to perform domain randomization based on the unlabeled target sample, with which multiple domains around the target sample can be explored during training. However, such a style-transfer module relies on an additional set of images as style reference for pre-training and also increases the memory demand for domain adaptation. Here we propose a new OSUDA method that can effectively relieve such computational burden. Specifically, we integrate several style-mixing layers into the segmentor which play the role of style-transfer module to stylize the source images without introducing any learned parameters. Moreover, we propose a patchwise prototypical matching (PPM) method to weighted consider the importance of source pixels during the supervised training to relieve the negative adaptation. Experimental results show that our method achieves new state-of-the-art performance on two commonly used benchmarks for domain adaptive semantic segmentation under the one-shot setting and is more efficient than all comparison approaches.

Downloads

Published

2022-06-28

How to Cite

Wu, X., Wu, Z., Lu, Y., Ju, L., & Wang, S. (2022). Style Mixing and Patchwise Prototypical Matching for One-Shot Unsupervised Domain Adaptive Semantic Segmentation. Proceedings of the AAAI Conference on Artificial Intelligence, 36(3), 2740-2749. https://doi.org/10.1609/aaai.v36i3.20177

Issue

Section

AAAI Technical Track on Computer Vision III