Zeroth-Order Sampling Methods for Non-Log-Concave Distributions: Alleviating Metastability by Denoising Diffusion
Ye He, Kevin Rojas, Molei Tao
Abstract
This paper considers the problem of sampling from non-logconcave distribution, based on queries of its unnormalized density. It first describes a framework, Denoising Diffusion Monte Carlo (DDMC), based on the simulation of a denoising diffusion process with its score function approximated by a generic Monte Carlo estimator. DDMC is an oracle-based meta-algorithm, where its oracle is the assumed access to samples that generate a Monte Carlo score estimator. Then we provide an implementation of this oracle, based on rejection sampling, and this turns DDMC into a true algorithm, termed Zeroth-Order Diffusion Monte Carlo (ZOD-MC). We provide convergence analyses by first constructing a general framework, i.e. a performance guarantee for DDMC, without assuming the target distribution to be log-concave or satisfying any isoperimetric inequality. Then we prove that ZOD-MC admits an inverse polynomial dependence on the desired sampling accuracy, albeit still suffering from the curse of dimensionality. Consequently, for low dimensional distributions, ZOD-MC is a very efficient sampler, with performance exceeding latest samplers, including also-denoising-diffusion-based RDMC and RSDMC. Last, we experimentally demonstrate the insensitivity of ZOD-MC to increasingly higher barriers between modes or discontinuity in non-convex potential.
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 4e0590cf-58ab-4a60-8dd3-5793cb8027ecCited by top-tier papers13
- Accelerating Diffusion Models with Parallel Sampling: Inference at Sub-Linear Time ComplexityHaoxuan Chen, Yinuo Ren, Lexing Ying, Grant M. RotskoffNeurIPS 2024 · 53 citations
- Evaluating the design space of diffusion-based generative modelsYuqing Wang, Ye He, Molei TaoNeurIPS 2024 · 30 citations
- MDNS: Masked Diffusion Neural Sampler via Stochastic Optimal ControlYuchen Zhu, Wei Guo, Jaemoo Choi, Guan-Horng Liu et al.NeurIPS 2025 · 24 citations
- Proximal Diffusion Neural SamplerWei Guo, Jaemoo Choi, Yuchen Zhu, Molei Tao et al.ICLR 2026 · 19 citations
- Complexity Analysis of Normalizing Constant Estimation: from Jarzynski Equality to Annealed Importance Sampling and beyondWei Guo, Molei Tao, Yongxin ChenICLR 2026 · 12 citations
Builds on13
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Improved Analysis of Score-based Generative Modeling: User-Friendly Bounds under Minimal Smoothness AssumptionsHongrui Chen, Holden Lee, Jianfeng LuICML 2023 · 212 citations
- Path Integral Sampler: A Stochastic Control Approach For SamplingQinsheng Zhang, Yongxin ChenICLR 2022 · 177 citations
- Improved sampling via learned diffusionsLorenz Richter, Julius BernerICLR 2024 · 103 citations
Related papers
- Sampling from multi-modal distributions with polynomial query complexity in fixed dimension via reverse diffusionAdrien Vacher, Omar Chehab, Anna KorbaNeurIPS 2025 · 5 citations
- Reverse Diffusion Sequential Monte Carlo SamplersLuhuan Wu, Yi Han, Christian Andersson Naesseth, John P. CunninghamNeurIPS 2025 · 12 citations
- Alternating Diffusion for Proximal Sampling with Zeroth Order QueriesHirohane Takagi, Atsushi NitandaICLR 2026 · 1 citation
- Provable Benefit of Annealed Langevin Monte Carlo for Non-log-concave SamplingWei Guo, Molei Tao, Yongxin ChenICLR 2025
- Denoising Diffusion SamplersFrancisco Vargas, Will Sussman Grathwohl, Arnaud DoucetICLR 2023 · 3 citations
