arXiv:2509.25604cs.CLcs.LG2025-09被引 8

无需奖励模型,用扩散语言模型自动生成推理路径。

RFG: Test-Time Scaling for Diffusion Large Language Model Reasoning with Reward-Free Guidance

  • 通过对比增强与基准模型的似然比,无监督构造引导信号。
  • 在数学与代码生成任务中提升准确率最高达9.2%。
  • 适合希望不训练额外奖励模型的推理优化研究者。

扩散大语言模型(dLLMs)在大规模语言建模中展现出巨大潜力,但如何逐步引导其解决复杂问题仍具挑战。传统自回归模型依赖密集标注的中间步骤奖励模型,而dLLMs生成过程为任意顺序,中间状态为部分掩码句子,难以直接应用。本文提出无奖励引导(RFG),一种无需显式过程奖励即可引导dLLMs推理轨迹的方法。核心思想是利用增强模型与参考模型的对数似然比参数化过程奖励,其中增强模型可通过任意经强化学习或监督微调后训练的dLLM获得。我们提供了理论依据,证明RFG可诱导出无额外奖励的奖励引导采样分布。在四个具有挑战性的数学推理与代码生成基准上,使用多种经不同后训练方法增强的dLLMs进行实验,RFG在所有任务和模型类型中均显著提升性能,最高准确率提升达9.2%。结果表明,RFG是一种通用、无需训练的测试时推理扩展框架。

原文摘要 · Abstract (English)

Diffusion large language models (dLLMs) have shown great potential in large-scale language modeling, and there is an increasing interest in further improving the capacity to solve complex problems by guiding the reasoning process step by step. Common practice for autoregressive language models typically learns a process reward model with dense annotation for each intermediate step. However, this is challenging for dLLMs where the generation is in an any-order fashion and intermediate states are partially masked sentences. To this end, in this paper, we propose reward-free guidance (RFG), a principled method for guiding the reasoning trajectory of dLLMs without explicit process reward. The key idea of RFG is to parameterize the process reward by log-likelihood ratios of the enhanced and reference dLLMs, where the enhanced model can be easily obtained by any off-the-shelf dLLM that has been post-trained with reinforcement learning (RL) or supervised fine-tuning (SFT). We provide theoretical justification that RFG induces the reward-guided sampling distribution with no additional reward. We conduct comprehensive experiments on four challenging mathematical reasoning and code generation benchmarks using a diverse suite of dLLMs enhanced with various post-training methods. RFG consistently yields significant improvements across all tasks and model types, achieving accuracy gains of up to 9.2%. These findings establish RFG as a general training-free framework that scales test-time reasoning without reliance on external reward models.

扩散模型推理优化无监督引导

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。