让搜索模型通过自蒸馏学会分步推理,提升长流程任务表现。
Beyond Outcome Rewards: Step-Level Self-Distilled Policy Optimization for Deep Search Agents

- 用网页提取的证据锚作为教师信号,指导每一步推理。
- 仅对错误轨迹进行梯度优化,避免正确路径被干扰。
- 在不增加太多计算量下,显著超越传统强化学习方法。
深度搜索代理需处理数十步的轨迹,但标准强化学习仅提供单个结果奖励,导致信用分配极度稀疏。有策略自蒸馏(OPSD)通过模型自身输出作为密集的词级教师信号缓解此问题,但应用于搜索任务时面临根本矛盾:教师因掌握正确答案等特权信息,其分布与学生探索式推理存在系统性差异,直接蒸馏会使学生继承这种信息不对称而非学习更优搜索策略。本文提出两种改进:首先构建证据锚(Evidence Anchors),即从网络提取的简洁、步骤级证据片段,作为不暴露完整解题路径的特权信息;其次提出步级自蒸馏策略优化(SSPO),将教师与学生间的分歧转化为GRPO中的步级优势权重,仅作用于错误轨迹。该设计将更新方向与更新强度解耦:最终奖励决定策略调整方向,教师控制每一步更新幅度。正确轨迹保持不变,保留多样性。在Qwen3-8B上,SSPO在BrowseComp、GAIA和FRAMES三个数据集上持续优于GRPO,性能超过或持平于使用两倍梯度步数训练的GRPO,且每步仅增加约5%开销(来自一次额外前向传播)。
原文摘要 · Abstract (English)
Deep search agents operate over trajectories spanning dozens of steps, yet standard reinforcement learning provides only a single outcome reward per trajectory, which is far too sparse for effective credit assignment. On-policy self-distillation (OPSD) addresses this by using the model's own logits as dense token-level teachers, but extending it to search agents introduces a fundamental tension: the teacher, having access to privileged information such as the correct answer, produces a distribution that differs systematically from the student's exploration-based reasoning, and naive distillation causes the student to inherit this information asymmetry rather than learn better search strategies. We resolve this tension through two contributions. First, we construct Evidence Anchors, which are concise, step-level evidence snippets extracted from the web, as privileged information that captures key reasoning steps without revealing the entire answer path. Second, we propose Step-Level Self-Distilled Policy Optimization (SSPO), which converts teacher-student disagreement into step-level advantage weights within GRPO, applied exclusively to incorrect trajectories. This design decouples what to update from how much to update: the outcome reward determines the direction of policy change, while the teacher modulates its magnitude at each step. Correct trajectories are left untouched, preserving their diversity. On Qwen3-8B, SSPO consistently outperforms GRPO across BrowseComp, GAIA, and FRAMES, surpassing or matching GRPO trained with twice as many gradient steps while adding only about 5 percent overhead per step from a single additional forward pass.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。