acceptodds
Under review as a conference paper at ICLR 2027

Revision Provably Reduces Sequential Computation in Diffusion Language Models

Abstract

Diffusion language models (DLMs) generate multiple tokens with a single forward pass per decoding round, achieving greater parallelism than autoregressive models (ARMs). Revision promises further reductions in sequential computation by allowing a DLM to overwrite previously generated tokens and thereby reducing the number of decoding rounds. Yet it remains unclear whether this advantage persists as DLMs evolve. To assess the expressive power of DLMs with and without revision, we work in a model where each Transformer's decoding round uses computation. We compare the minimum number of rounds each class needs to sample , where the word is uniform on and is the membership function of a binary regular language . We show that revision yields an asymptotic advantage. A DLM with revision exactly samples this distribution in at most two rounds without chain-of-thought (CoT) positions. However, a DLM without revision has tight round complexity even with a constant number of CoT positions for every , that is, every regular language whose membership cannot be computed by such circuits. Moreover, for every such and every constant , CoT positions do not suffice to sample this distribution in constant rounds without revision. We further show a separation when the goal is to approximately sample a length- word uniformly from some regular language . These results provide a theoretical framework for understanding how revision reduces sequential computation and expands the scope of parallel speedups achievable by DLMs.

open until 14 Dec 2026

est. 32% chance this paper gets accepted at ICLR 2027.

Reject 68%Accept 32%

What do you think this paper will get?

All positions stay anonymous.

Related papers

Loading the map…

Discussion (0)

Sign in to comment.