arXiv:2608.29622cs.MAcs.AI2026-08

用记忆栈和细粒度动作提升大模型多步推理的准确性和连贯性。

AgenticRag-R1: Agentic Reinforcement Learning with Stack Memory for Multi-Step Reasoning, Retrieval and Memorizing

论文配图:AgenticRag-R1: Agentic Reinforcement Learning with Stack Memory for Multi-Step Reasoning, Retrieval and Memorizing
图 1 · 摘自论文原文
  • 引入记忆栈与细粒度动作空间,实现推理、检索与记忆的深度协同。
  • 在多跳问答等任务上超越强基线,长程推理准确率显著提升。
  • 适合需要复杂逻辑推理与持续记忆的智能体系统研发者。

检索增强生成(RAG)提升了大语言模型的事实性,但现有系统在需要自适应检索和中间上下文持续修正的多步推理任务中仍表现不佳。基于强化学习的智能体RAG方法虽有所改进,但通常依赖粗粒度动作空间和轨迹级奖励,导致奖励分配弱,倾向于短周期、刻板的推理模式。为此,我们提出AgenticRag-R1,一种通过记忆栈与细粒度动作空间深度融合推理、检索与记忆的强化学习框架,结合分层动作感知奖励与信息感知轨迹拒绝策略,实现有效的长时程学习。在涵盖多跳、开放域及智能体推理等多种基准测试中,无论使用何种主干模型规模,AgenticRag-R1均持续优于强基线。此外,该方法学习到更鲁棒、可解释且具备记忆意识的推理行为,凸显了细粒度动作建模与信息感知优化对长时程推理的关键作用。代码已公开于https://github.com/jiangxinke/Harness-RL/tree/AgenticRAG-R1-Whitebox。

原文摘要 · Abstract (English)

Retrieval-Augmented Generation (RAG) improves the factuality of large language models (LLMs), yet existing RAG systems often struggle with complex, multi-step reasoning that requires adaptive retrieval and continuous revision of intermediate contexts. Recent reinforcement learning (RL)-based agentic RAG methods partially alleviate this issue, but typically rely on coarse-grained action spaces and trajectory-level rewards, resulting in weak reward assignment and a bias toward short-horizon, stereotyped reasoning template. To address, we propose AgenticRag-R1, a RL framework that deeply integrates reasoning, retrieval, and memory via a memory stack and fine-grained action space, supported by hierarchical action-aware rewards and an information-aware trajectory rejection strategy to enable effective long-horizon learning. Experiments across a diverse set of multi-hop, open-domain, and agentic reasoning benchmarks, spanning multiple backbone model sizes, demonstrate that AgenticRag-R1 consistently outperforms strong baselines. Moreover, AgenticRag-R1 learns more robust, interpretable, and memory-aware reasoning behaviors, highlighting the effect of fine-grained action modeling and information-aware optimization for long-horizon reasoning. Our code is anonymous available at https://github.com/jiangxinke/Harness-RL/tree/AgenticRAG-R1-Whitebox.

强化学习多步推理记忆机制RAG

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