Skip to content
Preprint

Answer First, Reason Later: When Commitment Order Costs Accuracy in Diffusion Language Models

Aug 2026 · 3 citations · 27 references
Computer Science

TL;DR

Masked diffusion language models revise many masked output positions in parallel and call a response answer-first when the final answer commits before the reasoning printed ahead of it, which limits how far the intervention result can be generalized.

Abstract

Masked diffusion language models revise many masked output positions in parallel. We call a token committed once it becomes visible and is never masked again, and call a response answer-first when the final answer commits before the reasoning printed ahead of it. On 1,069 GSM8K test questions, an explicit step-by-step instruction increases the accuracy difference between unrestricted decoding and a decoder that permits commitment only near the left-most unresolved position; unrestricted decoding also produces more answer-first trajectories. On MATH-500, the two LLaDA models spend most of a short output canvas on reasoning that commits after the answer, and the benefit of frontier gating decreases as that postanswer writing disappears. Dream-7B has little post-answer writing and follows a different accuracy pattern. A controlled four-option task reserves a one-token answer position before generation. Delaying that position outperforms an equally timed reasoning-token delay on LLaDA-8B, LLaDA-1.5, and Dream-7B. The raw difference is largest on Dream, whose free accuracy on the controlled task is lower. Answers commit much earlier under the reserved-position interface than in ordinary free-form generation, which limits how far the intervention result can be generalized. Commitment order affects the context used to complete a response and the allocation of a finite output canvas.

View source

Similar papers

#artificial intelligence Preprint Sep 2026

Reliable Parallel Decoding in Masked Diffusion Language Models

Masked diffusion language models (MDLMs) can generate text efficiently by predicting multiple masked tokens in parallel, but predictions from the same forward pass are not necessarily reliable when committed together. We study when parallel commitment is reliable. Our diagnostics show that confidence alone does not det...

Zheng-Hao He, Bo-Han Liu, Guang-Zhi Xiong et al. · 0 citations
Preprint Aug 2026

Shortcut Before Circuit: Document Statistics Time In-Context Conflict Resolution

The authors train 26M-parameter transformers on a synthetic language where recency and rarity are exactly coextensive, and separate them with a minimal causal edit that inverts one cue while holding the truth, token count and answer position fixed.

Yijun Liao, Fan-Wei Liang · 0 citations
#machine learning Preprint Sep 2026

Self-Repulsive Sampling for Diffusion Language Models

Sampling several responses and voting over their answers can improve a language model's accuracy, but repeated answers limit the benefit of additional samples. Raising temperature increases diversity at a potential cost to per-sample accuracy. We introduce Self-Repulsion (SR), a sampler for masked diffusion language mo...

Michael A. Helcig, Martin Jaggi · 0 citations
#artificial intelligence Preprint Sep 2026

Unmask the State: When Does State Adaptation Matter for Masked Diffusion Language Models

Masked diffusion language models (MDMs) admit flexible generation orders, making the unmasking strategy an inference decision. Existing methods vary in how they prioritize positions, control parallelism, restrict selection regions, revise predictions, or plan future denoising, yet it remains unclear when these choices...

InJin Kong, Sung-hwan Choi, Yohan Jo · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.