通过细粒度过程奖励,让智能检索生成系统更高效、更准确。
HiPRAG: Hierarchical Process Rewards for Efficient Agentic Retrieval Augmented Generation
- 将推理过程拆解为可分析步骤,动态评估每次检索必要性。
- 在7个问答数据集上准确率达65.4%(3B)和67.2%(7B),过检率降至2.3%。
- 适用于多种模型与强化学习算法,提升搜索效率和可靠性。
智能检索增强生成(Agentic RAG)是一种利用外部信息弥补大模型知识缺口的强大技术,有助于提升问题求解与问答能力。然而,普遍存在检索行为低效的问题,如过度检索(重复获取已知信息)和不足检索(应检索时未检索),导致计算开销增加且输出不可靠。现有基于结果的强化学习训练方法缺乏对检索过程的精细控制。为此,我们提出层次化过程奖励机制(HiPRAG),在强化学习训练中引入基于知识的细粒度过程奖励。该方法将智能体的推理轨迹分解为离散可解析的步骤,实时评估每一步检索的必要性,并采用分层奖励函数,在常用的结果与格式奖励基础上,额外根据最优检索与非检索步骤的比例给予奖励。在Qwen2.5和Llama-3.2模型上,于七个不同问答基准测试中,平均准确率分别达到65.4%(3B)和67.2%(7B),同时将过检率降低至2.3%,并减少欠检率。实验表明,优化推理过程本身比仅优化最终结果更有效。进一步分析显示,HiPRAG在多种强化学习算法、模型家族、规模及类型间具有良好泛化能力。本工作证明了通过强化学习实现细粒度控制在提升搜索智能体效率与最优性方面的潜力。
原文摘要 · Abstract (English)
Agentic RAG is a powerful technique for incorporating external information that LLMs lack, enabling better problem solving and question answering. However, suboptimal search behaviors exist widely, such as over-search (retrieving information already known) and under-search (failing to search when necessary), which leads to unnecessary overhead and unreliable outputs. Current training methods, which typically rely on outcome-based rewards in a RL framework, lack the fine-grained control needed to address these inefficiencies. To overcome this, we introduce Hierarchical Process Rewards for Efficient agentic RAG (HiPRAG), a training methodology that incorporates a fine-grained, knowledge-grounded process reward into the RL training. Our approach evaluates the necessity of each search decision on-the-fly by decomposing the agent's reasoning trajectory into discrete, parsable steps. We then apply a hierarchical reward function that provides an additional bonus based on the proportion of optimal search and non-search steps, on top of commonly used outcome and format rewards. Experiments on the Qwen2.5 and Llama-3.2 models across seven diverse QA benchmarks show that our method achieves average accuracies of 65.4% (3B) and 67.2% (7B). This is accomplished while improving search efficiency, reducing the over-search rate to just 2.3% and concurrently lowering the under-search rate. These results demonstrate the efficacy of optimizing the reasoning process itself, not just the final outcome. Further experiments and analysis demonstrate that HiPRAG shows good generalizability across a wide range of RL algorithms, model families, sizes, and types. This work demonstrates the importance and potential of fine-grained control through RL, for improving the efficiency and optimality of reasoning for search agents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。