让大模型在有限计算资源下更聪明地思考,避免无效步骤。
Spend Less, Reason Better: Budget-Aware Value Tree Search for LLM Agents
- 用动态搜索树结构,每步评估价值,智能决策下一步
- 低预算时性能比基线高4倍,证明精打细算胜过堆算力
- 无需训练,适合追求效率的推理部署场景
测试时扩展已成为提升大模型智能体可靠性的主流范式,但现有方法将计算视为充足资源,导致代理在冗余步骤或死胡同路径上耗尽令牌和工具预算。现有预算感知方法要么需要昂贵微调,要么依赖粗粒度的轨迹级启发式策略,无法在执行中干预。本文提出预算感知价值树(BAVT),一种无需训练的推理时框架,将多跳推理建模为由单一大模型主干驱动的动态搜索树,通过步骤级价值估计进行引导。另一关键创新是预算条件节点选择机制,利用剩余资源比例作为自然缩放指数作用于节点价值,实现从广泛探索到贪婪利用的无参数过渡。为应对大模型自评估常见的过度自信问题,BAVT采用残差价值预测器,评分相对进展而非绝对状态质量,从而可靠地剪枝无信息或冗余工具调用。我们进一步提供理论收敛保证,证明在显式有限预算约束下,BAVT以至少 $1-ε$ 的概率达到终止答案。在四个多跳问答基准上对两个模型家族的广泛评估表明,BAVT始终优于并行采样基线。尤其值得注意的是,在严格低预算条件下,BAVT表现超越基线在四倍资源分配下的水平,证明智能预算管理从根本上优于盲目算力扩展。
原文摘要 · Abstract (English)
Test-time scaling has become a dominant paradigm for improving LLM agent reliability, yet current approaches treat compute as an abundant resource, allowing agents to exhaust token and tool budgets on redundant steps or dead-end trajectories. Existing budget-aware methods either require expensive fine-tuning or rely on coarse, trajectory-level heuristics that cannot intervene mid-execution. We propose the Budget-Aware Value Tree (BAVT), a training-free inference-time framework that models multi-hop reasoning as a dynamic search tree guided by step-level value estimation within a single LLM backbone. Another key innovation is a budget-conditioned node selection mechanism that uses the remaining resource ratio as a natural scaling exponent over node values, providing a principled, parameter-free transition from broad exploration to greedy exploitation as the budget depletes. To combat the well-known overconfidence of LLM self-evaluation, BAVT employs a residual value predictor that scores relative progress rather than absolute state quality, enabling reliable pruning of uninformative or redundant tool calls. We further provide a theoretical convergence guarantee, proving that BAVT reaches a terminal answer with probability at least $1-ε$ under an explicit finite budget bound. Extensive evaluations on four multi-hop QA benchmarks across two model families demonstrate that BAVT consistently outperforms parallel sampling baselines. Most notably, BAVT under strict low-budget constraints surpasses baseline performance at $4\times$ the resource allocation, establishing that intelligent budget management fundamentally outperforms brute-force compute scaling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。