[edit]
Hyperspherical Variational Auto-Encoders
Proceedings of the 34th Conference on Uncertainty in Artificial Intelligence, PMLR R16:855-864, 2018.
Abstract
The Variational Auto-Encoder (VAE) is one of the most used unsupervised machine learn- ing models. But although the default choice of a Gaussian distribution for both the prior and posterior represents a mathematically con- venient distribution often leading to competi- tive results, we show that this parameterization fails to model data with a latent hyperspheri- cal structure. To address this issue we propose using a von Mises-Fisher (vMF) distribution in- stead, leading to a hyperspherical latent space. Through a series of experiments we show how such a hyperspherical VAE, or S-VAE, is more suitable for capturing data with a hyperspheri- cal latent structure, while outperforming a nor- mal, N-VAE, in low dimensions on other data types.