NeurIPS 2017poster76 citations
VAE Learning via Stein Variational Gradient Descent
Yuchen Pu, Zhe Gan, Ricardo Henao, Chunyuan Li, Shaobo Han, Lawrence Carin
Abstract
A new method for learning variational autoencoders (VAEs) is developed, based on Stein variational gradient descent. A key advantage of this approach is that one need not make parametric assumptions about the form of the encoder distribution. Performance is further enhanced by integrating the proposed encoder with importance sampling. Excellent performance is demonstrated across multiple unsupervised and semi-supervised problems, including semi-supervised analysis of the ImageNet data, demonstrating the scalability of the model to large datasets.
BibTeX
@inproceedings{NIPS2017_443dec30,
author = {Pu, Yuchen and Gan, Zhe and Henao, Ricardo and Li, Chunyuan and Han, Shaobo and Carin, Lawrence},
booktitle = {Advances in Neural Information Processing Systems},
editor = {I. Guyon and U. Von Luxburg and S. Bengio and H. Wallach and R. Fergus and S. Vishwanathan and R. Garnett},
pages = {},
publisher = {Curran Associates, Inc.},
title = {VAE Learning via Stein Variational Gradient Descent},
url = {https://proceedings.neurips.cc/paper_files/paper/2017/file/443dec3062d0286986e21dc0631734c9-Paper.pdf},
volume = {30},
year = {2017}
}