通过动态组合生成路径,显著提升扩散语言模型推理效果
Test-Time Scaling in Diffusion LLMs via Hidden Semi-Autoregressive Experts
- 不需训练,通过多数投票融合多种分块生成顺序
- 在GSM8K上准确率提升3.56倍,达88.10%
- 适合追求高精度推理的开发者和研究者
基于扩散的大型语言模型(dLLMs)训练时可灵活建模数据分布中的极端依赖关系;然而,如何在推理阶段有效利用这一信息仍是未解问题。本文发现:在文本数据上训练的dLLMs会隐式学习一组半自回归专家,不同生成顺序揭示出不同专业行为。若采用固定推理调度(常见做法),则会因无法利用该潜在集成而性能下降。为此,我们提出HEX(隐藏半自回归专家用于测试时扩展),一种无需训练的推理方法,通过在异构分块调度间进行集成。通过对多种分块大小生成路径进行多数投票,HEX能稳健避免单一固定调度带来的失败模式。在GSM8K等推理基准上,准确率最高提升3.56倍(从24.72%增至88.10%),优于Top-K边缘推理和专用微调方法GRPO,且无需额外训练。HEX还在MATH(16.40%→40.00%)、ARC-C科学推理(54.18%→87.80%)和TruthfulQA(28.36%→57.46%)上取得显著提升。结果确立了扩散语言模型测试时扩展的新范式,揭示掩码执行顺序在推理中起决定性作用。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。