Reasoning in Diffusion Large Language Models is Concentrated in Dynamic Confusion Zones
On the map: LLM reasoning → Diffusion reasoning and test-time scaling
Abstract (arXiv)
Diffusion Large Language Models (dLLMs) are rapidly emerging alongside autoregressive models as a powerful paradigm for complex reasoning, with reinforcement learning increasingly used for downstream alignment. Existing trajectory-based RL methods uniformly allocate policy gradients across denoising steps, implicitly treating all steps as equally important. We challenge this assumption by analyzing trajectories with several step-level metrics: entropy-based uncertainty, Confidence-Margin (CM) uncertainty, and Rate of Entropy Change (RoEC). These reveal structured "zones of confusion": transient spikes in uncertainty and instability that strongly predict final success or failure, while most steps remain stable. We propose Adaptive Trajectory Policy Optimization (ATPO), a lightweight step-selection strategy that dynamically reallocates gradient updates to these high-leverage steps without changing the RL objective, rewards, or compute budget. Using a hybrid RoEC+CM rule, ATPO delivers substantial gains in reasoning accuracy and training stability across benchmarks, showing that exploiting trajectory dynamics is key to advancing dLLM RL.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Diffusion reasoning and test-time scaling, one of 17 directions in LLM reasoning. Builds on 41 of its 43 references that are in the map. First posted on arXiv 19 Nov 2025.
Builds on
41 of 43 references.
Closest ancestors · 9
Same direction, newest first
- 2025Beyond Surface Reasoning: Unveiling the True Long Chain-of-Thought Capacity of Diffusion Large Language Models10 Oct
- 2025Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization9 Oct
- 2025Simple Policy Gradients for Reasoning with Diffusion Language Models5 Oct
- 2025Advancing Reasoning in Diffusion Language Models with Denoising Process Rewards2 Oct
- 2025RFG: Test-Time Scaling for Diffusion Large Language Model Reasoning with Reward-Free Guidance29 Sep
- 2025d2: Improving Reasoning in Diffusion Language Models via Trajectory Likelihood Estimation25 Sep
- 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models7 Jul
- 2025Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Models15 May
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learning16 Apr
Other references · 32
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2025Large Language Diffusion Modelscited by 912
- 2025Dream 7B: Diffusion Large Language Modelscited by 544
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 250
- 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generationcited by 187
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 178
- 2025dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Cachingcited by 169
20 more
- 2025LLaDA-V: Large Language Diffusion Models with Visual Instruction Tuningcited by 148
- 2025Fast-dLLM v2: Efficient Block-Diffusion LLMcited by 139
- 2025Dimple: Discrete Diffusion Multimodal Large Language Model with Parallel Decodingcited by 101
- 2025Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Modelscited by 88
- 2025LaViDa: A Large Diffusion Language Model for Multimodal Understandingcited by 84
- 2025Advantage Weighted Matching: Aligning RL with Pretraining in Diffusion Modelscited by 67
- 2025Sparse-dLLM: Accelerating Diffusion LLMs with Dynamic Cache Evictioncited by 65
- 2025LLaDA-MoE: A Sparse MoE Diffusion Language Modelcited by 57
- 2025SPG: Sandwiched Policy Gradient for Masked Diffusion Language Modelscited by 41
- 2025LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMscited by 40
- 2025dInfer: An Efficient Inference Framework for Diffusion Language Modelscited by 37
- 2025Continuously Augmented Discrete Diffusion model for Categorical Generative Modelingcited by 34
- 2025MDPO: Overcoming the Training-Inference Divide of Masked Diffusion Language Modelscited by 32
- 2025Adaptive Classifier-Free Guidance via Dynamic Low-Confidence Maskingcited by 21
- 2025Taming Masked Diffusion Language Models via Consistency Trajectory Reinforcement Learning with Fewer Decoding Stepcited by 17
- 2025Mask Tokens as Prophet: Fine-Grained Cache Eviction for Efficient dLLM Inferencecited by 17
- 2025Inpainting-Guided Policy Optimization for Diffusion Large Language Modelscited by 16
- 2025CreditDecoding: Accelerating Parallel Decoding in Diffusion Large Language Models with Trace Creditcited by 13
- 2025UltraLLaDA: Scaling the Context Length to 128K for Diffusion Large Language Modelscited by 13
- 2025Consolidating Reinforcement Learning for Multimodal Discrete Diffusion Modelscited by 12
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026LogicDiff: Logic-Guided Denoising Improves Zero-Shot Reasoning in Masked Diffusion Language Modelscited by 1
- 2026Escaping Confidence Trap: Evolutionary Decoding for Mathematical Reasoning in Diffusion LLMscited by 0
- 2026Pivot-SD: Efficient Self-Distillation for Masked Diffusion Language Modelscited by 0
- 2026Pivot-SD: Efficient Self-Distillation for Masked Diffusion Language Models2 Oct 2026
- 2026Escaping Confidence Trap: Evolutionary Decoding for Mathematical Reasoning in Diffusion LLMs1 Aug 2026
- 2026LogicDiff: Logic-Guided Denoising Improves Zero-Shot Reasoning in Masked Diffusion Language Models24 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2026Break the Block: Dynamic-size Reasoning Blocks for Diffusion Large Language Models via Monotonic Entropy Descent with Reinforcement Learning
- 2025Enhancing Reasoning for Diffusion LLMs via Distribution Matching Policy Optimization
- 2026The Flexibility Trap: Rethinking the Value of Arbitrary Order in Diffusion Language Models
- 2025EDIT: Early Diffusion Inference Termination for dLLMs Based on Dynamics of Training Gradients
- 2026Back on Track: Aligning Rewards and States for Reasoning in Diffusion Large Language Models
- 2026Beyond Mode-Seeking RL: Trajectory-Balance Post-Training for Diffusion Language Models
- 2026dTRPO: Trajectory Reduction in Policy Optimization of Diffusion Large Language Models
- 2026Read the Trace, Steer the Path: Trajectory-Aware Reinforcement Learning for Diffusion Language Models