Variational inference via Wasserstein gradient flows - Inria - Institut national de recherche en sciences et technologies du numérique Accéder directement au contenu
Communication Dans Un Congrès Année : 2022

Variational inference via Wasserstein gradient flows

Inférence variationnelle via les flots de gradient Wasserstein

Résumé

Along with Markov chain Monte Carlo (mcmc) methods, variational inference (vi) has emerged as a central computational approach to large-scale Bayesian inference. Rather than sampling from the true posterior π, vi aims at producing a simple but effective approximation π to π for which summary statistics are easy to compute. However, unlike the well-studied mcmc methodology, algorithmic guarantees for vi are still relatively less well-understood. In this work, we propose principled methods for vi, in which π is taken to be a Gaussian or a mixture of Gaussians, which rest upon the theory of gradient flows on the Bures-Wasserstein space of Gaussian measures. Akin to mcmc, it comes with strong theoretical guarantees when π is log-concave.
Fichier principal
Vignette du fichier
Bures-JKO_v3.pdf (2.14 Mo) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03953336 , version 1 (24-01-2023)

Identifiants

Citer

Marc Lambert, Sinho Chewi, Francis S Bach, Silvère Bonnabel, Philippe Rigollet. Variational inference via Wasserstein gradient flows. NeurIPS 2022 - Thirty-sixth Conference on Neural Information Processing Systems, Nov 2022, Nouvelle Orléans (Louisiane), United States. ⟨10.48550/arXiv.2205.15902⟩. ⟨hal-03953336⟩
76 Consultations
181 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More