arXiv:2602.01381cs.CLstat.ML2026-02被引 3

用近似奖励模型也能实现推理效率的指数级提升

On the Power of (Approximate) Reward Models for Inference-Time Scaling

  • 以贝尔曼误差为关键指标,分析近似奖励模型的性能边界
  • 当误差小于 $1/T$ 时,推理复杂度从指数降至多项式
  • 适合关注推理加速与奖励模型实用性的研究者

推理时缩放已成为提升大语言模型推理能力的重要范式。其中,序列蒙特卡洛(SMC)框架通过迭代生成、评估、拒绝和重采样中间推理路径,显著增强模型表现。该过程的核心是奖励模型,用于评估部分解并指导计算分配。然而实际部署中无法获取真实奖励模型,所有系统均依赖近似奖励模型。本文提出理论解答:近似奖励模型的有效性由其贝尔曼误差决定。对于长度为 $T$ 的推理过程,若贝尔曼误差不超过 $O(1/T)$,则结合 SMC 可将推理复杂度从 $T$ 的指数级降至多项式级,实现推理效率的指数级提升,即使仅使用近似奖励。

原文摘要 · Abstract (English)

Inference-time scaling has recently emerged as a powerful paradigm for improving the reasoning capability of large language models. Among various approaches, Sequential Monte Carlo (SMC) has become a particularly important framework, enabling iterative generation, evaluation, rejection, and resampling of intermediate reasoning trajectories. A central component in this process is the reward model, which evaluates partial solutions and guides the allocation of computation during inference. However, in practice, true reward models are never available. All deployed systems rely on approximate reward models, raising a fundamental question: Why and when do approximate reward models suffice for effective inference-time scaling? In this work, we provide a theoretical answer. We identify the Bellman error of the approximate reward model as the key quantity governing the effectiveness of SMC-based inference-time scaling. For a reasoning process of length $T$, we show that if the Bellman error of the approximate reward model is bounded by $O(1/T)$, then combining this reward model with SMC reduces the computational complexity of reasoning from exponential in $T$ to polynomial in $T$. This yields an exponential improvement in inference efficiency despite using only approximate rewards.

推理加速奖励模型序列蒙特卡洛理论分析

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