arXiv · 1902.01950
Meta-Amortized Variational Inference and Learning
Abstract
Despite the recent success in probabilistic modeling and their applications, generative models trained using traditional inference techniques struggle to adapt to new distributions, even when the target distribution may be closely related to the ones seen during training. In this work, we present a doubly-amortized variational inference procedure as a way to address this challenge. By sharing computation across not only a set of query inputs, but also a set of different, related probabilistic models, we learn transferable latent representations that generalize across several related distributions. In particular, given a set of distributions over images, we find the learned representations to transfer to different data transformations. We empirically demonstrate the effectiveness of our method by introducing the MetaVAE, and show that it significantly outperforms baselines on downstream image classification tasks on MNIST (10-50%) and NORB (10-35%).
Explore related subjects
Keep this discovery
Mike Wu, Kristy Choi, Noah Goodman, Stefano Ermon. 2019-02-05. Meta-Amortized Variational Inference and Learning. https://arxiv.org/abs/1902.01950
Cite the original work for its findings. Save a collection to share your selection of sources.