Variational Inference with Mixtures of Isotropic Gaussians

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Petit-Talamon, Marguerite, Lambert, Marc, Korba, Anna
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908658455019520
author Petit-Talamon, Marguerite
Lambert, Marc
Korba, Anna
author_facet Petit-Talamon, Marguerite
Lambert, Marc
Korba, Anna
contents Variational inference (VI) is a popular approach in Bayesian inference, that looks for the best approximation of the posterior distribution within a parametric family, minimizing a loss that is typically the (reverse) Kullback-Leibler (KL) divergence. In this paper, we focus on the following parametric family: mixtures of isotropic Gaussians (i.e., with diagonal covariance matrices proportional to the identity) and uniform weights. We develop a variational framework and provide efficient algorithms suited for this family. In contrast with mixtures of Gaussian with generic covariance matrices, this choice presents a balance between accurate approximations of multimodal Bayesian posteriors, while being memory and computationally efficient. Our algorithms implement gradient descent on the location of the mixture components (the modes of the Gaussians), and either (an entropic) Mirror or Bures descent on their variance parameters. We illustrate the performance of our algorithms on numerical experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2506_13613
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Variational Inference with Mixtures of Isotropic Gaussians
Petit-Talamon, Marguerite
Lambert, Marc
Korba, Anna
Machine Learning
Variational inference (VI) is a popular approach in Bayesian inference, that looks for the best approximation of the posterior distribution within a parametric family, minimizing a loss that is typically the (reverse) Kullback-Leibler (KL) divergence. In this paper, we focus on the following parametric family: mixtures of isotropic Gaussians (i.e., with diagonal covariance matrices proportional to the identity) and uniform weights. We develop a variational framework and provide efficient algorithms suited for this family. In contrast with mixtures of Gaussian with generic covariance matrices, this choice presents a balance between accurate approximations of multimodal Bayesian posteriors, while being memory and computationally efficient. Our algorithms implement gradient descent on the location of the mixture components (the modes of the Gaussians), and either (an entropic) Mirror or Bures descent on their variance parameters. We illustrate the performance of our algorithms on numerical experiments.
title Variational Inference with Mixtures of Isotropic Gaussians
topic Machine Learning
url https://arxiv.org/abs/2506.13613