arXiv:2604.02869cs.AI2026-04被引 7

通过迭代奖励校准,让小模型在客服任务上超越大模型。

Multi-Turn Reinforcement Learning for Tool-Calling Agents with Iterative Reward Calibration

  • 用多轮奖励校准优化每轮奖励设计,解决奖励方向错位问题。
  • 小模型(4B)在航空任务上从63.8%提升至66.7%,超过GPT-4.1。
  • 适合做低成本高效率工具调用智能体的开发者参考。

在多轮任务中训练工具调用智能体时,稀疏的结果奖励和跨对话回合的信用分配难题仍具挑战性。本文首次将多轮组相对策略优化(MT-GRPO)与广义词级别策略优化(GTPO)结合,用于基于大语言模型用户模拟器的真实客服任务训练。通过对训练轨迹的系统分析,发现简单设计的每轮密集奖励会导致性能下降最高达14个百分点,原因是奖励区分度与优势方向不匹配。为此提出迭代奖励校准方法,基于轨迹数据的实证区分度分析设计每轮奖励,并证明其混合优势形式可消除方向错位。在Tau-Bench航空基准测试中,该方法使Qwen3.5-4B准确率从63.8%提升至66.7%(+2.9pp),Qwen3-30B-A3B从58.0%提升至69.5%(+11.5pp)。其中4B模型虽仅为GPT-4.1(49.4%)的1/50,但表现更优;30.5B MoE模型接近Claude Sonnet 4.5(70.0%)。据我们所知,这是首个公开的在Tau-Bench上的强化学习训练结果。代码、奖励校准分析及训练方案均已开源。

原文摘要 · Abstract (English)

Training tool-calling agents with reinforcement learning on multi-turn tasks remains challenging due to sparse outcome rewards and difficult credit assignment across conversation turns. We present the first application of MT-GRPO (Multi-Turn Group Relative Policy Optimization) combined with GTPO (Generalized Token-level Policy Optimization) for training a tool-calling agent on realistic customer service tasks with an LLM-based user simulator. Through systematic analysis of training rollouts, we discover that naively designed dense per-turn rewards degrade performance by up to 14 percentage points due to misalignment between reward discriminativeness and advantage direction. We introduce Iterative Reward Calibration, a methodology for designing per-turn rewards using empirical discriminative analysis of rollout data, and show that our GTPO hybrid advantage formulation eliminates the advantage misalignment problem. Applied to the Tau-Bench airline benchmark, our approach improves Qwen3.5-4B from 63.8 percent to 66.7 percent (+2.9pp) and Qwen3-30B-A3B from 58.0 percent to 69.5 percent (+11.5pp) -- with the trained 4B model exceeding GPT-4.1 (49.4 percent) and GPT-4o (42.8 percent) despite being 50 times smaller, and the 30.5B MoE model approaching Claude Sonnet 4.5 (70.0 percent). To our knowledge, these are the first published RL training results on Tau-Bench. We release our code, reward calibration analysis, and training recipes.

强化学习工具调用奖励校准大模型优化

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