Coin Sampling: Gradient-Based Bayesian Inference without Learning Rates
Louis Sharrock, Christopher Nemeth
Abstract
In recent years, particle-based variational inference (ParVI) methods such as Stein variational gradient descent (SVGD) have grown in popularity as scalable methods for Bayesian inference. Unfortunately, the properties of such methods invariably depend on hyperparameters such as the learning rate, which must be carefully tuned by the practitioner in order to ensure convergence to the target measure at a suitable rate. In this paper, we introduce a suite of new particle-based methods for scalable Bayesian inference based on coin betting, which are entirely learning-rate free. We illustrate the performance of our approach on a range of numerical examples, including several high-dimensional models and datasets, demonstrating comparable performance to other ParVI algorithms with no need to tune a learning rate.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 393f9992-96ec-4b37-adc2-2edcd1d171dcCited by top-tier papers3
- Entropy-based Training Methods for Scalable Neural Implicit SamplersWeijian Luo, Boya Zhang, Zhihua ZhangNeurIPS 2023 · 15 citations
- Momentum Particle Maximum LikelihoodJen Ning Lim, Juan Kuntz, Samuel Power, Adam M. JohansenICML 2024 · 13 citations
- Learning Rate Free Bayesian Inference in Constrained DomainsLouis Sharrock, Lester Mackey, Christopher NemethNeurIPS 2023 · 3 citations
Builds on10
- A Non-Asymptotic Analysis for Stein Variational Gradient DescentAnna Korba, Adil Salim, Michael Arbel, Giulia Luise et al.NeurIPS 2020 · 102 citations
- SVGD as a kernelized Wasserstein gradient flow of the chi-squared divergenceSinho Chewi, Thibaut Le Gouic, Chen Lu, Tyler Maunu et al.NeurIPS 2020 · 92 citations
- Projected Stein Variational Gradient DescentPeng Chen, Omar GhattasNeurIPS 2020 · 84 citations
- The Wasserstein Proximal Gradient AlgorithmAdil Salim, Anna Korba, Giulia LuiseNeurIPS 2020 · 74 citations
- Kernel Stein Discrepancy DescentAnna Korba, Pierre-Cyril Aubin-Frankowski, Szymon Majewski, Pierre AblinICML 2021 · 64 citations
Related papers
- Better Parameter-Free Stochastic Optimization with ODE Updates for Coin-BettingKeyi Chen, John Langford, Francesco OrabonaAAAI 2022 · 23 citations
- A Finite-Particle Convergence Rate for Stein Variational Gradient DescentJiaxin Shi, Lester MackeyNeurIPS 2023 · 34 citations
- Variance Reduction and Quasi-Newton for Particle-Based Variational InferenceMichael Zhu, Chang Liu, Jun ZhuICML 2020 · 12 citations
- Provably Fast Finite Particle Variants of SVGD via Virtual Particle Stochastic ApproximationAniket Das, Dheeraj NagarajNeurIPS 2023 · 10 citations
- ELBOing Stein: Variational Bayes with Stein Mixture InferenceOla Rønning, Eric T. Nalisnick, Christophe Ley, Padhraic Smyth et al.ICLR 2025
