acceptodds
Published as a conference paper at NeurIPS 2026

Why Deterministic PRM Guidance Underperforms in Discrete Diffusion Reasoning

Yan Zhan, Shaobo Liu, Zhijun Gao

Abstract

We present a diagnostic study of reward guided reasoning in discrete diffusion language models (dLLMs). On in-distribution GSM8K with Dream-v0-Instruct-7B, deterministic segmental top- guidance using outcome-supervised intermediate-state reward models underperforms a simpler matched-compute baseline: independent best-of- sampling plus a GSM8K-trained final-answer outcome reward model (ORM) reranker. The setup appears well suited for PRM guidance because dLLM intermediate states partially expose solution evidence; however, these states are unordered masked subsets rather than autoregressive prefixes, and prior reward-guided decoding results rarely report matched forward-pass compute against simple verifier reranking. We study outcome-supervised PRMs trained with binary final-correctness labels on dLLM intermediate states. Under a protocol that charges denoising, PRM scoring, and ORM scoring in the same forward-pass unit, deterministic PRM-guided search trails ORM Rerank by and percentage points (pp) in seed-mean aggregate at matched candidate budgets and ; paired bootstrap CIs on seed exclude zero. Two mechanisms account for much of the headline gap: bidirectional PRM ROC-AUC (area under the receiver operating characteristic curve) decays from to with mask ratio, and deterministic top- pruning drops the guided pool's perfect-selector ceiling by  pp. A third diagnostic shows that mean pooling also weakens causal PRM variants. Matched-compute ORM Rerank thus emerges as the strong baseline when an in-distribution final-answer verifier is available; closing the gap requires PRM guidance that preserves candidate diversity and queries the scorer at denoising stages where it remains discriminative, neither of which the deterministic top- recipe satisfies.

Related papers

See more related submissions when the market opens.