Registers for diffusion LLMs: tokens that let models keep reasoning after history is cleared

charles_irl · x · 2026-09-17

Researchers introduce register tokens for diffusion LLMs: special tokens the model can write to and read from during text generation. The key benefit is that the model can keep its chain of reasoning even after its generation history is cleared, offering a new mechanism for state persistence in dLLMs. Details are laid out in the accompanying thread.

Related event: Register Tokens Let Diffusion LLMs Keep Reasoning After Clearing Context(2 posts)→

Original post →

More from Research

Research channel →