Affine Variational Autoencoders: An Efficient Approach for Improving Generalization and Robustness to Distribution Shift

Rene Bidart, Alexander Wong

In this study, we propose the Affine Variational Autoencoder (AVAE), a variant of Variational Autoencoder (VAE) designed to improve robustness by overcoming the inability of VAEs t…