TTCD gives each token its own diffusion clock, wins conditional
TL;DR
- TTCD is a continuous-space diffusion language model where each token runs on its own denoising clock, so high-confidence tokens finish faster than uncertain ones.
- At 160M parameters trained on OpenWebText with self-distillation, it matches similar-sized baselines on unconditional generation at high speedups.
- It outperforms existing models on conditional generation and shows gains on structured prediction, specifically Sudoku solving.
A new paper from Parikshit Bansal and Sujay Sanghavi, Token Time Continuous Diffusion for Language Modeling, proposes a small but interesting tweak to diffusion language models: give every token its own generation clock. High-confidence tokens finish sooner, uncertain ones keep refining. At the 160M parameter scale the authors tested, a continuous-space diffusion model matches similar-sized baselines on unconditional text and beats them where diffusion LMs have historically fallen short, which is conditional generation.
Why this is worth paying attention to. Diffusion language models have been the waiting-for-their-moment alternative to autoregressive transformers for a while now. They promise parallel decoding, which in principle should be a big speed win, and in practice quality tends to collapse when you crank the parallelism up. TTCD's answer is that treating every token as if it is on the same denoising schedule wastes compute on tokens the model already knows, and forcing them all to move together drags accuracy down when you sample in parallel. Per-token times let the model be lazy about the easy tokens and careful about the hard ones. The authors describe the method as operating in continuous space and deterministically mapping Gaussian noise to a final token canvas, with training done via self-distillation on OpenWebText.
The other result the authors highlight is structured prediction, specifically Sudoku solving, where the method shows gains. That is a niche benchmark, but it is the kind of task diffusion LMs should be well-suited to in theory. Global constraints, no natural left-to-right order. A concrete win there matters as a validation of the underlying story.
The honest caveats. 160M parameters is small, the paper's headline unconditional result is comparable rather than better, and the self-distillation training recipe is not free. What the write-up does not give you is a wall-clock comparison against a modern autoregressive baseline with KV caching, or evidence that the trick generalizes past small scale. Take the specifics as reported, not settled.
If per-token timing does hold up as models get bigger, the payoff is a cheap architectural knob that closes diffusion's biggest weakness against transformers on the very regime where it was losing worst. That is the version worth watching.
Shared on Bluesky by 1 AI expert
Originally reported by paper
Read the original article →Original headline: TTCD Gives Each Token Its Own Diffusion Timeline, Beats Baselines on Conditional Text