ICML 2026: Self-Aware Scheduling (SAS) for Diffusion LMs

furongh · x · 2026-07-06

In a paper published at ICML 2026, researchers proposed the Self-Aware Scheduling (SAS) method to optimize reasoning in diffusion language models. The core idea: rather than forcing the model to "think longer," it is better to let the model know "which token to determine first."

By introducing a self-aware scheduling mechanism into diffusion LMs, SAS enables the model to dynamically decide the token generation order, thereby improving reasoning quality. This approach diverges from the mainstream CoT/long chain-of-thought route, offering a novel diffusion model perspective for LLM reasoning.

Related event: ICML 2026 Paper SAS: Optimizing Thought Scheduling in Diffusion LMs(15 posts)→

Original post →

More from Research

Research channel →