Alternating Diffusion for Proximal Sampling with Zeroth Order Queries
Hirohane Takagi, Atsushi Nitanda
Abstract
This work introduces a new approximate proximal sampler that operates solely with zeroth-order information of the potential function. Prior theoretical analyses have revealed that proximal sampling corresponds to alternating forward and backward iterations of the heat flow. The backward step was originally implemented by rejection sampling, whereas we directly simulate the dynamics. Unlike diffusion-based sampling methods that estimate scores via learned models or by invoking auxiliary samplers, our method treats the intermediate particle distribution as a Gaussian mixture, thereby yielding a Monte Carlo score estimator from directly samplable distributions. Theoretically, when the score estimation error is sufficiently controlled, our method inherits the exponential convergence of proximal sampling under isoperimetric conditions on the target distribution. In practice, the algorithm avoids rejection sampling, permits flexible step sizes, and runs with a deterministic runtime budget. Numerical experiments demonstrate that our approach converges rapidly to the target distribution, driven by interactions among multiple particles and by exploiting parallel computation.
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.
Builds on14
- DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 StepsCheng Lu, Yuhao Zhou, Fan Bao, Jianfei Chen et al.NeurIPS 2022 · 2,653 citations
- Improved sampling via learned diffusionsLorenz Richter, Julius BernerICLR 2024 · 103 citations
- On the Generalization Properties of Diffusion ModelsPuheng Li, Zhong Li, Huishuai Zhang, Jiang BianNeurIPS 2023 · 86 citations
- Accelerating Convergence of Score-Based Diffusion Models, ProvablyGen Li, Yu Huang, Timofey Efimov, Yuting Wei et al.ICML 2024 · 75 citations
- Reverse Diffusion Monte CarloXunpeng Huang, Hanze Dong, Yifan Hao, Yian Ma et al.ICLR 2024 · 46 citations
Related papers
- Zeroth-Order Sampling Methods for Non-Log-Concave Distributions: Alleviating Metastability by Denoising DiffusionYe He, Kevin Rojas, Molei TaoNeurIPS 2024 · 25 citations
- Beyond Scores: Proximal Diffusion ModelsZhenghan Fang, Mateo Díaz, Sam Buchanan, Jeremias SulamNeurIPS 2025 · 6 citations
- Proximal-Based Generative Modeling for Bayesian Inverse ProblemsBoyang Zhang, Zhiguo Wang, Ya-Feng LiuICML 2026
- Reverse Diffusion Sequential Monte Carlo SamplersLuhuan Wu, Yi Han, Christian Andersson Naesseth, John P. CunninghamNeurIPS 2025 · 12 citations
- Proximal Diffusion Neural SamplerWei Guo, Jaemoo Choi, Yuchen Zhu, Molei Tao et al.ICLR 2026 · 19 citations
