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.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.