Embedding-reparameterization procedure for manifold-valued latent variables in generative models

Eugene Golikov, Maksim Kretov

Conventional prior for Variational Auto-Encoder (VAE) is a Gaussian distribution. Recent works demonstrated that choice of prior distribution affects learning capacity of VAE model…