1

Residual context is emerging as a key ingredient for unlocking the speed-quality Pareto frontier of diffusion language models (DLMs) by carrying forward predictive distributions across denoising steps.Yet turning this seemingly simple mechanism of computation recirculation into an effective and efficient learning paradigm is far from simple: there is no clear recipe for how residual context should be learned.We introduce ORL, a framework for on-policy residual learning in DLMs.Algorithmically, we identify that residual learning is inherently on-policy.ORL learns directly from these model-induced residual trajectories on the fly and introduces an entropy-contraction objective that encourages sharpening predictions as the model revisits its own intermediate predictions.System-wise, exact residual supervision requires capturing, retaining, and repeatedly processing many trajectory-dependent intermediate states.ORL co-designs the learning algorithm with asynchronous residual capture and transport, specialized blockdiffusion attention, and packed compute-balanced training, making on-policy residual learning practical at scale.On a 4B DLM converted from Qwen3-4B, ORL improves tokens per forward (TPF) by up to 6.2× and end-to-end throughput by approximately 5× at comparable or better generation quality.On a single H100, ORL-trained models achieve a stronger speed-quality Pareto frontier than autoregressive models accelerated with EAGLE3 and DFlashV1 speculative decoding, under both strict and relaxed verification.On LLaDA-2.1-mini, 36 H100hours and 4.2M self-generated tokens, about 10× less post-training data than prior LLaDA parallel-decoding recipes, yield up to 1.5× higher TPF and up to 3× higher throughput at matched quality.

Authors

Publication Details

Published
2026-10-26
DOI
https://doi.org/10.66977/xsci.2610.002r
Primary Topic
Natural Language Processing Techniques
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

1

1
Natural Language Processing Techniques
preprint

1

1
preprint en

Abstract

Residual context is emerging as a key ingredient for unlocking the speed-quality Pareto frontier of diffusion language models (DLMs) by carrying forward predictive distributions across denoising steps.Yet turning this seemingly simple mechanism of computation recirculation into an effective and efficient learning paradigm is far from simple: there is no clear recipe for how residual context should be learned.We introduce ORL, a framework for on-policy residual learning in DLMs.Algorithmically, we identify that residual learning is inherently on-policy.ORL learns directly from these model-induced residual trajectories on the fly and introduces an entropy-contraction objective that encourages sharpening predictions as the model revisits its own intermediate predictions.System-wise, exact residual supervision requires capturing, retaining, and repeatedly processing many trajectory-dependent intermediate states.ORL co-designs the learning algorithm with asynchronous residual capture and transport, specialized blockdiffusion attention, and packed compute-balanced training, making on-policy residual learning practical at scale.On a 4B DLM converted from Qwen3-4B, ORL improves tokens per forward (TPF) by up to 6.2× and end-to-end throughput by approximately 5× at comparable or better generation quality.On a single H100, ORL-trained models achieve a stronger speed-quality Pareto frontier than autoregressive models accelerated with EAGLE3 and DFlashV1 speculative decoding, under both strict and relaxed verification.On LLaDA-2.1-mini, 36 H100hours and 4.2M self-generated tokens, about 10× less post-training data than prior LLaDA parallel-decoding recipes, yield up to 1.5× higher TPF and up to 3× higher throughput at matched quality.

Natural Language Processing Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.

Laika access coming soon