arXiv Machine Learning

Lower Bounds for Parallel Diffusion Sampling

The paper establishes the first polynomial lower bounds on the number of parallel rounds required for diffusion sampling with approximate scores. It proves a “~Ω(d^{1/3})” lower bound for sampling smooth, near‑isotropic Gaussian mixtures in ℝ^d, and an Ω(d) lower bound for uniform sampling from anisotropic axis‑aligned boxes inside the unit ball. These bounds apply to any randomized algorithm that can make polynomially many queries per round at arbitrary locations and noise levels, while maintaining inverse‑polynomial score error and constant total variation accuracy.

arXiv Machine Learning
Jul 13

A Fourier analytique approach to Gaussian mixture learning

arXiv:2004. 05813v3 Announce Type: replace-cross Abstract: Suppose that we are given independent, identically distributed random samples $x_1,\cdots,x_n$ from a mixture at most $k$ many $d$-dimensional spherical Gaussian distributions $\mu_1,\cdots,\mu_{k_0}$ of identical and known variance $\sigma^2$ in each coordinate, such that the minimum $\ell^2$ distance between two distinct centers $y_l$ and $y_j$ is greater than $2\Delta\sigma \min\{\sqrt{d},\sqrt k\}$, where $\Delta>C_0$, and $C_0$ is a sufficiently large universal constant.

By Somnath Chakraborty, Hariharan Narayanan
arXiv Machine Learning
Oct 8

Walk fast but be careful: Understanding Parallel Sampling in Masked Diffusion

The paper introduces random walks on graphs as a controllable sandbox for studying parallel sampling strategies in masked diffusion models (MDMs). It shows that common parallel unmasking methods, such as lowest‑entropy scoring, are not universally superior to random sampling and that performance depends on the graph’s conditional dependence structure. The authors also propose training‑free bisection samplers that are provably exact for random walks and demonstrate their effectiveness on both graph‑walk tasks and pretrained MDMs for language generation and reasoning benchmarks.

By Vansh Bansal, Cholyeon Cho, Syamantak Kumar, Sujay Sanghavi, Purnamrita Sarkar
arXiv Machine Learning
Oct 9

How Many Directions Must a Truncated Diffusion Sampler Retain? Matching Bounds Under Power-Law Spectra

The paper investigates how many spectral directions a truncated diffusion sampler must retain when data exhibit power‑law covariance spectra. For Gaussian data compared to a smoothed target, it establishes matching upper and lower bounds on the necessary number of retained directions, assuming a sufficiently large ambient dimension. The analysis shows that truncation error depends on the combined Wiener gains of omitted directions, meaning that even weak individual directions can collectively contribute significant error, and it provides practical guidelines for selecting the retained subspace and diffusion noise level.

By Radmehr Karimian, Ali Mohades, Johannes Lederer
arXiv Statistics ML
Aug 25

Provably adaptive sampling with uniform and remasking discrete diffusion models

The paper proves that for discrete diffusion models using uniform or remasking forward processes, an adaptive sampler based on a leave‑one‑out denoiser can achieve sampling error proportional to the score‑estimation error plus a small tolerance. The required number of discretization steps scales with the dual total correlation of the target distribution, not directly with the ambient dimension. This result shows that sampling complexity is governed by the intrinsic dependence structure of the distribution, and the authors provide an information‑theoretic analysis linking discretization error to mutual information between coordinates.

By Daniil Dmitriev, Zhihan Huang, Yuting Wei