用过程奖励提升智能体RAG的训练效率和稳定性
Process vs. Outcome Reward: Which is Better for Agentic RAG Reinforcement Learning
- 采用细粒度过程奖励替代结果奖励,提升训练稳定性
- 仅用5000次训练即超越需9万次的现有方法
- 适合需要高效迭代与复杂推理的智能体系统研究者
检索增强生成(RAG)通过引入外部知识提升大语言模型的文本生成能力。传统RAG受限于静态流程,难以应对多步推理与复杂任务。为此,智能体RAG系统(如DeepResearch)被提出,支持动态检索、上下文迭代优化与自适应工作流。近期方法Search-R1利用最终答案正确性作为奖励信号,在强化学习中取得进展,但面临探索效率低、梯度冲突和奖励稀疏等问题。本文提出ReasonRAG,通过自动构建RAG-ProGuide数据集,为查询生成、证据提取和答案生成三个环节提供高质量过程级奖励,实现基于过程监督的强化学习。该框架使模型能自主完成搜索、提问、证据抽取与回答生成。在五个基准数据集上,ReasonRAG仅用5000个训练实例即达到优于Search-R1(需90,000实例)的性能。
原文摘要 · Abstract (English)
Retrieval-augmented generation (RAG) enhances the text generation capabilities of large language models (LLMs) by integrating external knowledge and up-to-date information. However, traditional RAG systems are limited by static workflows and lack the adaptability required for multistep reasoning and complex task management. To address these limitations, agentic RAG systems (e.g., DeepResearch) have been proposed, enabling dynamic retrieval strategies, iterative context refinement, and adaptive workflows for handling complex search queries beyond the capabilities of conventional RAG. Recent advances, such as Search-R1, have demonstrated promising gains using outcome-based reinforcement learning, where the correctness of the final answer serves as the reward signal. Nevertheless, such outcome-supervised agentic RAG methods face challenges including low exploration efficiency, gradient conflict, and sparse reward signals. To overcome these challenges, we propose to utilize fine-grained, process-level rewards to improve training stability, reduce computational costs, and enhance efficiency. Specifically, we introduce a novel method ReasonRAG that automatically constructs RAG-ProGuide, a high-quality dataset providing process-level rewards for (i) query generation, (ii) evidence extraction, and (iii) answer generation, thereby enhancing model inherent capabilities via process-supervised reinforcement learning. With the process-level policy optimization, the proposed framework empowers LLMs to autonomously invoke search, generate queries, extract relevant evidence, and produce final answers. Compared to existing approaches such as Search-R1 and traditional RAG systems, ReasonRAG, leveraging RAG-ProGuide, achieves superior performance on five benchmark datasets using only 5k training instances, significantly fewer than the 90k training instances required by Search-R1.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。