Sampling with Riemannian Hamiltonian Monte Carlo in a Constrained Space
Yunbum Kook, Yin Tat Lee, Ruoqi Shen, Santosh S. Vempala
Abstract
We demonstrate for the first time that ill-conditioned, non-smooth, constrained distributions in very high dimension, upwards of 100,000, can be sampled efficiently . Our algorithm incorporates constraints into the Riemannian version of Hamiltonian Monte Carlo and maintains sparsity. This allows us to achieve a mixing rate independent of smoothness and condition numbers. On benchmark data sets in systems biology and linear programming, our algorithm outperforms existing packages by orders of magnitude. In particular, we achieve a 1,000-fold speed-up for sampling from the largest published human metabolic network (RECON3D). Our package has been incorporated into the COBRA toolbox.
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 d96da76e-ca48-45f9-9dfe-58fa8158d2d7Cited by top-tier papers13
- Metropolis Sampling for Constrained Diffusion ModelsNic Fishman, Leo Klarner, Emile Mathieu, Michael J. Hutchinson et al.NeurIPS 2023 · 37 citations
- In-and-Out: Algorithmic Diffusion for Sampling Convex BodiesYunbum Kook, Santosh S. Vempala, Matthew Shunshi ZhangNeurIPS 2024 · 25 citations
- Sampling in Constrained Domains with Orthogonal-Space Variational Gradient DescentRuqi Zhang, Qiang Liu, Xin T. TongNeurIPS 2022 · 23 citations
- Constrained Diffusers for Safe Planning and ControlJichen Zhang, Liqun Zhao, Antonis Papachristodoulou, Jack UmenbergerNeurIPS 2025 · 17 citations
- Constrained Sampling with Primal-Dual Langevin Monte CarloLuiz F. O. Chamon, Mohammad Reza Karimi Jaghargh, Anna KorbaNeurIPS 2024 · 15 citations
Builds on2
Related papers
- Reflective Hamiltonian Monte Carlo: Mixing Analysis and Application to Sampling on Stiefel ManifoldKwangmin Lee, Yeonhee Park, Sewon ParkICML 2026
- A Gradient Based Strategy for Hamiltonian Monte Carlo Hyperparameter OptimizationAndrew Campbell, Wenlong Chen, Vincent Stimper, José Miguel Hernández-Lobato et al.ICML 2021 · 20 citations
- Unbiased constrained sampling with Self-Concordant Barrier Hamiltonian Monte CarloMaxence Noble, Valentin De Bortoli, Alain DurmusNeurIPS 2023 · 11 citations
- Practical Hamiltonian Monte Carlo on Riemannian Manifolds via Relativity TheoryKai Xu, Hong GeICML 2024
- Accelerating Hamiltonian Monte Carlo via Chebyshev Integration TimeJun-Kun Wang, Andre WibisonoICLR 2023
