arXiv:2608.30952cs.LGcs.CL2026-08

单策略模型比树搜索更高效地解决化学工具调用问题

One Policy Is Enough: Single-Agent Reinforcement Learning Outperforms Tree Search for Chemistry Tool Learning

  • 用单个策略模型端到端生成推理、工具调用和返回,无需多步搜索
  • 在ChemToolBench上提升工具准确率5.5%,答案正确率也更高
  • 适合需要精准工具调用的化学任务,尤其适用于资源受限场景

化学问题常需精确计算与数据库查询,语言模型无法仅凭参数完成,必须调用外部工具。工具使用涉及三要素:从大量工具中选择合适的、正确填写参数、合理串联调用链。此前的CheMatAgent采用分层进化蒙特卡洛树搜索,依赖两个学习的策略模型与两个由GPT评分引导的评价器。本文证明仅需一个策略模型即可。我们的模型将推理、工具调用与结果返回以左到右的方式统一生成,先经监督预训练,再通过直接读取真实调用链的程序化奖励进行结果级强化学习,训练过程不依赖任何学习的评价器或裁判。在ChemToolBench多工具综合化学任务上,无论使用Qwen-2.5-7B还是Llama-3.1-8B骨干,相比其最强搜索配置,工具F1分别提升5.5%与3.7%,返回F1提升9.6%与3.9%,且每题仅需一次模型调用,而搜索成本随调用树增长;同时在Qwen-2.5-7B上领先答案通过率。

原文摘要 · Abstract (English)

Chemistry questions often demand exact computation and database lookups that a language model cannot supply from its parameters, so it must reach for external tools. Tool use here is a three-part problem: select the right tool from a large pool, fill it with correctly typed arguments, and chain calls so that each consumes the outputs of the last. CheMatAgent, a previously published system, addresses this with hierarchical evolutionary MCTS: separate policy and execution models searching tool-call trees under two learned critics, one regressed partly onto GPT-assigned scores. We show that a single policy suffices. Our model interleaves reasoning, tool calls, and returns in one left-to-right generation, trained by a supervised warm-up and then outcome-level reinforcement learning against a programmatic reward read directly off the gold call chain, which leaves no learned critic and no judge in the training loop. On ChemToolBench multiple-tool comprehensive chemistry, on both backbones CheMatAgent use, we improve Tool F1 by 5.5% and Return F1 by 9.6% on Qwen-2.5-7B, and by 3.7% and 3.9% on Llama-3.1-8B, compared with their strongest search configuration, at one model invocation per question, against a search whose cost grows with the tree; we also lead answer Pass Rate on Qwen-2.5-7B.

强化学习工具调用化学智能单策略

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