Abstract:Variational inference with Gaussian mixture models (GMMs) enables learning of highly tractable yet multi-modal approximations of intractable target distributions with up to a few hundred dimensions. The two currently most effective methods for GMM-based variational inference, VIPS and iBayes-GMM, both employ independent natural gradient updates for the individual components and their weights. We show for the first time, that their derived updates are equivalent, although their practical implementations and theoretical guarantees differ. We identify several design choices that distinguish both approaches, namely with respect to sample selection, natural gradient estimation, stepsize adaptation, and whether trust regions are enforced or the number of components adapted. We argue that for both approaches, the quality of the learned approximations can heavily suffer from the respective design choices: By updating the individual components using samples from the mixture model, iBayes-GMM often fails to produce meaningful updates to low-weight components, and by using a zero-order method for estimating the natural gradient, VIPS scales badly to higher-dimensional problems. Furthermore, we show that information-geometric trust-regions (used by VIPS) are effective even when using first-order natural gradient estimates, and often outperform the improved Bayesian learning rule (iBLR) update used by iBayes-GMM. We systematically evaluate the effects of design choices and show that a hybrid approach significantly outperforms both prior works. Along with this work, we publish our highly modular and efficient implementation for natural gradient variational inference with Gaussian mixture models, which supports 432 different combinations of design choices, facilitates the reproduction of all our experiments, and may prove valuable for the practitioner.

Variational Inference for Gaussian Process Modulated Poisson Processes

Variational Inference for Gaussian Process with Panel Count Data

Variational inference for probabilistic Poisson PCA

Stein Variational Gaussian Processes

Amortized Variational Inference for Deep Gaussian Processes

Constrained Gaussian Process for Signal Integrity applications using Variational Inference

Variational Inference for Uncertainty on the Inputs of Gaussian Process Models

Variational Inference for Diffusion Processes.

Variational Gaussian Process Diffusion Processes

Distributed Variational Inference in Sparse Gaussian Process Regression and Latent Variable Models

Variational Gaussian Processes For Linear Inverse Problems

Variational Bayesian Inference for Nonlinear Hawkes Process with Gaussian Process Self-Effects

Sparse Orthogonal Variational Inference for Gaussian Processes

Variational methods for the Dirichlet process

A Unified Perspective on Natural Gradient Variational Inference with Gaussian Mixture Models

Variational Inference for Continuous-Time Switching Dynamical Systems

Inference in Deep Gaussian Processes using Stochastic Gradient Hamiltonian Monte Carlo

Variational inference for Dirichlet process mixtures

Flexible and Efficient Inference with Particles for the Variational Gaussian Approximation

Variational Inference for Infinite Mixtures of Gaussian Processes with Applications to Traffic Flow Prediction

Deep Gaussian Processes with Importance-Weighted Variational Inference