Test-Time Scaling in Diffusion LLMs via Hidden Semi-Autoregressive Experts
On the map: LLM reasoning → Diffusion reasoning and test-time scaling
Abstract (arXiv)
Diffusion-based large language models (dLLMs) are trained flexibly to model extreme dependence in the data distribution; however, how to best utilize this information at inference time remains an open problem. In this work, we uncover an interesting property of these models: dLLMs trained on textual data implicitly learn a mixture of semi-autoregressive experts, where different generation orders reveal different specialized behaviors. We show that committing to any single, fixed inference time schedule, a common practice, collapses performance by failing to leverage this latent ensemble. To address this, we introduce HEX (Hidden semiautoregressive EXperts for test-time scaling), a training-free inference method that ensembles across heterogeneous block schedules. By doing a majority vote over diverse block-sized generation paths, HEX robustly avoids failure modes associated with any single fixed schedule. On reasoning benchmarks such as GSM8K, it boosts accuracy by up to 3.56X (from 24.72% to 88.10%), outperforming top-K margin inference and specialized fine-tuned methods like GRPO, without additional training. HEX even yields significant gains on MATH benchmark from 16.40% to 40.00%, scientific reasoning on ARC-C from 54.18% to 87.80%, and TruthfulQA from 28.36% to 57.46%. Our results establish a new paradigm for test-time scaling in diffusion-based LLMs (dLLMs), revealing that the sequence in which masking is performed plays a critical role in determining performance during inference.
Key facts. Cited by 8 papers 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 44 of its 47 references that are in the map. First posted on arXiv 6 Oct 2025.
Builds on
44 of 47 references.
Closest ancestors · 2
Same direction, newest first
- 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models7 Jul
- 2024Diffusion of Thoughts: Chain-of-Thought Reasoning in Diffusion Language Models12 Feb
Other references · 42
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2023Let's Verify Step by Stepcited by 3,598
- 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,919
- 2020Learning to summarize from human feedbackcited by 2,562
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
- 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,722
30 more
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2025s1: Simple test-time scalingcited by 1,311
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
- 2025Large Language Diffusion Modelscited by 872
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 385
- 2025L1: Controlling How Long A Reasoning Model Thinks With Reinforcement Learningcited by 381
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Modelscited by 268
- 2025Training Language Models to Reason Efficientlycited by 242
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 241
- 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusionscited by 208
- 2022Continuous diffusion for categorical datacited by 202
- 2024Scaling up Masked Diffusion Models on Textcited by 195
- 2025LLaDA-V: Large Language Diffusion Models with Visual Instruction Tuningcited by 145
- 2025AdaptThink: Reasoning Models Can Learn When to Thinkcited by 141
- 2023Controlled Decoding from Language Modelscited by 133
- 2025Towards Thinking-Optimal Scaling of Test-Time Compute for LLM Reasoningcited by 118
- 2025Thinkless: LLM Learns When to Thinkcited by 107
- 2024ARGS: Alignment as Reward-Guided Searchcited by 99
- 2024Decoding-Time Language Model Alignment with Multiple Objectivescited by 90
- 2025Think Only When You Need with Large Hybrid-Reasoning Modelscited by 64
- 2024DeAL: Decoding-time Alignment for Large Language Modelscited by 55
- 2025AdaCtrl: Towards Adaptive and Controllable Reasoning via Difficulty-Aware Budgetingcited by 50
- 2024Transfer Q Star: Principled Decoding for LLM Alignmentcited by 46
- 2025ThinkSwitcher: When to Think Hard, When to Think Fastcited by 37
- 2025Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Modelscited by 33
- 2025LLaDA-VLA: Vision Language Diffusion Action Modelscited by 30
- 2025When to Continue Thinking: Adaptive Thinking Mode Switching for Efficient Reasoningcited by 26
- 2025LLaDA-MedV: Exploring Large Language Diffusion Models for Biomedical Image Understandingcited by 12
- 2025DIFFA: Large Language Diffusion Models Can Listen and Understandcited by 7
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026CORE: Context-Robust Remasking for Diffusion Language Modelscited by 10
- 2026Prism: Efficient Test-Time Scaling via Hierarchical Search and Self-Verification for Discrete Diffusion Language Modelscited by 7
- 2026S3: Stratified Scaling Search for Test-Time in Diffusion Language Modelscited by 2
- 2026VGB for Masked Diffusion Model: Efficient Test-time Scaling for Reward Satisfaction and Sample Editingcited by 1
- 2026Adaptive Block Diffusion: Resolving Training-Inference Mismatch in Diffusion Language Modelscited by 1
- 2026Block-R1: Rethinking the Role of Block Size in Multi-domain Reinforcement Learning for Diffusion Large Language Modelscited by 0
- 2026Escaping Confidence Trap: Evolutionary Decoding for Mathematical Reasoning in Diffusion LLMscited by 0
- 2026Self-Repulsive Sampling for Diffusion Language Modelscited by 0
- 2026Self-Repulsive Sampling for Diffusion Language Models30 Sep 2026
- 2026Escaping Confidence Trap: Evolutionary Decoding for Mathematical Reasoning in Diffusion LLMs1 Aug 2026
- 2026Adaptive Block Diffusion: Resolving Training-Inference Mismatch in Diffusion Language Models28 Jun 2026
- 2026VGB for Masked Diffusion Model: Efficient Test-time Scaling for Reward Satisfaction and Sample Editing26 Jun 2026
- 2026Block-R1: Rethinking the Role of Block Size in Multi-domain Reinforcement Learning for Diffusion Large Language Models12 May 2026
- 2026S3: Stratified Scaling Search for Test-Time in Diffusion Language Models7 Apr 2026
- 2026CORE: Context-Robust Remasking for Diffusion Language Models4 Feb 2026
- 2026Prism: Efficient Test-Time Scaling via Hierarchical Search and Self-Verification for Discrete Diffusion Language Models2 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2023David helps Goliath: Inference-Time Collaboration Between Small Specialized and Large General Diffusion LMs
- 2026Accelerating Masked Diffusion Large Language Models: A Survey of Efficient Inference Techniques
- 2025Diffusion Language Model Inference with Monte Carlo Tree Search
- 2026Adaptation to Intrinsic Dependence in Diffusion Language Models
- 2026Scaling Beyond Masked Diffusion Language Models
- 2025Diffusion Language Models are Super Data Learners
- 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMs
- 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Models