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.

Analytic natural gradient updates for Cholesky factor in Gaussian variational approximation

Second order stochastic gradient update for Cholesky factor in Gaussian variational approximation from Stein's Lemma

Natural Gradient Variational Bayes Without Fisher Matrix Analytic Calculation and Its Inversion

Natural Gradients in Practice: Non-Conjugate Variational Inference in Gaussian Process Models

Natural gradient hybrid variational inference with application to deep mixed models

Understanding Stochastic Natural Gradient Variational Inference

Fast and Simple Natural-Gradient Variational Inference with Mixture of Exponential-family Approximations

Manifold Optimization Assisted Gaussian Variational Approximation

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

Iterative Methods for Vecchia-Laplace Approximations for Latent Gaussian Process Models

A Sharp Analysis of Covariate Adjusted Precision Matrix Estimation Via Alternating Projected Gradient Descent

Relative Gradient Speeding Up Additive Updates for Nonnegative Matrix Factorization

A Solution to the Ill-Conditioning of Gradient-Enhanced Covariance Matrices for Gaussian Processes

SLANG: Fast Structured Covariance Approximations for Bayesian Deep Learning with Natural Gradient

Manifold Gaussian Variational Bayes on the Precision Matrix

Representing Additive Gaussian Processes by Sparse Matrices

Precision and Cholesky Factor Estimation for Gaussian Processes

A Sharp Algorithmic Analysis of Covariate Adjusted Precision Matrix Estimation with General Structural Priors

Sparse inversion for derivative of log determinant