mlsanityData to 7 Oct 2026

Zeroth-Order Non-Log-Concave Sampling with Variance Reduction and Applications to Inverse Problems

arXiv 2605.30573 ↗First posted 28 May 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Sampling from high-dimensional, non-log-concave distributions with unnormalized densities remains a fundamental challenge in machine learning, particularly in black-box settings where gradient information is inaccessible or computationally prohibitive. While Langevin dynamics provides a principled framework for sampling when gradients are accessible, its extension to the black-box settings suffers from high variance and lacks non-asymptotic convergence guarantees for non-log-concave sampling. To address these limitations, we propose a variance-reduced zeroth-order Langevin sampling method. Our method employs a gradient estimator that substantially reduces the variance of the classical batched zeroth-order estimator and eliminates the unfavorable dimensional dependence of the batch size required for accurate estimation, enabling practical and stable sampling. We establish the first non-asymptotic convergence guarantees for zeroth-order non-log-concave sampling in terms of varepsilon-relative Fisher information, and, under a Poincar'e inequality assumption, squared total variation distance. We further propose ZO-APMC, a posterior sampling algorithm for black-box inverse problems with pre-trained score-based generative priors, establishing the first non-asymptotic convergence guarantees for such methods. We validate our theory through synthetic experiments and demonstrate strong empirical performance on practical linear and nonlinear inverse problems.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 32 of its 65 references that are in the map. First posted on arXiv 28 May 2026.

Builds on

32 of 65 references.

References · 32

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2020Denoising Diffusion Implicit Modelscited by 9,655
  3. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  4. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
  5. 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
  6. 2014Deep Learning Face Attributes in the Wildcited by 3,679
  7. 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
  8. 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
  9. 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,036
  10. 2022Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Modelcited by 574
  11. 2021Solving Inverse Problems in Medical Imaging with Score-Based Generative Modelscited by 514
  12. 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 302
20 more
  1. 2021Robust Compressed Sensing MRI with Deep Generative Priorscited by 284
  2. 2021SNIPS: Solving Noisy Inverse Problems Stochasticallycited by 191
  3. 2022Convergence for score-based generative modeling with polynomial complexitycited by 176
  4. 2023Solving Linear Inverse Problems Provably via Posterior Sampling with Latent Diffusion Modelscited by 156
  5. 2023Score-Based Diffusion Models as Principled Priors for Inverse Imagingcited by 126
  6. 2020PAGE: A Simple and Optimal Probabilistic Gradient Estimator for Nonconvex Optimizationcited by 116
  7. 2022Parallel Diffusion Models of Operator and Image for Blind Inverse Problemscited by 113
  8. 2021Bayesian imaging using Plug & Play priors: when Langevin meets Tweediecited by 96
  9. 2024Principled Probabilistic Imaging using Diffusion Models as Plug-and-Play Priorscited by 93
  10. 2022DOLCE: A Model-Based Probabilistic Diffusion Framework for Limited-Angle CT Reconstructioncited by 73
  11. 2025InverseBench: Benchmarking Plug-and-Play Diffusion Priors for Inverse Problems in Physical Sciencescited by 67
  12. 2022Towards a Theory of Non-Log-Concave Sampling: First-Order Stationarity Guarantees for Langevin Monte Carlocited by 63
  13. 2020Deep Probabilistic Imaging: Uncertainty Quantification and Multi-modal Solution Characterization for Computational Imagingcited by 57
  14. 2024Symbolic Music Generation with Non-Differentiable Rule Guided Diffusioncited by 52
  15. 2023Provable Probabilistic Imaging using Score-Based Generative Priorscited by 51
  16. 2023Plug-and-Play split Gibbs sampler: embedding deep generative priors in Bayesian inferencecited by 41
  17. 2023Generative Plug and Play: Posterior Sampling for Inverse Problemscited by 24
  18. 2024Zeroth-Order Sampling Methods for Non-Log-Concave Distributions: Alleviating Metastability by Denoising Diffusioncited by 19
  19. 2024Ensemble Kalman Diffusion Guidance: A Derivative-free Method for Inverse Problemscited by 18
  20. 2024Solving General Noisy Inverse Problem via Posterior Sampling: A Policy Gradient Viewpointcited by 5

Built on by

Not cited yet.