让大模型更聪明地用工具:按难易程度分配学习信号,提升复杂任务解决能力
HiDiffTIR: Hierarchical Difficulty-Aware Policy Optimization for Multi-Turn Tool-Integrated Reasoning

- 分层评估每轮推理的难度,动态调整学习信号
- 在三个基准上显著提升工具调用准确率和任务完成度
- 无需额外标注,仅靠现有训练数据就能实现精细优化
工具集成推理(TIR)是大模型代理通过迭代调用外部工具解决复杂任务的核心能力。强化学习已成为实现该能力的主流方法。然而,现有方法通常对整个轨迹赋予统一的优势值,且将所有正确的工具调用视为等价,忽略了不同轨迹与推理步骤间的难易差异和学习价值。这导致学习信号不够精确,难以区分简单与困难的工具使用模式。为此,我们提出HiDiffTIR,一种面向多轮工具集成推理的分层难度感知策略优化框架。该框架在轨迹与轮次两个层级上实现难度感知的信用分配,使策略能聚焦于更具信息量的轨迹和更难的推理步骤。值得注意的是,这种细粒度优化无需额外监督,仅依赖标准强化学习滚动采样所得的群体统计量即可实现。在三个工具使用基准上的大量实验表明,HiDiffTIR在多轮TIR性能和工具调用准确性方面持续优于强基线,凸显了难度感知信用分配在工具集成大模型代理策略优化中的必要性。
原文摘要 · Abstract (English)
Tool-Integrated Reasoning (TIR) is a fundamental capability for LLM agents to solve complex tasks by interacting with external tools iteratively. Reinforcement Learning (RL) has become the dominant paradigm for enabling this capability. However, existing approaches typically assign uniform trajectory-level advantages and treat all correct tool calls equally, ignoring the varying difficulty and learning value across trajectories and reasoning steps. This can lead to imprecise learning signals that do not adequately distinguish between trivial and challenging tool-use patterns. To address this limitation, we propose HiDiffTIR, a Hierarchical Difficulty-aware policy optimization framework for multi-turn TIR. HiDiffTIR performs difficulty-aware credit assignment at both trajectory and turn levels, enabling the policy to focus on more informative trajectories and harder reasoning steps. Notably, this fine-grained optimization is achieved without additional supervision, relying solely on group-level statistics derived from standard RL rollouts. Extensive experiments on three tool-using benchmarks demonstrate that HiDiffTIR consistently improves multi-turn TIR performance and tool invocation accuracy over strong RL baselines, highlighting the necessity of difficulty-aware credit assignment for effective policy optimization in tool-integrated LLM agents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。