1804.00891

Total: 1

#1 Hyperspherical Variational Auto-Encoders [PDF8] [Copy] [Kimi3]

Authors: Tim R. Davidson ; Luca Falorsi ; Nicola De Cao ; Thomas Kipf ; Jakub M. Tomczak

The Variational Auto-Encoder (VAE) is one of the most used unsupervised machine learning models. But although the default choice of a Gaussian distribution for both the prior and posterior represents a mathematically convenient distribution often leading to competitive results, we show that this parameterization fails to model data with a latent hyperspherical structure. To address this issue we propose using a von Mises-Fisher (vMF) distribution instead, leading to a hyperspherical latent space. Through a series of experiments we show how such a hyperspherical VAE, or $\mathcal{S}$-VAE, is more suitable for capturing data with a hyperspherical latent structure, while outperforming a normal, $\mathcal{N}$-VAE, in low dimensions on other data types. Code at http://github.com/nicola-decao/s-vae-tf and https://github.com/nicola-decao/s-vae-pytorch

Subjects: Machine Learning ; Machine Learning

Publish: 2018-04-03 09:57:26 UTC