SOURCE-LINKED INTELLIGENCE

Survival-Guided Length Control for Efficient Diffusion Language Models

arXiv · AI, language, vision and robotics · article · Aug 26, 2026 · UTC

Diffusion language models (DLMs) generate text by iteratively denoising masked sequences, but standard decoding either fixes the sequence length or relies on ad hoc stopping rules, often leading to unnecessary denoising steps. We recast length selection as a discrete-time survival problem over the end-of-sequence token and propose a plug-in, training-free length predictor that can be added to any existing DLM. Across reasoning and code-generation benchmarks, survival-guided length decoding speeds up inference by up to 7 times while preserving task accuracy. We further find that predicted lengt

Read original source ↗ Open in workspace

recordType

paper

region

Global