Learning latent subspaces in variational autoencoders

J Klys, J Snell, R Zemel - Advances in neural information …, 2018 - proceedings.neurips.cc
J Klys, J Snell, R Zemel
Advances in neural information processing systems, 2018proceedings.neurips.cc
Variational autoencoders (VAEs) are widely used deep generative models capable of
learning unsupervised latent representations of data. Such representations are often difficult
to interpret or control. We consider the problem of unsupervised learning of features
correlated to specific labels in a dataset. We propose a VAE-based generative model which
we show is capable of extracting features correlated to binary labels in the data and
structuring it in a latent subspace which is easy to interpret. Our model, the Conditional …
Abstract
Variational autoencoders (VAEs) are widely used deep generative models capable of learning unsupervised latent representations of data. Such representations are often difficult to interpret or control. We consider the problem of unsupervised learning of features correlated to specific labels in a dataset. We propose a VAE-based generative model which we show is capable of extracting features correlated to binary labels in the data and structuring it in a latent subspace which is easy to interpret. Our model, the Conditional Subspace VAE (CSVAE), uses mutual information minimization to learn a low-dimensional latent subspace associated with each label that can easily be inspected and independently manipulated. We demonstrate the utility of the learned representations for attribute manipulation tasks on both the Toronto Face and CelebA datasets.
proceedings.neurips.cc
Showing the best result for this search. See all results