为搜索增强型智能体设计双向奖励分配机制,提升多步推理稳定性
BiCAA: Bidirectional Credit Assignment for Search-Augmented Agent

- 提出双向信用分配框架,基于过程奖励对每一步搜索行为评分
- 实验表明能减少冗余搜索,训练更稳定,准确率媲美主流方法
- 适合需要复杂推理的问答系统研发者参考
多步搜索是搜索智能体的核心能力,使其能够迭代获取、优化并整合外部证据以完成复杂问答。然而,传统GRPO仅根据最终输出分配奖励,导致仅有结果监督,中间推理步骤缺乏反馈信号,易引发训练不稳定和冗余搜索行为。为此,本文引入过程奖励,提出两个互补判别标准:该步骤是否引入新证据以推动问题求解,以及是否在整体推理路径中形成关键性中间决策。基于此,提出BiCAA框架,通过融合前向可解性提升与事后成功必要性评估,生成密集且区分度高的过程奖励。该框架将两种信号加权聚合后与结果奖励融合,显著提升了策略优化稳定性,减少了冗余搜索,在多个搜索增强型QA基准上取得竞争力表现。
原文摘要 · Abstract (English)
Multi-step search is a fundamental capability for search agents, enabling them to iteratively acquire, refine, and integrate external evidence for complex reasoning QA. However, vanilla GRPO allocates rewards exclusively based on the model's final outputs, yielding outcome-only supervision with no supervisory signals for intermediate reasoning steps. Such sparse supervision easily causes training instability and redundant search behaviors on multi-step search tasks. To mitigate this limitation, we adopt process reward to deliver stepwise supervision signals. For this process reward, we propose two complementary criteria to judge each search step: whether the step yields new evidence to facilitate problem solving, and whether it forms an efficient, pivotal intermediate decision within the overall reasoning trajectory. Building on this insight, we propose BiCAA: a bidirectional credit assignment framework that delivers dense, distinguishing process rewards for search-augmented agents. BiCAA builds bidirectional process rewards by fusing two complementary signals: forward solvability gain and hindsight success criticality. The former quantifies step-wise improvements in answer plausibility, while the latter evaluates each step's necessity for final success via hindsight outcome-based criticality scoring. We modulate and aggregate the two signals and then fuse them with the outcome reward. Experiments on search-augmented QA benchmarks show that BiCAA stabilizes policy optimization, reduces redundant search behavior, and achieves competitive performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。