arXiv:2502.01618cs.LGcs.AI2025-02NeurIPS被引 6

用概率推断方法提升大模型推理效率,4次尝试超越GPT-4o。

Rollout Roulette: A Probabilistic Inference Approach to Inference-Time Scaling of LLMs using Particle-Based Monte Carlo Methods

  • 将推理扩增转为概率推断,用粒子蒙特卡洛采样探索状态分布。
  • 在数学推理任务中,性能提升达4至16倍,32次采样达到o1模型水平。
  • 避免奖励欺骗,适合追求高精度与鲁棒性的推理场景。

大语言模型通过增大参数量或数据量取得了显著性能提升,但近期研究显示此类方法收益递减,促使人们关注推理时计算资源的扩展。现有方法通常依赖奖励模型,将任务视为搜索问题,易受奖励模型近似误差导致的奖励劫持影响。本文将推理时扩展重构为概率推断任务,利用基于粒子的蒙特卡洛方法,在近似似然下探索状态空间模型的状态分布典型集,而非直接优化其极值点。我们提出一种新颖的推理时扩展方法,适配粒子蒙特卡洛技术。实验表明,该方法在多个挑战性数学推理任务上,相比确定性搜索基线,缩放率提升4至16倍。使用该方法,Qwen2.5-Math-1.5B-Instruct仅需4次推演即超越GPT-4o准确率,而Qwen2.5-Math-7B-Instruct在32次推演内达到o1级别准确率。本工作不仅提供了一种高效的推理时扩展方案,更将概率推断领域的丰富文献与大模型推理时扩展相结合,为未来开发更鲁棒算法奠定基础。代码、视频及更多详情见 https://probabilistic-inference-scaling.github.io。

原文摘要 · Abstract (English)

Large language models (LLMs) have achieved significant performance gains via scaling up model sizes and/or data. However, recent evidence suggests diminishing returns from such approaches, motivating scaling the computation spent at inference time. Existing inference-time scaling methods, usually with reward models, cast the task as a search problem, which tends to be vulnerable to reward hacking as a consequence of approximation errors in reward models. In this paper, we instead cast inference-time scaling as a probabilistic inference task and leverage sampling-based techniques to explore the typical set of the state distribution of a state-space model with an approximate likelihood, rather than optimize for its mode directly. We propose a novel inference-time scaling approach by adapting particle-based Monte Carlo methods to this task. Our empirical evaluation demonstrates that our methods have a 4-16x better scaling rate over our deterministic search counterparts on various challenging mathematical reasoning tasks. Using our approach, we show that Qwen2.5-Math-1.5B-Instruct can surpass GPT-4o accuracy in only 4 rollouts, while Qwen2.5-Math-7B-Instruct scales to o1 level accuracy in only 32 rollouts. Our work not only presents an effective method to inference-time scaling, but also connects the rich literature in probabilistic inference with inference-time scaling of LLMs to develop more robust algorithms in future work. Code, videos, and further information available at https://probabilistic-inference-scaling.github.io.

大模型推理概率推断数学推理采样优化

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