English

On the $\epsilon$-Free Inference Complexity of Absorbing Discrete Diffusion

Machine Learning 2026-03-03 v2

Abstract

Absorbing discrete diffusion has emerged as a dominant framework for discrete data generation. However, a significant disparity remains between its empirical success and theoretical understanding: existing analyses fail to demonstrate a complexity advantage over the O(dln(d/ϵ))\mathcal{O}(d \ln(d/\epsilon)) baseline established for \emph{uniform} discrete diffusion. We bridge this gap by identifying a critical structural advantage: whereas uniform diffusion redundantly re-denoises valid elements, the absorbing scheme denoises each absorbing state exactly once. Leveraging this insight, we introduce \emph{Absorbing-Aware Truncated Uniformization} (AATU). We prove that AATU achieves ϵ\epsilon-TV convergence with O(dlnd)\mathcal{O}(d \ln d) complexity-\emph{independent} of the error tolerance ϵ\epsilon-thereby strictly outperforming existing uniform baselines. Beyond improving convergence rates, our analysis eliminates the restrictive bounded-score assumption commonly required in prior studies of uniformization-based inference. Furthermore, we extend AATU to time-invariant parameterizations, showing that it naturally adopts an imputation-type inference with a uniformly randomized denoising order. When combined with a lazy update strategy, TV convergence requires only O(d)\mathcal{O}(d) discrete score evaluations. These results not only establish a rigorous foundation for absorbing discrete diffusion -- confirming its efficiency in high-accuracy generation -- but also open new avenues for analyzing diffusion-based language models under the masking paradigm.

Keywords

Cite

@article{arxiv.2509.21835,
  title  = {On the $\epsilon$-Free Inference Complexity of Absorbing Discrete Diffusion},
  author = {Xunpeng Huang and Yingyu Lin and Nishant Jain and Kaibo Wang and Difan Zou and Yian Ma and Tong Zhang},
  journal= {arXiv preprint arXiv:2509.21835},
  year   = {2026}
}

Comments

48 pages