English

Adversarial Bayesian Augmentation for Single-Source Domain Generalization

Computer Vision and Pattern Recognition 2023-10-04 v2

Abstract

Generalizing to unseen image domains is a challenging problem primarily due to the lack of diverse training data, inaccessible target data, and the large domain shift that may exist in many real-world settings. As such data augmentation is a critical component of domain generalization methods that seek to address this problem. We present Adversarial Bayesian Augmentation (ABA), a novel algorithm that learns to generate image augmentations in the challenging single-source domain generalization setting. ABA draws on the strengths of adversarial learning and Bayesian neural networks to guide the generation of diverse data augmentations -- these synthesized image domains aid the classifier in generalizing to unseen domains. We demonstrate the strength of ABA on several types of domain shift including style shift, subpopulation shift, and shift in the medical imaging setting. ABA outperforms all previous state-of-the-art methods, including pre-specified augmentations, pixel-based and convolutional-based augmentations.

Keywords

Cite

@article{arxiv.2307.09520,
  title  = {Adversarial Bayesian Augmentation for Single-Source Domain Generalization},
  author = {Sheng Cheng and Tejas Gokhale and Yezhou Yang},
  journal= {arXiv preprint arXiv:2307.09520},
  year   = {2023}
}

Comments

Accepted to ICCV 2023

R2 v1 2026-06-28T11:33:56.751Z