PACT: giving intermediate draft steps of diffusion language models informative RL training signals

burkov · x · 2026-11-12

Diffusion language models generate answers through a sequence of partial drafts, revealing several tokens at a time rather than writing left to right. This speeds up generation but complicates RL training: the probability of a finished answer is hard to compute exactly, and whole-answer rewards can't distinguish which intermediate decisions helped or hurt.

The article introduces PACT, which jointly learns the model's response policy and a score for each unfinished answer. The score estimates the promise of a partial answer, anchored to the completed response's reward. During training, PACT encourages each generation step's preference change relative to the original model to match the anchored reward change, yielding more informative signals for intermediate steps.

Original post →

More from Research

Research channel →