Agentic RAG-R1:让大模型从「检索助手」跃升为「思考+搜索王者」!

 

 

 

Agentic RAG-R1 是由北京大学研发的一项开源研究项目,旨在推动语言模型在自主检索与推理能力方面的能力边界。该项目通过引入强化学习策略(GRPO),构建了一个可自我规划、检索、推理与总结的智能体式 RAG 系统。

核心亮点

  1. 1. Agentic RAG 架构:融合检索增强生成(RAG)与 Agentic AI 机制,模型不仅生成答案,还能“决定如何生成答案”。
  2. 2. 强化学习优化(GRPO):借助 Generalized Relevance Policy Optimization,让模型学会更合理地选择检索和推理步骤。
  3. 3. 多轮推理与回溯能力:支持计划、回溯、总结等多种 agent 行为,实现人类式的问题解决流程。
  4. 4. LoRA 与量化支持:低成本微调与高效推理并存,轻松部署大模型至生产环境。
  5. 5. 丰富奖励机制:引入格式、准确性、RAG 表现等多个维度的奖励,训练出更“懂业务”的智能体。

Github项目地址: https://github.com/jiangxinke/Agentic-RAG-R1

“模型自主、工具自选、推理自洽”——Agentic RAG-R1 用强化学习把 RAG 带进智能体时代。


📚 背景:为什么 RAG 需要 “Agentic”?

  • • 事实性:RAG 通过外部检索解决 “幻觉” 问题,但仍依赖人工提示来决定何时检索。
  • • 上下文爆炸:检索结果越多,拼接进上下文越长,反而稀释关键信息。
  • • 多跳推理:复杂任务需要 “查-思-查-思” 循环,仅一次检索难以覆盖。

Agentic RAG-R1 让模型在每一步“思考”时都能自主决定:

  1. 1. 是否检索? —— 省掉无关调用,提高效率
  2. 2. 检索什么? —— 人类不再手写复杂 prompt
  3. 3. 如何引用? —— 自动将证据融入推理链

🏗️ 体系结构:全面的 Agentic 思考

✨ 核心理念:两大王牌技术的强强联合

– 🔍 检索增强生成 (RAG):在生成过程中即时从外部知识库检索信息,兼具语言模型的创造力与实时、可信的事实。

– 🤖 Agentic AI 智能体:让模型自主决定何时检索、检索什么,以及如何把检索证据编织进推理链,真正做到“会思考、会行动”。

🏗️ 架构:基于 TC-RAG 的智能体思考循环

目前支持如下动作:

#
动作
说明
状态
1
🤔 Reasoning(推理)
展开思考、提出假设
2
🔄 Backtrack(回溯)
回到上一节点,修正思路
3
📝 Summary(总结)
汇总已有证据,压缩上下文
4
🛠️ Tool Observation(工具调用)
访问 Wiki / 文档 / 知识图谱等
5
✅ Conclusion(结论)
输出最终答案

🔬 技术细节深挖

Features ✨

组件 关键点 优势
GRPO (Generalized Relevance Policy Optimization)
采样多条推理-检索轨迹,对“高相关、高准确、高格式”路径赋正奖励
训练稳定

、收敛快,避免 RLHF 里的 Reward Hacking
LoRA + NF4 量化
10 % 参数可训练,int-4 存储
GPU 省钱

,多实验迭代无压力
Deepspeed Zero-3
权重 & 优化器拆分到 CPU / NVMe
3×A100 → 32B

 轻松起飞
多模态工具接口
支持文本、代码、数据库、REST API
让模型在“真实工作流”里落地

奖励公式: ( 

其中 r_rag 由 RAGAS 自动评测检索片段是否被有效引用。

Rollout Generation 🔄


📊 结果:数据说话

数据集MedQA(中英双语) | Judge Model:Qwen-2.5-72B

设置 格式准确率 ↑ 答案准确率 ↑
微调前
39 %
84 %
微调前 + 检索
56 %
79 %
微调后 + 检索 92 % (+53 %) 87 % (+3 %)
  • • 跨语言:中/英两份测试集均显著提升
  • • 复杂推理:多跳问题正确率提升 8 % 以上
  • • 工具调用成功率:> 95 %,日志可追溯

实际测试结果:


💬 FAQ

Q1:必须用 32B 模型吗?

A1:不需要!我们默认用 Qwen-2.5-7B-Instruct;你也可以换成 Llama-3-8B / Baichuan-13B,只需改配置。

Q2:RL 训练很复杂吗?

A2:脚本参数与常规 LoRA 差不多,多加一份奖励配置即可。CPU 显存不足?Zero-3 + Offload 轻松搞定。


📢 结语 & 口号

“模型自主,检索在手;深度推理,靠谱出口!”

“让 LLM 会自己找资料,再也不用 Ctrl + C / Ctrl + V!”

 


(文:PaperAgent)

发表评论

×

下载每时AI手机APP

 

和大家一起交流AI最新资讯!

立即前往