cs.LGFeb 26, 2026

Sharp Convergence Rates for Masked Diffusion Models

Authors: Yuchen Liang, Zhiheng Tan, Ness Shroff, Yingbin Liang

Organizations: The Ohio State University

Abstract

Discrete diffusion models have achieved strong empirical performance in text and other symbolic domains, with masked (absorbing-rate) variants emerging as competitive alternatives to autoregressive models. Among existing samplers, the Euler method remains the standard choice in many applications, and more recently, the First-Hitting Sampler (FHS) has shown considerable promise for masked diffusion models. Despite their practical success, the theoretical understanding of these samplers remains limited. Existing analyses are conducted in Kullback-Leibler (KL) divergence, which often yields loose parameter dependencies and requires strong assumptions on score estimation. Moreover, these guarantees do not cover recently developed high-performance sampler of FHS. In this work, we first develop a direct total-variation (TV) based analysis for the Euler method that overcomes these limitations. Our results relax assumptions on score estimation, improve parameter dependencies, and establish convergence guarantees without requiring any surrogate initialization. Also for this setting, we provide the first convergence lower bound for the Euler sampler, establishing tightness with respect to both the data dimension dd and the target accuracy ε\varepsilon. Finally, we analyze the FHS sampler and show that it incurs no sampling error beyond that induced by score estimation, which we show to be tight with a matching lower error bound. Overall, our analysis introduces a direct TV-based error decomposition along the CTMC trajectory and a decoupling-based path-wise analysis for FHS, which may be of independent interest.

Figures & tables

Explore similar work

CardsList
  1. Efficient Sampling with Discrete Diffusion Models: Sharp and Adaptive Guarantees

    Feb 16, 2026Daniil Dmitriev, Zhihan Huang, Yuting WeiSample ComplexityDiffusion Model Sampling

  2. Vocabulary-size-independent Convergence of Discrete Diffusion Models: adjoint equations induce the right space

    May 17, 2026Kelvin Kan, Xingjian Li, Benjamin J. Zhang +3Integral Probability MetricsDiffusion Models

  3. Low-dimensional adaptation of diffusion models: Convergence in total variation

    Jan 22, 2025Jiadong Liang, Zhihan Huang, Yuxin ChenDiffusion SamplingDiffusion Models