math.STSep 14, 2026

Tight Sampling Complexity with stochastic gradient oracles in Fixed Dimensions

Authors: Weiming OuXiao Wang

Abstract

We investigate the stochastic-gradient query complexity of sampling smooth strongly log-concave distributions in any fixed Euclidean dimension. The potential is μ\mu-strongly convex and LL-smooth, with an unknown mode in the ball of radius μ1/2\mu^{-1/2} about the origin. We have access to unbiased stochastic oracles with the variance at most σ2\sigma^2. For every σ20\sigma^2\ge0 and total variation (TV) accuracy 0<ε1/100<\varepsilon\le1/10, we prove that the tight complexity of sampling a distribution within ϵ\epsilon-TV distance from the target distribution is

NTV=Θ ⁣(log(1+κ)+σ2μϵ),N^\star_{\text{TV}}=\Theta\!\left(\log(1+\kappa)+ \frac{\sigma^2}{\mu\epsilon}\right),

where κ:=Lμ\kappa:=\frac L\mu is the condition number. Note that this complexity bound is simultaneously tight for the condition number κ\kappa and accuracy ϵ\epsilon. Besides, our tight complexity bound is adaptive to noiseless setting σ=0\sigma=0, which is NTV=Θ ⁣(log(1+κ)) N^\star_{\text{TV}}=\Theta\!\left(\log(1+\kappa)\right).

Explore similar work

May 26, 2026cs.DS

Smoothed Score Queries and the Complexity of Sampling

We study the query complexity of sampling from high-dimensional Gaussian distributions using gradient information. In the standard oracle model, exact gradients expose only matrix-vector products with the precision matrix, leading to polynomial approximation barriers and a characteristic κ\sqrtκ dependence on the condition number. We show that this barrier disappears when the sampler is allowed to query \emph{smoothed scores}, namely gradients of the logarithms of the Gaussian-convolved densities. For a Gaussian target with precision matrix ΛΛ, a smoothed-score query at noise level ττ gives access to the resolvent (Λ+τ1I)1(Λ+τ^{-1}I)^{-1}. Combining geometrically spaced noise levels with sinc-quadrature rational approximation, we obtain a sampler with q=O ⁣((logκ+log(ed/δTV))log(ed/δTV))q=O\!\left(\bigl(\logκ+\log(e\sqrt d/δ_{\rm TV})\bigr)\log(e\sqrt d/δ_{\rm TV})\right) smoothed-score queries for total variation error δTVδ_{\rm TV}, improving the condition-number dependence from κ\sqrtκ to logarithmic. We also study finite-bit gradient oracles. Using coordinatewise quantization of the transformed smoothed-score answers and a final dithering step, we obtain a sampling scheme whose total communicated gradient information is polylogarithmic in κκ; in particular, for fixed dimension and accuracy, the bit complexity is O(log2κ)O(\log^2κ). To complement these upper bounds, we introduce a channel-synthesis, or reverse-Shannon, converse technique for sampling lower bounds. This converts total-variation simulation guarantees into communication requirements and yields an Ω(logκ)Ω(\logκ) lower bound on the required gradient information. Together, these results identify smoothed scores as a provably more informative oracle for sampling and give nearly matching upper and lower bounds for its finite-bit complexity.
Jingbo Liu
May 12, 2026math.ST

A proximal gradient algorithm for composite log-concave sampling

We propose an algorithm to sample from composite log-concave distributions over Rd\mathbb{R}^d, i.e., densities of the form πefgπ\propto e^{-f-g}, assuming access to gradient evaluations of ff and a restricted Gaussian oracle (RGO) for gg. The latter requirement means that we can easily sample from the density RGOg,h,y(x)exp(g(x)12hyx2)\text{RGO}_{g,h,y}(x) \propto \exp(-g(x) -\frac{1}{2h}||y-x||^2), which is the sampling analogue of the proximal operator for gg. If f+gf + g is αα-strongly convex and ff is ββ-smooth, our sampler achieves ε\varepsilon error in total variation distance in O~(κdlog4(1/ε))\widetilde{\mathcal O}(κ\sqrt d \log^4(1/\varepsilon)) iterations where κ:=β/ακ:= β/α, which matches prior state-of-the-art results for the case g=0g=0. We further extend our results to cases where (1) ππ is non-log-concave but satisfies a Poincaré or log-Sobolev inequality, and (2) ff is non-smooth but Lipschitz.
Linghai Liu, Sinho Chewi
Aug 10, 2026math.OC

A Tight Lower Bound for Smooth Nonconvex Stochastic Optimization with Bounded Gradient Noise

We prove a sharp lower bound for smooth nonconvex stochastic optimization with uniformly bounded gradient noise. In the K=1K=1 fresh-sample model, every randomized adaptive algorithm requires Ω(ΔLε2+ΔLσ2ε4)Ω\left( \frac{ΔL}{ε^2} + \frac{ΔLσ^2}{ε^4} \right) queries to find a point with expected gradient norm at most εε. This matches the standard upper bound and, to the best of our knowledge, resolves the question raised by [Arjevani et al. 2023] of whether almost-surely bounded oracle error permits a better rate than bounded variance. The proof was independently generated with GPT-5.6 Sol in Codex's Ultra mode during a two-hour session. The human author supplied the prompt and was responsible only forchecking the proof and revising and polishing the manuscript.
Jikai Jin