math.STSep 14, 2026
SaveTight Sampling Complexity with stochastic gradient oracles in Fixed Dimensions
Abstract
We investigate the stochastic-gradient query complexity of sampling smooth strongly log-concave distributions in any fixed Euclidean dimension. The potential is -strongly convex and -smooth, with an unknown mode in the ball of radius about the origin. We have access to unbiased stochastic oracles with the variance at most . For every and total variation (TV) accuracy , we prove that the tight complexity of sampling a distribution within -TV distance from the target distribution is
where is the condition number. Note that this complexity bound is simultaneously tight for the condition number and accuracy . Besides, our tight complexity bound is adaptive to noiseless setting , which is .
Explore similar work
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 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 . Combining geometrically spaced noise levels with sinc-quadrature rational approximation, we obtain a sampler with smoothed-score queries for total variation error , improving the condition-number dependence from 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 . 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 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.
A proximal gradient algorithm for composite log-concave sampling
We propose an algorithm to sample from composite log-concave distributions over , i.e., densities of the form , assuming access to gradient evaluations of and a restricted Gaussian oracle (RGO) for . The latter requirement means that we can easily sample from the density , which is the sampling analogue of the proximal operator for . If is -strongly convex and is -smooth, our sampler achieves error in total variation distance in iterations where , which matches prior state-of-the-art results for the case . We further extend our results to cases where (1) is non-log-concave but satisfies a Poincaré or log-Sobolev inequality, and (2) is non-smooth but Lipschitz.
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 fresh-sample model, every randomized adaptive algorithm requires 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.