让无效查询重用于训练,提升大模型搜索效率
Effective Reinforcement Learning for Agentic Search by Recycling Zero-Variance Queries During Training
- 将无方差查询回收再利用,动态优化训练数据分布
- 1.7B模型在7个多跳问答任务上达66.0%准确率,媲美7B模型
- 适合做智能搜索代理训练、强化学习调优的研究者
GRPO类算法已成为基于结果奖励训练大型语言模型搜索代理的标准方法。这类算法中,仅当回溯组同时包含成功与失败时,查询才贡献于参数更新;全正确(太简单)和全错误(太难)的组为零方差,造成回溯成本浪费。现有方法将零方差视为静态属性,直接丢弃或预过滤。我们假设并实证验证:随着策略在训练中演化,查询会在这两种状态间切换。基于此,我们提出查询回收机制,将零方差组返回可变池中供未来重采样,使有效训练分布随策略共同演化。采用该技术后,1.7B参数模型在合成数据上训练,可在7个多跳问答基准上达到66.0%平均Pass@1准确率,与使用基准数据监督的高达7B参数系统相当或更优。对回收模式的分析显示,训练结束时约四分之三的有效批次由回收查询提供,其贡献来自策略改进后的恢复和策略漂移。
原文摘要 · Abstract (English)
The use of GRPO-style algorithms has become the standard strategy for training LLM search agents under outcome-only rewards. With these algorithms, a query contributes to parameter updates only when its rollout group mixes successes and failures; all-correct (too-easy) and all-incorrect (too-hard) groups are zero-variance and waste rollout cost. Existing approaches treat zero-variance as a static property and either discard or pre-filter such groups. We hypothesize and empirically validate that queries flip between zero-variance and signal-bearing states as the policy evolves during training. Building on this intuition, we propose query recycling, which returns zero-variance groups to a mutable pool for future resampling, so that the effective training distribution co-evolves with the policy. With the proposed technique, a 1.7B parameter model trained on synthetic data can reach 66.0 average Pass@1 accross seven multi-hop QA benchmarks, matching or surpassing systems with up to 7B parameters trained on benchmark-derived supervision. Analysis of recycling patterns shows that recycled queries supply roughly three quarters of the effective batch by the end of training, with contributions split between recovery from policy improvement and policy drift.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。