Provable Smoothness Guarantees for Black-Box Variational Inference
Justin Domke
Abstract
Black-box variational inference tries to approximate a complex target distribution though a gradient-based optimization of the parameters of a simpler distribution. Provable convergence guarantees require structural properties of the objective. This paper shows that for location-scale family approximations, if the target is M-Lipschitz smooth, then so is the objective, if the entropy is excluded. The key proof idea is to describe gradients in a certain inner-product space, thus permitting use of Bessel's inequality. This result gives insight into how to parameterize distributions, gives bounds the location of the optimal parameters, and is a key ingredient for convergence guarantees.
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.
Cited by top-tier papers23
- Variational inference via Wasserstein gradient flowsMarc Lambert, Sinho Chewi, Francis R. Bach, Silvère Bonnabel et al.NeurIPS 2022 · 123 citations
- Forward-Backward Gaussian Variational Inference via JKO in the Bures-Wasserstein SpaceMichael Ziyang Diao, Krishna Balasubramanian, Sinho Chewi, Adil SalimICML 2023 · 47 citations
- Provable convergence guarantees for black-box variational inferenceJustin Domke, Robert M. Gower, Guillaume GarrigosNeurIPS 2023 · 35 citations
- Amortized Variational Inference for Simple Hierarchical ModelsAbhinav Agrawal, Justin DomkeNeurIPS 2021 · 32 citations
- On the Convergence of Black-Box Variational InferenceKyurae Kim, Jisu Oh, Kaiwen Wu, Yi-An Ma et al.NeurIPS 2023 · 27 citations
Related papers
- Nearly Dimension-Independent Convergence of Mean-Field Black-Box Variational InferenceKyurae Kim, Yian Ma, Trevor Campbell, Jacob R. GardnerNeurIPS 2025 · 1 citation
- Practical and Matching Gradient Variance Bounds for Black-Box Variational Bayesian InferenceKyurae Kim, Kaiwen Wu, Jisu Oh, Jacob R. GardnerICML 2023 · 8 citations
- Stochastic Gradient Variational Inference with Price's Gradient Estimator from Bures-Wasserstein to Parameter SpaceKyurae Kim, Qiang Fu, Yian Ma, Jacob Gardner et al.ICML 2026
- 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
- Understanding Stochastic Natural Gradient Variational InferenceKaiwen Wu, Jacob R. GardnerICML 2024 · 11 citations
