mlsanityData to 7 Oct 2026

Test-Time Scaling in Diffusion LLMs via Hidden Semi-Autoregressive Experts

arXiv 2510.05040 ↗First posted 6 Oct 2025cs.LG cs.AIcited by 8

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

  1. 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models7 Jul
  2. 2024Diffusion of Thoughts: Chain-of-Thought Reasoning in Diffusion Language Models12 Feb

Other references · 42

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  3. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  4. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  5. 2020Denoising Diffusion Implicit Modelscited by 9,655
  6. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,601
  7. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  8. 2023Let's Verify Step by Stepcited by 3,598
  9. 2021TruthfulQA: Measuring How Models Mimic Human Falsehoodscited by 2,919
  10. 2020Learning to summarize from human feedbackcited by 2,562
  11. 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
  12. 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,722
30 more
  1. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
  2. 2025s1: Simple test-time scalingcited by 1,311
  3. 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
  4. 2025Large Language Diffusion Modelscited by 872
  5. 2025Dream 7B: Diffusion Large Language Modelscited by 515
  6. 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 385
  7. 2025L1: Controlling How Long A Reasoning Model Thinks With Reinforcement Learningcited by 381
  8. 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Modelscited by 268
  9. 2025Training Language Models to Reason Efficientlycited by 242
  10. 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 241
  11. 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusionscited by 208
  12. 2022Continuous diffusion for categorical datacited by 202
  13. 2024Scaling up Masked Diffusion Models on Textcited by 195
  14. 2025LLaDA-V: Large Language Diffusion Models with Visual Instruction Tuningcited by 145
  15. 2025AdaptThink: Reasoning Models Can Learn When to Thinkcited by 141
  16. 2023Controlled Decoding from Language Modelscited by 133
  17. 2025Towards Thinking-Optimal Scaling of Test-Time Compute for LLM Reasoningcited by 118
  18. 2025Thinkless: LLM Learns When to Thinkcited by 107
  19. 2024ARGS: Alignment as Reward-Guided Searchcited by 99
  20. 2024Decoding-Time Language Model Alignment with Multiple Objectivescited by 90
  21. 2025Think Only When You Need with Large Hybrid-Reasoning Modelscited by 64
  22. 2024DeAL: Decoding-time Alignment for Large Language Modelscited by 55
  23. 2025AdaCtrl: Towards Adaptive and Controllable Reasoning via Difficulty-Aware Budgetingcited by 50
  24. 2024Transfer Q Star: Principled Decoding for LLM Alignmentcited by 46
  25. 2025ThinkSwitcher: When to Think Hard, When to Think Fastcited by 37
  26. 2025Time Is a Feature: Exploiting Temporal Dynamics in Diffusion Language Modelscited by 33
  27. 2025LLaDA-VLA: Vision Language Diffusion Action Modelscited by 30
  28. 2025When to Continue Thinking: Adaptive Thinking Mode Switching for Efficient Reasoningcited by 26
  29. 2025LLaDA-MedV: Exploring Large Language Diffusion Models for Biomedical Image Understandingcited by 12
  30. 2025DIFFA: Large Language Diffusion Models Can Listen and Understandcited by 7

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 8.

8 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2023David helps Goliath: Inference-Time Collaboration Between Small Specialized and Large General Diffusion LMs
  2. 2026Accelerating Masked Diffusion Large Language Models: A Survey of Efficient Inference Techniques
  3. 2025Diffusion Language Model Inference with Monte Carlo Tree Search
  4. 2026Adaptation to Intrinsic Dependence in Diffusion Language Models
  5. 2026Scaling Beyond Masked Diffusion Language Models
  6. 2025Diffusion Language Models are Super Data Learners
  7. 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMs
  8. 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Models