用模型自信心指导推理,高效提升大模型数学能力
Guided by Gut: Efficient Test-Time Scaling with Reinforced Intrinsic Confidence
- 仅用模型内部信号做树搜索,不依赖外部验证器
- 1.5B小模型达32B-70B大模型水平,内存降低10倍
- 适合资源有限但需强推理的部署场景
测试时扩展(TTS)方法常因依赖外部过程奖励模型(PRM)或采样策略(如Best-of-N)导致计算开销大。本文提出无需外部验证器的自引导框架Guided by Gut(GG),通过轻量级树搜索结合令牌级置信度与步骤新颖性实现媲美PRM的性能。关键创新在于通过针对性强化学习微调提升内部置信估计可靠性。在复杂数学推理基准上的实证表明,GG使1.5B参数模型达到甚至超越32B-70B参数模型的准确率,同时将GPU内存使用减少高达10倍。相比基于PRM的方法,GG实现相当准确率下8倍更快推理速度与4-5倍更低内存消耗;此外,相比BoN策略,其KV缓存占用减少约50%,显著提升TTS技术的实际部署效率。
原文摘要 · Abstract (English)
Test-Time Scaling (TTS) methods for enhancing Large Language Model (LLM) reasoning often incur substantial computational costs, primarily due to extensive reliance on external Process Reward Models (PRMs) or sampling methods like Best-of-N (BoN). This paper introduces Guided by Gut (GG), an efficient self-guided TTS framework that achieves PRM-level performance without costly external verifier models. Our method employs a lightweight tree search guided solely by intrinsic LLM signals, token-level confidence and step novelty. One critical innovation is improving the reliability of internal confidence estimates via a targeted reinforcement learning fine-tuning phase. Empirical evaluations on challenging mathematical reasoning benchmarks demonstrate that GG enables smaller models (e.g., 1.5B parameters) to achieve accuracy matching or surpassing significantly larger models (e.g., 32B-70B parameters), while reducing GPU memory usage by up to 10x. Compared to PRM-based methods, GG achieves comparable accuracy with 8x faster inference speeds and 4-5x lower memory usage. Additionally, GG reduces KV cache memory usage by approximately 50% compared to the BoN strategy, facilitating more efficient and practical deployment of TTS techniques.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。