stat.MLNov 2, 2026

Subpolynomial-Query Proximal Sampling Beyond Log-Concavity

Authors: Wei Guo, Bo Yuan, Ye He, Jiaming Liang, Molei Tao, Yongxin Chen

Organizations: Georgia Institute of Technology · University of Colorado Boulder · University of Rochester

Abstract

We study the gradient query complexity of sampling from a density π(x)∝e−V(x)π(x)\propto\mathrm{e}^{-V(x)} on Rd\mathbb{R}^d, where VV has a ββ-Lipschitz gradient but need not be convex. Our approach uses the proximal sampler, which alternates Gaussian perturbation with sampling a position conditional on the noisy observation. We implement its observation chain using the high-order noisy conditional sampler of [OpenAI 2026], obtaining improved dimension dependence under functional inequalities. If ππ satisfies a log-Sobolev inequality (LSI) with constant αα, then for every fixed ν>0ν>0, there is a finite constant CνC_ν depending only on νν such that the algorithm returns a sample within total-variation distance ε\varepsilon using at most Cνκ(d+κ+ε−1+∥∇V(x0)∥2/β)νC_νκ(d+κ+\varepsilon^{-1}+\|\nabla V(x_0)\|^2/β)^ν queries, where κ=β/ακ=β/α and x0x_0 is an arbitrary reference point for initialization. The analysis controls center-dependent local errors along exact proximal histories and removes the final observation noise by conditional decoding. Under a Poincaré inequality (PI), we obtain a corresponding guarantee from a warm start with a known bound on its Rényi divergence. An oracle lower bound shows that a stationary point alone, which suffices under LSI, is insufficient under PI.

Figures & tables

Explore similar work

CardsList
  1. Complexity of Non-Log-Concave Sampling in Fisher Information

    May 15, 2026Sinho Chewi, Andre WibisonoOracle ComplexityProximal Sampling

  2. Smoothed Score Queries and the Complexity of Sampling

    May 26, 2026Jingbo LiuSample ComplexityCommunication Complexity