一句话定位

把 R1 式 RL 推到”边推理边搜索”:用结果奖励 + 检索 token 掩码稳定训练,让 LLM 在推理链中自主发起多轮搜索查询,是 2025 “agentic RL + 工具”浪潮的代表作之一。

摘要

Search-R1 把推理框架的强化学习扩展到搜索引擎使用:LLM 在逐步推理中自主生成(可多个)搜索查询并实时检索。它用多轮搜索交互优化 LLM 推理轨迹,采用”检索 token 掩码(retrieved token masking)“实现稳定的 RL 训练,并用简单的基于结果(outcome-based)的奖励函数。在七个问答数据集上,相比各种 RAG 基线(相同设置),Search-R1 使 Qwen2.5-7B 提升 41%、Qwen2.5-3B 提升 20%。论文还给出 RL 优化方法、LLM 选择、回复长度动态的经验洞见。

关键技术细节

  • 基座:Qwen2.5-3B / 7B(base 与 instruct)。
  • RL 算法:支持 PPO 与 GRPO;奖励为简单的 outcome-based(答案是否正确)。
  • 关键技巧 retrieved token masking:对检索回来的文档 token 在损失中做掩码,避免对外部内容计算策略梯度,稳定训练。
  • 交互格式:推理链中以 query 触发检索, 回填结果,多轮交错。
  • 结果:7 个 QA 数据集上较 RAG 基线 +41%(7B) / +20%(3B);分析了 response length 动态(随训练增长)等。
  • 与同期 R1-Searcher、ReSearch、DeepResearcher 等同属”RL 训练搜索 agent”方向。

原始链接

一手源存档(sources/)