小模型通过优化推理计算,能超越大模型的数学解题能力。
Can 1B LLM Surpass 405B LLM? Rethinking Compute-Optimal Test-Time Scaling
- 根据模型和题目难度动态调整推理计算策略,实现算力最优。
- 10亿参数模型在MATH-500上超越4050亿参数模型,0.5B模型胜过GPT-4o。
- 适合追求高效推理、资源受限场景下的大模型性能提升应用。
测试时扩展(Test-Time Scaling, TTS)是一种通过推理阶段增加计算量来提升大语言模型性能的重要方法。然而,现有研究缺乏对策略模型、过程奖励模型(PRM)及题目难度如何影响TTS的系统分析,限制了其理解与应用。本文聚焦两个核心问题:(1)在不同策略模型、PRM及题目难度下,如何实现测试时计算的最优扩展?(2)扩展计算能在多大程度上提升模型在复杂任务上的表现?能否让小模型超越大模型?在MATH-500和挑战性AIME24任务上进行综合实验后发现:(1)最优TTS策略高度依赖于策略模型、PRM与题目难度;(2)采用最优策略时,极小模型可超越大模型。例如,10亿参数模型在MATH-500上超过4050亿参数模型;0.5亿参数模型在两项任务上均优于GPT-4o;3亿参数模型击败4050亿参数模型;7亿参数模型战胜o1与DeepSeek-R1,且推理效率更高。结果表明,针对任务特性适配TTS策略至关重要,证明TTS是提升大模型推理能力的有力路径。
原文摘要 · Abstract (English)
Test-Time Scaling (TTS) is an important method for improving the performance of Large Language Models (LLMs) by using additional computation during the inference phase. However, current studies do not systematically analyze how policy models, Process Reward Models (PRMs), and problem difficulty influence TTS. This lack of analysis limits the understanding and practical use of TTS methods. In this paper, we focus on two core questions: (1) What is the optimal approach to scale test-time computation across different policy models, PRMs, and problem difficulty levels? (2) To what extent can extended computation improve the performance of LLMs on complex tasks, and can smaller language models outperform larger ones through this approach? Through comprehensive experiments on MATH-500 and challenging AIME24 tasks, we have the following observations: (1) The compute-optimal TTS strategy is highly dependent on the choice of policy model, PRM, and problem difficulty. (2) With our compute-optimal TTS strategy, extremely small policy models can outperform larger models. For example, a 1B LLM can exceed a 405B LLM on MATH-500. Moreover, on both MATH-500 and AIME24, a 0.5B LLM outperforms GPT-4o, a 3B LLM surpasses a 405B LLM, and a 7B LLM beats o1 and DeepSeek-R1, while with higher inference efficiency. These findings show the significance of adapting TTS strategies to the specific characteristics of each task and model and indicate that TTS is a promising approach for enhancing the reasoning abilities of LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。