首页 > AI前沿 > One Block Ahead: Promoting Sampler State in Diffusion Language Models

One Block Ahead: Promoting Sampler State in Diffusion Language Models

arXiv自然语言 2026-10-14 11:28 4 阅读 查看原文

Block-diffusion language models denoise a whole block of token positions in parallel so each forward pass can emit many tokens. However, blocks are still denoised in strict sequence, a cost that within-block acceleration leaves untouched.

Here, we investigate decoding blocks concurrently and propose One Block Ahead (OBA), a training-free decoding algorithm that denoises the next block alongside the current one.

We demonstrate that for uniform-state diffusion LMs, where a block's denoising progress is not visible in its intermediate tokens, concurrent decoding is advantageous if the next block's tokens are promoted together with the corresponding sampler state, including the self-conditioning signal and the denoising time.

We also derive a break-even criterion that relates the benefit of OBA to the relative cost of a paired-block forward pass.

On DiffusionGemma, a state-of-the-art diffusion LM, OBA increases the tokens per forward pass by 42.3% (from 15.64 to 22.25) and the single-request denoising throughput by 16.3%, averaged over ten benchmarks, with a small accuracy regression.