Learning Rate Free Bayesian Inference in Constrained Domains
Louis Sharrock, Lester Mackey, Christopher Nemeth
Abstract
We introduce a suite of new particle-based algorithms for sampling in constrained domains which are entirely learning rate free. Our approach leverages coin betting ideas from convex optimisation, and the viewpoint of constrained sampling as a mirrored optimisation problem on the space of probability measures. Based on this viewpoint, we also introduce a unifying framework for several existing constrained sampling algorithms, including mirrored Langevin dynamics and mirrored Stein variational gradient descent. We demonstrate the performance of our algorithms on a range of numerical examples, including sampling from targets on the simplex, sampling with fairness constraints, and constrained sampling problems in postselection inference. Our results indicate that our algorithms achieve competitive performance with existing constrained sampling methods, without the need to tune any hyperparameters.
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 541f8212-9645-49c2-a79f-2f8458b42619Cited by top-tier papers2
- Mirror and Preconditioned Gradient Descent in Wasserstein SpaceClément Bonet, Théo Uscidda, Adam David, Pierre-Cyril Aubin-Frankowski et al.NeurIPS 2024 · 19 citations
- Constrained Sampling with Primal-Dual Langevin Monte CarloLuiz F. O. Chamon, Mohammad Reza Karimi Jaghargh, Anna KorbaNeurIPS 2024 · 15 citations
Builds on16
- Minimax Pareto Fairness: A Multi Objective PerspectiveNatalia Martínez, Martín Bertrán, Guillermo SapiroICML 2020 · 232 citations
- 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
- Efficient constrained sampling via the mirror-Langevin algorithmKwangjun Ahn, Sinho ChewiNeurIPS 2021 · 77 citations
Related papers
- Sampling with Mirrored Stein OperatorsJiaxin Shi, Chang Liu, Lester MackeyICLR 2022 · 23 citations
- Coin Sampling: Gradient-Based Bayesian Inference without Learning RatesLouis Sharrock, Christopher NemethICML 2023 · 10 citations
- Sampling with Trusthworthy Constraints: A Variational Gradient FrameworkXingchao Liu, Xin Tong, Qiang LiuNeurIPS 2021 · 2 citations
- Mirror Langevin Monte Carlo: the Case Under IsoperimetryQijia JiangNeurIPS 2021 · 28 citations
- Fast Non-Log-Concave Sampling under Nonconvex Equality and Inequality Constraints with LandingKijung Jeon, Michael Muehlebach, Molei TaoNeurIPS 2025 · 5 citations
