The Flexibility Trap: Rethinking the Value of Arbitrary Order in Diffusion Language Models

View PDF HTML (experimental)

Abstract:Diffusion Large Language Models (dLLMs) break the rigid left-to-right constraint of traditional LLMs, enabling token generation in arbitrary orders. Intuitively, this flexibility implies a solution space that strictly supersets the fixed autoregressive trajectory, theoretically unlocking superior reasoning potential. However, in this paper, we find that for general reasoning tasks (e.g., mathematics and coding), arbitrary order generation may in fact limit the reasoning potential of dLLMs. We observe that dLLMs tend to exploit this order flexibility to bypass high-uncertainty tokens that are crucial for exploration, which can lead to a premature collapse of solution coverage. This observation motivates a rethink of RL approaches for dLLMs, where considerable complexities, such as handling combinatorial trajectories and intractable likelihoods, are often devoted to preserving this flexibility. We show that effective reasoning can be elicited by simply forgoing arbitrary order and applying standard Group Relative Policy Optimization (GRPO) instead. Our approach, JustGRPO, is minimalist yet surprisingly effective (e.g., 89.1% accuracy on GSM8K) while fully retaining the parallel decoding ability of dLLMs. Project page: this https URL

Submission history

From: Zanlin Ni [view email]
[v1] Wed, 21 Jan 2026 16:41:58 UTC (2,006 KB)
[v2] Mon, 26 Jan 2026 08:29:32 UTC (2,272 KB)
[v3] Thu, 19 Mar 2026 02:53:39 UTC (2,274 KB)
[v4] Mon, 8 Jun 2026 15:43:52 UTC (2,081 KB)