arXiv:2509.25779cs.AI2025-09被引 4

用奖励塑造让小模型高效完成智能体强化学习任务

Planner-R1: Reward Shaping Enables Efficient Agentic RL with Smaller LLMs

  • 通过密集过程奖励引导80亿参数模型,提升学习效率
  • 仅180次训练查询即达56.9%成功率,性能超大模型2.7倍
  • 小模型更易受奖励调节,兼具高效率与强泛化能力

我们在《TravelPlanner》基准上研究了基于大语言模型的智能体强化学习。提出的《Planner-R1》方法在仅180次训练查询下达到56.9%的最终通过率,较GPT-5的21.2%基线提升2.7倍,为公开排行榜最强结果。核心发现:80亿参数的小模型对奖励塑造极为敏感,使用密集过程级信号即可达到竞争力表现,同时比320亿参数模型节省3.5倍计算量、1.5倍内存。大模型虽在稀疏奖励下更鲁棒,但奖励塑造带来的相对增益较小且运行方差更高。课程学习未带来显著收益,而奖励塑造始终强化学习动态,使80亿模型成为最高效的智能体强化学习配置。关键的是,微调模型未出现过拟合,其在跨领域任务(包括Multi-IF、NaturalPlan和τ-Bench)上大多保持或超越基线性能。这些结果确立奖励塑造是扩展智能体强化学习的关键杠杆,凸显小模型的竞争力,并证明效率与泛化可兼得。

原文摘要 · Abstract (English)

We investigated Agentic RL with large language models on the \textsc{TravelPlanner} benchmark. Our approach, \textsc{Planner-R1}, achieved a \textbf{56.9\%} final-pass rate with only 180 training queries, a $2.7\times$ improvement over GPT-5's $21.2\%$ baseline and the strongest agentic result on the public leaderboard. A central finding was that smaller models (8B) were highly responsive to reward shaping: with dense process-level signals, they reached competitive performance while being $3.5\times$ more compute-efficient and $1.5\times$ more memory-efficient than 32B models. Larger models were more robust under sparse rewards but exhibited smaller relative gains from shaping and higher variance across runs. While curriculum learning offered no significant benefit, shaped rewards consistently amplified learning dynamics, making 8B models the most efficient setting for agentic RL. Crucially, these gains did not come at the cost of overfitting: fine-tuned models mostly maintained or exceeded baseline performance on out-of-domain tasks, including \textsc{Multi-IF}, \textsc{NaturalPlan}, and $τ$-\textsc{Bench}. These results establish reward shaping as a decisive lever for scaling agentic RL, highlight the competitive strength of smaller models, and demonstrate that efficiency can be achieved without sacrificing generalization.

智能体强化学习奖励塑造小模型高效

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