Trust-Region Variational Inference with Gaussian Mixture Models
arXiv:1907.04710
Abstract
Many methods for machine learning rely on approximate inference from intractable probability distributions. Variational inference approximates such distributions by tractable models that can be subsequently used for approximate inference. Learning sufficiently accurate approximations requires a rich model family and careful exploration of the relevant modes of the target distribution. We propose a method for learning accurate GMM approximations of intractable probability distributions based on insights from policy search by using information-geometric trust regions for principled exploration. For efficient improvement of the GMM approximation, we derive a lower bound on the corresponding optimization objective enabling us to update the components independently. Our use of the lower bound ensures convergence to a stationary point of the original objective. The number of components is adapted online by adding new components in promising regions and by deleting components with negligible weight. We demonstrate on several domains that we can learn approximations of complex, multimodal distributions with a quality that is unmet by previous variational inference methods, and that the GMM approximation can be used for drawing samples that are on par with samples created by state-of-the-art MCMC samplers while requiring up to three orders of magnitude less computational resources.
References in corpus (16)
- Auto-Encoding Variational Bayes
- Trust Region Policy Optimization
- NICE: Non-linear Independent Components Estimation
- Variational Inference with Normalizing Flows
- Evolution Strategies as a Scalable Alternative to Reinforcement Learning
- Density estimation using Real NVP
- MADE: Masked Autoencoder for Distribution Estimation
- Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithm
- Elliptical slice sampling
- Auxiliary Deep Generative Models
- Boosting Variational Inference
- Nonparametric variational inference
- Fast Second-Order Stochastic Backpropagation for Variational Inference
- Model-Free Trajectory-based Policy Optimization with Monotonic Improvement
- A trust-region method for stochastic variational inference with applications to streaming data
- Inference Trees: Adaptive Inference with Exploration