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.

A unifying view for the mixture model of sparse Gaussian processes

Pseudo Independent Conditional Approximation for Training the Mixtures of Gaussian Processes

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

Multiview Learning with Variational Mixtures of Gaussian Processes

An Improved Mixture Model of Gaussian Processes and Its Classification Expectation–Maximization Algorithm

Multiview Variational Sparse Gaussian Processes.

The Hard-Cut EM Algorithm for Mixture of Sparse Gaussian Processes

Nonparametric Bayesian Mixed-effect Model: a Sparse Gaussian Process Approach

Joint Multi-Gaussian Mixture Model and Its Application to Multi-Model Multi-Bernoulli Filter

Bayesian mixtures of common factor analyzers: Model, variational inference, and applications

Variational Inference and Sparsity in High-Dimensional Deep Gaussian Mixture Models

Variational Mixtures of Gaussian Processes for Classification.

Mixture of Gaussian Processes and its Applications

Sparse Multimodal Gaussian Processes

An Effective Model Selection Criterion for Mixtures of Gaussian Processes.

Extended Variational Inference for Gamma Mixture Model in Positive Vectors Modeling

Mixture of robust Gaussian processes and its hard-cut EM algorithm with variational bounding approximation

An Mcmc Based Em Algorithm For Mixtures Of Gaussian Processes

Probabilistic Inference from Arbitrary Uncertainty using Mixtures of Factorized Generalized Gaussians

MAP approximation to the variational Bayes Gaussian mixture model and application

Gaussian mixture model with an extended ultrametric covariance structure