acceptodds
Under review as a conference paper at ICLR 2027

S2D2: Fast Decoding for Diffusion LLMs via Training-Free Self-Speculation

Abstract

Block-diffusion language models offer a promising path toward faster-than-autoregressive generation by combining block-wise autoregressive decoding with within-block parallel denoising. However, in the few-step regime needed for practical acceleration, standard confidence-thresholded decoding is often brittle: aggressive thresholds hurt quality, while conservative thresholds require unnecessary denoising steps. Existing approaches that address this issue either require additional training or incur extra test-time compute. We present S2D2, a training-free self-speculative decoding framework for block-diffusion language models. Our key observation is that a block-diffusion model becomes autoregressive when the block size is reduced to one, allowing the same pretrained model to act as both drafter and verifier. S2D2 inserts a speculative verification step into standard block-diffusion decoding and uses lightweight routing policies to decide when verification is worth its cost. This yields a hybrid decoding trajectory in which the autoregressive mode acts as a local sequence-level critic, without claiming global AR distribution preservation. Across three mainstream block-diffusion families, S2D2 consistently improves the accuracy–speed tradeoff over strong confidence-thresholding baselines. On SDAR, we observe up to speedup over autoregressive decoding, and approximately over a tuned dynamic decoding baseline while improving accuracy by points. On LLaDA2.1-Mini, S2D2 remains complementary to built-in self-correction, including a conservative setting where it is faster than the static baseline with comparable accuracy.

Then back it, or bet against it.

Related papers

Open the market on this paper to see 7 more related papers.