纯强化学习训练语言模型自主用工具,通用性更强。
Tool Zero: Training Tool-Augmented LLMs via Pure RL from Scratch
- 用动态奖励机制引导模型从探索到利用工具的转变
- 相比监督微调和混合强化学习,性能提升超7%
- 适合需要强泛化能力的复杂任务场景
训练具备工具使用能力的语言模型已成为提升其处理复杂任务能力的有前景方向。当前基于监督微调的方法依赖构建大量领域特定数据集,难以有效泛化至陌生或复杂的工具使用场景。最近,强化学习(RL)展现出赋予大模型更强推理与泛化能力的潜力。本文回答一个关键问题:能否仅通过纯强化学习充分激发模型内在推理能力并提升无工具依赖的泛化性能?为此,我们提出一种基于规则的动态泛化引导奖励设计,逐步将奖励从探索性转向利用性工具使用模式。基于此,我们构建了Tool-Zero系列模型。这些模型通过直接从零开始放大基础模型(即未经后训练的基座模型)的强化学习实现,使大模型能够自主使用通用工具。实验表明,在相同设置下,我们的模型性能相比监督微调(SFT)和基于监督微调的强化学习(RL-with-SFT)模型提升超过7%。该增益在跨数据集和同数据集评估中均一致复现,验证了方法的有效性与鲁棒性。
原文摘要 · Abstract (English)
Training tool-augmented LLMs has emerged as a promising approach to enhancing language models' capabilities for complex tasks. The current supervised fine-tuning paradigm relies on constructing extensive domain-specific datasets to train models. However, this approach often struggles to generalize effectively to unfamiliar or intricate tool-use scenarios. Recently, reinforcement learning (RL) paradigm can endow LLMs with superior reasoning and generalization abilities. In this work, we address a key question: Can the pure RL be used to effectively elicit a model's intrinsic reasoning capabilities and enhance the tool-agnostic generalization? We propose a dynamic generalization-guided reward design for rule-based RL, which progressively shifts rewards from exploratory to exploitative tool-use patterns. Based on this design, we introduce the Tool-Zero series models. These models are trained to enable LLMs to autonomously utilize general tools by directly scaling up RL from Zero models (i.e., base models without post-training). Experimental results demonstrate that our models achieve over 7% performance improvement compared to both SFT and RL-with-SFT models under the same experimental settings. These gains are consistently replicated across cross-dataset and intra-dataset evaluations, validating the effectiveness and robustness of our methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。