Register Tokens for Bounded-State Reasoning in Diffusion Language Models

Albert Ge, Chandan Singh, Yufan Zhuang +3

Masked diffusion language models (dLLMs) generate text by iteratively denoising masked tokens with bidirectional attention. Extending reasoning across generation chunks normally re…