On theoretical guarantees and a blessing of dimensionality for nonconvex sampling
Martin Chak
TL;DR
The paper develops explicit, scalable guarantees for sampling from high-dimensional nonconvex targets by introducing a tail-matching flattened proposal $\pi$ and correcting with importance sampling. It proves a polynomial-time sampling regime when the nonconvexity radius $R$ scales no faster than $\sqrt{d}$, and provides precise bounds on the chi-square divergence between the target and the proposal, controlled by Gaussian-tail assumptions. Conversely, it establishes sharp intractability results: under strong convexity outside a ball or mere dissipativity, there exist targets for which any algorithm requires exponential work in dimension to achieve accurate estimates. The results are illustrated with Gaussian mixtures and Bayesian neural networks, showing how the framework yields tractable sampling under concrete architectural constraints and tail behavior, thus bridging theory and scalable practice in nonconvex sampling.
Abstract
Existing guarantees for algorithms sampling from nonlogconcave measures on $\mathbb{R}^d$ are generally inexplicit or unscalable. Even for the class of measures with logdensities that have bounded Hessians and are strongly concave outside a Euclidean ball of radius $R$, no available theory is comprehensively satisfactory with respect to both $R$ and $d$. In this paper, it is shown that complete polynomial complexity can in fact be achieved if $R\leq c\sqrt{d}$, whilst an exponential number of point evaluations is generally necessary for any algorithm as soon as $R\geq C\sqrt{d}$ for constants $C>c>0$. Importance sampling with a tail-matching proposal achieves the former, owing to a blessing of dimensionality. On the other hand, if strong concavity outside a ball is replaced by a distant dissipativity condition, then sampling guarantees must generally scale exponentially with $d$ in all parameter regimes.
