English

Uncertainty-aware Generalized Adaptive CycleGAN

Computer Vision and Pattern Recognition 2021-02-24 v1 Machine Learning Image and Video Processing

Abstract

Unpaired image-to-image translation refers to learning inter-image-domain mapping in an unsupervised manner. Existing methods often learn deterministic mappings without explicitly modelling the robustness to outliers or predictive uncertainty, leading to performance degradation when encountering unseen out-of-distribution (OOD) patterns at test time. To address this limitation, we propose a novel probabilistic method called Uncertainty-aware Generalized Adaptive Cycle Consistency (UGAC), which models the per-pixel residual by generalized Gaussian distribution, capable of modelling heavy-tailed distributions. We compare our model with a wide variety of state-of-the-art methods on two challenging tasks: unpaired image denoising in the natural image and unpaired modality prorogation in medical image domains. Experimental results demonstrate that our model offers superior image generation quality compared to recent methods in terms of quantitative metrics such as signal-to-noise ratio and structural similarity. Our model also exhibits stronger robustness towards OOD test data.

Keywords

Cite

@article{arxiv.2102.11747,
  title  = {Uncertainty-aware Generalized Adaptive CycleGAN},
  author = {Uddeshya Upadhyay and Yanbei Chen and Zeynep Akata},
  journal= {arXiv preprint arXiv:2102.11747},
  year   = {2021}
}

Comments

13 pages, 9 figures

R2 v1 2026-06-23T23:26:32.123Z