Think First, Diffuse Fast: Improving Diffusion Language Model Reasoning via Autoregressive Plan Conditioning
On the map: LLM reasoning → Diffusion reasoning and test-time scaling
also in 1 more
Abstract (arXiv)
Diffusion large language models (dLLMs) generate text via iterative denoising but consistently underperform on multi-step reasoning. We hypothesize this gap stems from a coordination problem: AR models build coherence token-by-token, while diffusion models must coordinate all positions simultaneously. We propose plan conditioning, a training-free method that prepends a short (~100-token) natural-language plan from an AR model to the diffusion model's prompt. The plan serves as a frozen scaffold -- globally visible context that every token position can attend to from the first denoising step. On GSM8K, plan conditioning improves LLaDA-8B-Instruct from 75.6% to 87.2% (+11.6 percentage points), matching a same-size AR model (LLaMA 3.1 8B, 87.7%) despite a 6.4pp weaker baseline. On HumanEval, the gain is +12.8pp (37.2% to 50.0%), showing plans generalize to code. The same plans improve LLaMA by only +5.7pp on GSM8K and +1.3pp on HumanEval -- diffusion models benefit 2-10x more, supporting the coordination-problem hypothesis. Across 5 random seeds, plan-conditioned GSM8K accuracy has zero standard deviation, making diffusion inference highly stable. Ablations reveal the model follows plan strategy (wrong-strategy plans cause -16.3pp) but is robust to plan values (perturbed numbers: -1.1pp), and that planner quality has a sharp threshold: smaller Llama-class plans hurt (-1.6 to -6.8pp) while frontier plans provide the full lift. Attention analysis confirms the mechanism: plan tokens receive 1.8x excess attention during early denoising, declining to uniform as completion tokens solidify. Plan conditioning costs ~0.002 per problem and adds ~2s of latency.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Diffusion reasoning and test-time scaling, one of 17 directions in LLM reasoning. Builds on 14 of its 29 references that are in the map. First posted on arXiv 20 Feb 2026.
Builds on
14 of 29 references.
Closest ancestors · 3
Same direction, newest first
- 2025Planner and Executor: Collaboration between Discrete Diffusion And Autoregressive Models in Reasoning17 Oct
- 2025Mercury: Ultra-Fast Language Models Based on Diffusion17 Jun
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learning16 Apr
Other references · 11
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2023Let's Verify Step by Stepcited by 3,598
- 2025Large Language Diffusion Modelscited by 872
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2023Weak-to-Strong Generalization: Eliciting Strong Capabilities With Weak Supervisioncited by 445
- 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generationcited by 184
- 2025Planned Diffusioncited by 8
- 2026Diffusion In Diffusion: Reclaiming Global Coherence in Semi-Autoregressive Diffusioncited by 7
- 2026Plan, Verify and Fill: A Structured Parallel Decoding Approach for Diffusion Language Modelscited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025EDIT: Early Diffusion Inference Termination for dLLMs Based on Dynamics of Training Gradients
- 2025Advancing Reasoning in Diffusion Language Models with Denoising Process Rewards
- 2026CAI-DLLM: Convergence Aware Inference for Diffusion Language Models
- 2026Diffusion Language Models Are Natively Length-Aware
- 2026Just on Time: Token-Level Early Stopping for Diffusion Language Models
- 2026MetaState: Persistent Working Memory Enhances Reasoning in Discrete Diffusion Language Models
- 2025Unlocking the Potential of Diffusion Language Models through Template Infilling
- 2026Register Tokens for Bounded-State Reasoning in Diffusion Language Models