Fisher meets Feynman: score-based variational inference with a product of experts
Diana Cai, Robert M. Gower, David M. Blei, Lawrence K. Saul
Abstract
We introduce a highly expressive yet distinctly tractable family for black-box variational inference (BBVI). Each member of this family is a weighted product of experts (PoE), and each weighted expert in the product is proportional to a multivariate -distribution. These products of experts can model distributions with skew, heavy tails, and multiple modes, but to use them for BBVI, we must be able to sample from their densities. We show how to do this by reformulating these products of experts as latent variable models with auxiliary Dirichlet random variables. These Dirichlet variables emerge from a Feynman identity, originally developed for loop integrals in quantum field theory, that expresses the product of multiple fractions (or in our case, -distributions) as an integral over the simplex. We leverage this simplicial latent space to draw weighted samples from these products of experts -- samples which BBVI then uses to find the PoE that best approximates a target density. Given a collection of experts, we derive an iterative procedure to optimize the exponents that determine their geometric weighting in the PoE. At each iteration, this procedure minimizes a regularized Fisher divergence to match the scores of the variational and target densities at a batch of samples drawn from the current approximation. This minimization reduces to a convex quadratic program, and we prove under general conditions that these updates converge exponentially fast to a near-optimal weighting of experts. We conclude by evaluating this approach on a variety of synthetic and real-world target distributions.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on11
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Tails of Lipschitz Triangular FlowsPriyank Jaini, Ivan Kobyzev, Yaoliang Yu, Marcus A. BrubakerICML 2020 · 60 citations
- Provable convergence guarantees for black-box variational inferenceJustin Domke, Robert M. Gower, Guillaume GarrigosNeurIPS 2023 · 35 citations
- On the Convergence of Black-Box Variational InferenceKyurae Kim, Jisu Oh, Kaiwen Wu, Yi-An Ma et al.NeurIPS 2023 · 27 citations
- Variational Inference with Gaussian Score MatchingChirag Modi, Robert M. Gower, Charles Margossian, Yuling Yao et al.NeurIPS 2023 · 24 citations
Related papers
- Challenges and Opportunities in High Dimensional Variational InferenceAkash Kumar Dhaka, Alejandro Catalina, Manushi Welandawe, Michael Riis Andersen et al.NeurIPS 2021 · 54 citations
- EigenVI: score-based variational inference with orthogonal function expansionsDiana Cai, Chirag Modi, Charles Margossian, Robert M. Gower et al.NeurIPS 2024 · 17 citations
- Batch and match: black-box variational inference with a score-based divergenceDiana Cai, Chirag Modi, Loucas Pillaud-Vivien, Charles Margossian et al.ICML 2024 · 18 citations
- The Quotient Bayesian Learning RuleMykola Lukashchuk, Raphaël Trésor, Wouter W. L. Nuijten, Ismail Senöz et al.NeurIPS 2025 · 1 citation
- Phase-Type Variational Autoencoders for Heavy-Tailed DataAbdelhakim Ziani, Andras Horvath, Paolo BallariniICML 2026
