用缓存回放和混合奖励训练小模型,92%准确率逼近GPT-5
CacheRL:Multi-Turn Tool-Calling Agents via Cached Rollouts and Hybrid Reward

- 通过LLM生成推理轨迹,教模型何时调用工具及原因
- 三阶模糊缓存消除真实执行开销,保持轨迹真实度
- 缓存感知奖励避免惩罚模型因缓存导致的误差
我们提出CacheRL,一种训练小型代理基础模型的系统,在多步工具调用任务中达到92%的过程准确率,接近GPT-5的94%,但仅需其1/100的计算量。该方法解决三个实际训练挑战:大规模迁移大模型的工具调用知识、无需昂贵实时工具执行的强化学习,以及从噪声缓存环境稳健学习。CacheRL引入三项关键创新:首先,混合思维轨迹管道通过LLM生成的推理痕迹增强代理轨迹,使模型不仅学习调用哪些工具,还理解为何调用;其次,CacheAgentLoop通过三阶模糊缓存消除实时执行成本,同时使用标记级掩码保持轨迹保真度;第三,缓存层级感知奖励动态调整答案质量权重,避免因缓存限制惩罚模型。通过迭代监督微调(SFT)与组相对策略优化(GRPO),Qwen3-4B-Thinking的验证奖励从0.43提升至0.78。在公开的代理工具调用基准上,模型性能可与前沿模型如GPT-5媲美。消融实验表明,移除知识迁移使性能下降41%,而缓存感知奖励带来17%提升。有趣的是,强化学习虽提升训练稳定性,但收益有限,超过强监督微调后效果趋平,说明数据质量和奖励设计对构建实用小模型比复杂优化方法更为关键。
原文摘要 · Abstract (English)
We present CacheRL, a system for training small agent foundation models that achieves 92 percent process accuracy on multi-step tool-calling tasks, approaching GPT-5's 94 percent while requiring 100 times less compute. Our approach addresses three challenges in practical agent training: transferring tool-calling knowledge from large models at scale, enabling reinforcement learning without costly live tool execution, and learning robustly from noisy cached environments. CacheRL introduces three key innovations. First, a hybrid thinking trajectory pipeline augments agent trajectories with LLM-generated reasoning traces, producing training examples that teach models not only what tools to call but also why. Second, the CacheAgentLoop eliminates live execution costs through a three-tier fuzzy cache while preserving trajectory fidelity using token-level masking. Third, a cache-tier-aware reward dynamically adjusts answer-quality weights to avoid penalizing models for cache-induced limitations. Through iterative supervised fine-tuning (SFT) and Group Relative Policy Optimization (GRPO), CacheRL improves Qwen3-4B-Thinking's validation reward from 0.43 to 0.78. On public agentic tool-calling benchmarks, our model achieves competitive performance against frontier models such as GPT-5. Ablation studies show that removing knowledge transfer reduces performance by 41 percent, while cache-aware rewards contribute a 17 percent improvement. Interestingly, reinforcement learning improves training stability but yields limited gains beyond strong supervised fine-tuning, suggesting that data quality and reward design play a more important role than complex optimization methods in building practical small agent models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。