DRLMs: parallel draft refinement beats autoregressive and diffusion reasoning baselines

microsoft · hf · 2026-10-21

KAIST and Microsoft Research Asia introduce Draft Refinement Language Models (DRLMs), which use a single denoiser to propose parallel token revisions and evaluate candidate drafts. The denoiser is trained to reconstruct every token from uniformly corrupted drafts, learning both preservation and repair; using the known corruption process, a cavity energy is computed in one forward pass without an extra scoring model. Each round proposes a revision and keeps it only if energy drops. The method outperforms autoregressive and discrete diffusion baselines on Countdown, Sudoku-Extreme and Maze-Hard.

Original post →

More from Research

Research channel →