arXiv · 1611.07800
Infinite Variational Autoencoder for Semi-Supervised Learning
Abstract
This paper presents an infinite variational autoencoder (VAE) whose capacity adapts to suit the input data. This is achieved using a mixture model where the mixing coefficients are modeled by a Dirichlet process, allowing us to integrate over the coefficients when performing inference. Critically, this then allows us to automatically vary the number of autoencoders in the mixture based on the data. Experiments show the flexibility of our method, particularly for semi-supervised learning, where only a small number of training samples are available.
Explore related subjects
Keep this discovery
Ehsan Abbasnejad, Anthony Dick, Anton van den Hengel. 2016-11-23. Infinite Variational Autoencoder for Semi-Supervised Learning. https://arxiv.org/abs/1611.07800
Cite the original work for its findings. Save a collection to share your selection of sources.