arXiv:2601.06922cs.CL2026-01被引 4

用树结构在线监督代理式RAG的每步推理,提升效果与效率

TreePS-RAG: Tree-based Process Supervision for Reinforcement Learning in Agentic RAG

  • 构建推理树,通过后代结果估算每步价值,实现无标注的逐步奖励
  • 在7个问答基准上超越主流方法,多跳和通用问题均显著提效
  • 无需中间标注或离线数据,适合资源受限的实时智能系统

代理式检索增强生成(RAG)将问答建模为推理与信息检索的多步交互,近期通过基于结果的强化学习(RL)获得进展。然而,仅依赖稀疏的最终奖励限制了步骤级信用分配,对中间推理和行为指导不足。现有过程监督方法通常依赖离线训练数据,存在分布偏移风险,或需高成本的中间标注。我们提出TreePS-RAG,一种面向代理式RAG的在线树结构强化学习框架,支持逐步信用分配并保留标准的结果奖励。核心思想是将代理式RAG推理建模为回溯树,每一步推理对应一个节点,通过蒙特卡洛估计其后代结果来评估步骤效用,从而获得细粒度的过程优势,无需中间标签。为使该范式实用,我们设计了一种高效的在线树构建策略,在有限计算预算下保持探索多样性。实验表明,当回溯开销接近强基线Search-R1时,TreePS-RAG在多个模型规模下的七个多跳及通用问答基准上,持续且显著优于基于结果监督和领先过程监督的RL方法。

原文摘要 · Abstract (English)

Agentic retrieval-augmented generation (RAG) formulates question answering as a multi-step interaction between reasoning and information retrieval, and has recently been advanced by reinforcement learning (RL) with outcome-based supervision. While effective, relying solely on sparse final rewards limits step-wise credit assignment and provides weak guidance for intermediate reasoning and actions. Recent efforts explore process-level supervision, but typically depend on offline constructed training data, which risks distribution shift, or require costly intermediate annotations. We present TreePS-RAG, an online, tree-based RL framework for agentic RAG that enables step-wise credit assignment while retaining standard outcome-only rewards. Our key insight is to model agentic RAG reasoning as a rollout tree, where each reasoning step naturally maps to a node. This tree structure allows step utility to be estimated via Monte Carlo estimation over its descendant outcomes, yielding fine-grained process advantages without requiring intermediate labels. To make this paradigm practical, we introduce an efficient online tree construction strategy that preserves exploration diversity under a constrained computational budget. With a rollout cost comparable to strong baselines like Search-R1, experiments on seven multi-hop and general QA benchmarks across multiple model scales show that TreePS-RAG consistently and significantly outperforms both outcome-supervised and leading process-supervised RL methods.

强化学习RAG推理优化树结构

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。