自动修复和奖励工具使用轨迹,提升大模型复杂任务推理能力
Guided by Trajectories: Repairing and Rewarding Tool-Use Trajectories for Tool-Integrated Reasoning
- 两阶段框架:先修复低质量轨迹,再基于偏好训练奖励模型
- 在真实基准上显著提升工具集成推理效果,优于现有方法
- 适合需要可靠工具调用能力的智能助手、自动化系统研发者
工具集成推理(TIR)使大语言模型通过与外部工具交互解决复杂任务,但现有方法依赖高质量合成轨迹和稀疏的结果奖励,监督信号有限且存在偏差。为此,本文提出AutoTraj,一个两阶段框架,通过自动修复和奖励工具使用轨迹来学习TIR。在监督微调(SFT)阶段,AutoTraj为每个查询生成多个候选轨迹,并从多维度评估其质量;高质量轨迹直接保留,低质量轨迹则由大模型(即LLM-as-Repairer)修复。修复后的高质量轨迹构成合成SFT数据集,修复前后轨迹对构成轨迹偏好建模数据集。在强化学习(RL)阶段,基于偏好数据训练轨迹级奖励模型,结合结果和格式奖励,明确引导优化向可靠TIR行为收敛。在真实世界基准上的实验表明,AutoTraj在TIR中具有显著有效性。
原文摘要 · Abstract (English)
Tool-Integrated Reasoning (TIR) enables large language models (LLMs) to solve complex tasks by interacting with external tools, yet existing approaches depend on high-quality synthesized trajectories selected by scoring functions and sparse outcome-based rewards, providing limited and biased supervision for learning TIR. To address these challenges, in this paper, we propose AutoTraj, a two-stage framework that automatically learns TIR by repairing and rewarding tool-use trajectories. Specifically, in the supervised fine-tuning (SFT) stage, AutoTraj generates multiple candidate tool-use trajectories for each query and evaluates them along multiple dimensions. High-quality trajectories are directly retained, while low-quality ones are repaired using a LLM (i.e., LLM-as-Repairer). The resulting repaired and high-quality trajectories form a synthetic SFT dataset, while each repaired trajectory paired with its original low-quality counterpart constitutes a dataset for trajectory preference modeling. In the reinforcement learning (RL) stage, based on the preference dataset, we train a trajectory-level reward model to assess the quality of reasoning paths and combine it with outcome and format rewards, thereby explicitly guiding the optimization toward reliable TIR behaviors. Experiments on real-world benchmarks demonstrate the effectiveness of AutoTraj in TIR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。