References & Citations
Statistics > Machine Learning
Title: Stick-Breaking Variational Autoencoders
(Submitted on 20 May 2016 (v1), last revised 3 Apr 2017 (this version, v3))
Abstract: We extend Stochastic Gradient Variational Bayes to perform posterior inference for the weights of Stick-Breaking processes. This development allows us to define a Stick-Breaking Variational Autoencoder (SB-VAE), a Bayesian nonparametric version of the variational autoencoder that has a latent representation with stochastic dimensionality. We experimentally demonstrate that the SB-VAE, and a semi-supervised variant, learn highly discriminative latent representations that often outperform the Gaussian VAE's.
Submission history
From: Eric Nalisnick [view email][v1] Fri, 20 May 2016 02:20:13 GMT (2437kb,D)
[v2] Tue, 6 Dec 2016 03:16:25 GMT (2578kb,D)
[v3] Mon, 3 Apr 2017 22:36:33 GMT (2577kb,D)
Link back to: arXiv, form interface, contact.