Rényi-infinity constrained sampling with d3 membership queries
Yunbum Kook, Matthew S. Zhang
Abstract
Uniform sampling over a convex body is a fundamental algorithmic problem, yet the convergence in KL or Rényi divergence of most samplers remains poorly understood. In this work, we propose a constrained proximal sampler, a principled and simple algorithm that possesses elegant convergence guarantees. Leveraging the uniform ergodicity of this sampler, we show that it converges in the Rényi-infinity divergence (R ∞ ) with no query complexity overhead when starting from a warm start. This is the strongest of commonly considered performance metrics, implying rates in R q , KL convergence as special cases.
By applying this sampler within an annealing scheme, we propose an algorithm which can approximately sample ε-close to the uniform distribution on convex bodies in R ∞ -divergence with O(d 3 polylog 1 ε ) query complexity. This improves on all prior results in R q , KL-divergences, without resorting to any algorithmic modifications or post-processing of the sample. It also matches the prior best known complexity in total variation distance.
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 67413efe-fa95-4b1f-8f4c-adcfa9143803Cited by top-tier papers5
- In-and-Out: Algorithmic Diffusion for Sampling Convex BodiesYunbum Kook, Santosh S. Vempala, Matthew Shunshi ZhangNeurIPS 2024 · 25 citations
- Faster Logconcave Sampling from a Cold Start in High DimensionYunbum Kook, Santosh S. VempalaFOCS 2025 · 11 citations
- Covariance estimation using Markov chain Monte CarloYunbum Kook, Shunshi ZhangICML 2026 · 6 citations
- Riemannian Proximal Sampler for High-accuracy Sampling on ManifoldsYunrui Guan, Krishnakumar Balasubramanian, Shiqian MaNeurIPS 2025 · 4 citations
- Sampling and Integration of Logconcave Functions by Algorithmic DiffusionYunbum Kook, Santosh S. VempalaSTOC 2025 · 2 citations
Builds on8
- Efficient constrained sampling via the mirror-Langevin algorithmKwangjun Ahn, Sinho ChewiNeurIPS 2021 · 77 citations
- Sampling with Riemannian Hamiltonian Monte Carlo in a Constrained SpaceYunbum Kook, Yin Tat Lee, Ruoqi Shen, Santosh S. VempalaNeurIPS 2022 · 53 citations
- Faster Differentially Private Samplers via Rényi Divergence Analysis of Discretized Langevin MCMCArun Ganesh, Kunal TalwarNeurIPS 2020 · 44 citations
- Mirror Langevin Monte Carlo: the Case Under IsoperimetryQijia JiangNeurIPS 2021 · 28 citations
- In-and-Out: Algorithmic Diffusion for Sampling Convex BodiesYunbum Kook, Santosh S. Vempala, Matthew Shunshi ZhangNeurIPS 2024 · 25 citations
Related papers
- Faster high-accuracy log-concave sampling via algorithmic warm startsJason M. Altschuler, Sinho ChewiFOCS 2023 · 6 citations
- Provable Benefit of Annealed Langevin Monte Carlo for Non-log-concave SamplingWei Guo, Molei Tao, Yongxin ChenICLR 2025
- Double-Loop Unadjusted Langevin AlgorithmPaul Rolland, Armin Eftekhari, Ali Kavis, Volkan CevherICML 2020 · 3 citations
- Primal Dual Interpretation of the Proximal Stochastic Gradient Langevin AlgorithmAdil Salim, Peter RichtárikNeurIPS 2020 · 53 citations
- Sampling from Convex Sets with a Cold Start using Multiscale DecompositionsHariharan Narayanan, Amit Rajaraman, Piyush SrivastavaSTOC 2023 · 2 citations
