← 返回 2026-05-12

DeepRefine:通过强化学习实现智能体编译知识库的精炼 DeepRefine: Agent-Compiled Knowledge Refinement via Reinforcement Learning

Haoyu Huang, Jiaxin Bai, Shujie Liu, Yang Wei, Hong Ting Tsang, Yisen Gao, Zhongwei Xie, Yufei Li, Yangqiu Song 📅 2026-05-11 👍 3 2026-07-13 08:36
GRPO LLM智能体 RAG 强化学习 知识图谱 知识库精炼

首个面向已构建RAG知识库的强化学习后精炼框架

前置知识

检索增强生成(RAG)

Retrieval-Augmented Generation 通过在生成时从外部知识库(如向量索引、文档集或图结构)检索相关信息,再将其作为上下文输入到大语言模型中,从而缓解参数化知识的时效性与幻觉问题。常见做法包括稠密向量检索(DPR/Embedding-based)和图结构检索(HippoRAG、ToG、Subgraph RAG)等。

DeepRefine 直接优化 RAG 系统的上游知识库质量,理解它的输入输出链路(query→retrieve→augment→generate)才能看懂论文为何要"精炼知识库"而非"换个更好的 retriever"。

Agent-Compiled Knowledge Base(智能体编译式知识库)

由具备工具调用能力的 LLM 智能体(如 LLM-Wiki、Mem0、Zep、Graphify)持续从多源文档中抽取、组织并维护的结构化知识库,通常以图三元组 $(h, r, t)$ 形式存储,节点可以是实体、事件或文档片段,关系涵盖分类、时间、因果、跨文档链接。

本文研究对象正是这类"可演化"知识库;理解智能体在构建过程中的不可控缺陷(不完整/错误/冗余)是动机章节的关键。

GRPO(Group Relative Policy Optimization)

一种无需 critic 网络的强化学习算法,由 Shao et al. (2024) 提出。它对同一 prompt 采样 $G$ 个输出组成一个组,使用组内奖励的均值 $\mu_R$ 与标准差 $\sigma_R$ 归一化得到 group-relative advantage $\hat{A}_{i,t}=\frac{R_i-\mu_R}{\sigma_R}$,并以 PPO 风格 clip 目标函数 $\min\big(r_{i,t}(\theta)\hat{A}_{i,t},\ \text{clip}(r_{i,t}(\theta),1-\epsilon,1+\epsilon)\hat{A}_{i,t}\big)$ 进行策略更新。

DeepRefine 用 GRPO 训练精炼策略,去掉 KL 项以降开销;公式 (9) 给出的目标函数是论文方法学核心。

GBD 奖励(Gain Beyond Draft)

作者设计的下游效用奖励信号:$\text{GBD}(q) = \text{ACC}(A_{\text{refined}}, A) - \text{ACC}(A_{\text{draft}}, A)$,即精炼后 RAG 答案相对原始知识库下 RAG 答案在任务指标(如 F1)上的差值,反映精炼动作带来的下游收益。

这是论文最核心的创新之一:缺乏 gold refinement actions 的情形下,如何把"精炼"转化为可学习的奖励信号;公式 (8) 即定义。

Subgraph Retrieval 与 HippoRAG

Subgraph RAG 通过稠密检索得到 top-$N$ 三元组作为 0-hop 子图,再沿实体边扩展;HippoRAG (Jimenez Gutierrez et al., 2024) 借鉴海马体记忆索引机制,将 LLM 抽取的概念节点与 Personalized PageRank 结合实现图增强检索;ToG (Tree of Graph) 则把图遍历过程交给 LLM 进行多跳推理。

论文在 5 个 OOD 数据集上把这 4 类 retriever(Subgraph、ToG、HippoRAG、HippoRAG2)作为评估目标,DeepRefine 需对它们均透明有效。

研究动机

现有 RAG 系统的外部知识库一旦构建完成便基本静态,无法从用户后续查询中累积经验。更糟的是,由 LLM 智能体自动编译的知识库(如 LLM-Wiki、Graphify)普遍存在三类系统化缺陷:不完整(缺失证据链或跨文档链接)、不正确(低置信或含糊断言)、冗余(指代消解与歧义节点)。这些缺陷在迭代使用中复合放大,既降低检索 fidelity,也损害最终任务性能。例如在 Graphify 构建的子图上,HippoRAG2 在 NQ 上仅得 13.62 F1、PopQA 仅 6.19,远低于 AR1 构造函数下的 37.25 与 64.69;LOCOMO 长程对话任务上整体 F1 也只有 11.89。先前的工作(Mem-α、Memory-R1、AutoGraph-R1)大多聚焦"构建策略"优化,但在已部署场景中,重训构造函数并全量重建知识库既昂贵又不现实,催生了"已构建知识库的后精炼"这一新范式的需求。

本文的目标是本文提出 DeepRefine,目标是打造一个通用、基于 LLM 推理的智能体知识库后精炼模型:对任意预构建知识库 $G_f$ 与一组用户查询 $Q=\{q_1,\ldots,q_N\}$,生成精炼动作序列,使下游 RAG 任务(QA / 长程对话)的生成准确率得到稳定提升,同时显著降低相对全量重建的计算开销。

与已有工作不同的是,已有工作要么优化"构建期"策略(无法部署后改进),要么是 KG 补全 / 实体对齐类方法(只覆盖单一操作、忽略冗余;TRAIL 类基于 LLM 的迭代更新也未用下游效用作为奖励)。DeepRefine 的独特切入角度在于:把"精炼"建模为可在没有 gold reference 的条件下、用 GBD 奖励 + GRPO 端到端学习的策略优化问题,并以多轮 query-conditioned 子图扩展做缺陷定位,从而兼顾有效性、可扩展性与部署友好性。

核心方法

DeepRefine 是一种以 LLM 为推理核心、以知识库 $G_f$ 为交互对象的智能体精炼框架。它遵循"先定位—再归因—后修复"的直觉流程:首先让模型对每个 query 与知识库做多轮稠密+扩展式子图检索,判断"是否已可回答";若不可回答,则从交互历史中归纳可能的缺陷类型;最后产出代码式的精炼动作(如 $\texttt{insert\_edge()}$、$\texttt{delete\_edge()}$、$\texttt{replace\_node()}$)就地编辑图谱。整个推理过程通过 GRPO 端到端优化,并引入 GBD 奖励衡量"精炼相对草稿"的下游增益,因此无需任何 gold 精炼动作监督。

与 AutoGraph-R1 等"重新生成全图"的策略、以及 Mem-α 等优化构造函数的方法不同,DeepRefine 的本质创新是"后精炼 + 下游效用奖励":它把精炼动作空间约束为三种面向图数据库的原语(insert/delete/replace),从而通用且低 token;并以 GBD $=\text{ACC}(\hat{G}_f)-\text{ACC}(G_f)$ 作为可微替代信号,让策略在没有参考动作的情形下也能从下游任务反馈中学到"哪些动作真的有用"。

方法步骤详情

整个方法由三个紧密耦合的推理步骤组成。第一步是 Answerability Judgement Loop:对 query $q$ 先做稠密 top-$N$ 检索得到 0-hop 子图 $G^{(0)}_q=\text{Top-}k(q,G_f,N)$,再用 LLM 判断可回答性;若不可回答则沿子图中实体做 1-hop 扩展,得到候选集 $G^{(i)}_{\text{cand}}=\{(h,r,t)\in G_f \mid h\text{ or }t\in E^{(i-1)}_q\}$,剪枝 $G^{(i)}_{\text{pruned}}=\text{Top-}k(q,G^{(i)}_{\text{cand}},M)$,并入历史 $G^{(i)}_q=G^{(i-1)}_q\cup G^{(i)}_{\text{pruned}}$;只保留最近 $L_h$ 步的交互历史 $H_q[-L_h:]$ 以满足上下文限制。第二步是 Error Abduction:基于 $H_q[-L_h:]$,模型从 incompleteness / incorrectness / redundancy 三个视角归纳潜在缺陷 $I_q$,输出被 $\langle\text{abduction}\rangle\cdots\langle/\text{abduction}\rangle$ 包裹。第三步是 Refinement Actions Generation:在 $\langle\text{refinement}\rangle$ 标签内生成代码式动作序列 $A_q=\arg\max_{A\in\mathcal{A}} p_\theta(A\mid G^{(L)}_q,I_q)$,三类算子分别对应补全缺失边、删除错误/冗余边、替换歧义节点,可直接对接 NetworkX 等图数据库接口。训练阶段把这些推理轨迹作为 GRPO rollout,按公式 $\text{GBD}(q)=\text{ACC}(A_{\text{refined}},A)-\text{ACC}(A_{\text{draft}},A)$ 计算奖励,并以简化版目标(去掉 KL 项) $J_{\text{GRPO}}(\theta)=\mathbb{E}[\frac{1}{G}\sum_{i=1}^G\frac{1}{|o_i|}\sum_{t=1}^{|o_i|}\min(r_{i,t}(\theta)\hat{A}_{i,t},\text{clip}(r_{i,t}(\theta),1-\epsilon,1+\epsilon)\hat{A}_{i,t})]$ 更新策略 $\pi_\theta$。推理时 DeepRefine 冻结,可异步挂在用户 query 流与 KB 演化流之上,不引入在线服务延迟。

技术新颖性

技术新颖性体现在三方面:其一,**精炼算子抽象**——首次把后精炼操作收敛到与图数据库原语对齐的三种接口,兼顾通用性与可执行性;其二,**无参考学习范式**——GBD 奖励巧妙地把"知识库质量"映射到"下游 RAG 准确率差值",回避了对 gold refinement 轨迹的需求,填补 TRAIL 等方法只做无奖励探索的空白;其三,**交互式缺陷定位**——用多轮子图扩展代替全图遍历,将搜索复杂度限制在与 query 相关的邻域,且通过 $L_h$ 滚动窗口维持上下文可行。整体上把"构造端 RL"(AutoGraph-R1)与"使用端补全"(KGRefiner)的能力合二为一。

Demonstration of GRPO training with agent-compiled knowledge refinement (DeepRefine), where the orange arrows denote the multi-turn interactions of DeepRefine with knowledge bases and the green arrows denote the refinement actions on knowledge bases.
Figure 2: Demonstration of GRPO training with agent-compiled knowledge refinement (DeepRefine), where the orange arrows denote the multi-turn interactions of DeepRefine with knowledge bases and the green arrows denote the refinement actions on knowledge bases.
An overview of the inference process of DeepRefine.
Figure 3: An overview of the inference process of DeepRefine.

实验结果

在 5 个 OOD 数据集(NQ、PopQA、2WikiMultihopQA、Musique、LOCOMO)×4 种 retriever(Subgraph、ToG、HippoRAG、HippoRAG2)×3 种构造函数(Naive、AR1、Graphify)的全因子评测中,DeepRefine-8B 在绝大多数组合上稳定提升 RAG F1。在 AR1 构造函数上,HippoRAG2 + DeepRefine-8B 取得 45.27 的总体均值 F1,相对未精炼的 43.65 提升 +1.62,其中 Musique 单项从 27.96 提升到 30.30(+2.34),LOCOMO 从 35.75 提升到 36.75(+1.00);同构造下 HippoRAG + DeepRefine-8B 整体均值由 43.17 升至 44.11,2WikiQA 单项从 50.34 跃升至 52.75。在 Naive 构造函数上精炼后甚至能反超 AR1,例如 HippoRAG2 + DeepRefine-8B(42.24)已接近 AR1 单独的水平,Musique 上 28.27 显著高于 AR1 + HippoRAG 的 26.21,验证"针对关键区域的精炼胜过全图重建"。Graphify(LLM-Wiki)构造函数原始分数较低,但 DeepRefine-8B 仍带来大幅相对提升,例如 HippoRAG2 整体均值由 11.89 升至 13.18(+1.29),PopQA 由 6.19 升至 6.69。效率方面(Table 2),DeepRefine 在三类任务上分别为 3201.7s / 3357.5s / 1115.8s,比 AutoGraph-R1 的 6782.8s / 9780.4s / 2826.5s 快约 2–3 倍,主要受益于"只改相关区域 + 代码式动作 + 最大覆盖 query 子集"。消融实验(Table 3)显示,移除 RL 训练的 DeepRefine-8B w/o RL 在多数组合上仅得边际改善甚至退化,例如 AR1 + HippoRAG2 w/o RL 为 44.49,弱于完整训练的 45.27;Graphify + HippoRAG w/o RL(11.91)反而低于原始 11.99。这印证了 GBD 奖励 + GRPO 是性能增益的关键。Figure 4 的训练曲线显示 4B 与 8B 模型的平均 GBD 奖励均随训练步数稳定上升,8B 收敛值高于 4B,与论文"更大模型带来更高精炼效用"的观察一致。

Performance of our GBD reward finetuned DeepRefine with Qwen3-8B on various knowledge retrievers.
Table 1: Performance of our GBD reward finetuned DeepRefine with Qwen3-8B on various knowledge retrievers.
The average time consumptions (s) of the knowledge refinement process of DeepRefine and the reconstruction process of AutoGraph-R1.
Table 2: The average time consumptions (s) of the knowledge refinement process of DeepRefine and the reconstruction process of AutoGraph-R1.
Performance comparisons between GBD reward finetuned DeepRefine and the DeepRefine without RL training under AR1 and Graphify (LLM-Wiki) constructors.
Table 3: Performance comparisons between GBD reward finetuned DeepRefine and the DeepRefine without RL training under AR1 and Graphify (LLM-Wiki) constructors.
The average GBD reward curves during the end-to-end RL training process of both Qwen3-4B-Instruct-2507en (left) and Qwen3-8B (right).
Figure 4: The average GBD reward curves during the end-to-end RL training process of both Qwen3-4B-Instruct-2507en (left) and Qwen3-8B (right).
查看结构化数据
任务指标本文基线提升
AR1 构造函数下 HippoRAG2 总体均值(5 个 OOD 数据集) RAG F1(%) HippoRAG2 + DeepRefine-8B = 45.27 HippoRAG2(AR1,未精炼)= 43.65 +1.62 绝对提升(约 +3.7%)
AR1 构造函数下 HippoRAG 总体均值 RAG F1(%) HippoRAG + DeepRefine-8B = 44.11 HippoRAG(AR1)= 43.17 +0.94 绝对提升
Naive 构造函数下 HippoRAG2 总体均值 RAG F1(%) HippoRAG2 + DeepRefine-8B = 42.24 HippoRAG2(Naive)= 41.82 +0.42 绝对提升;其中 Musique 单项 28.27 vs 26.21(+2.06)
Graphify(LLM-Wiki)构造函数下 HippoRAG2 总体均值 RAG F1(%) HippoRAG2 + DeepRefine-8B = 13.18 HippoRAG2(Graphify)= 11.89 +1.29 绝对提升(约 +10.8% 相对提升)
Simple QA 任务上精炼与重建的耗时对比 平均时间消耗(秒) DeepRefine = 3201.7s AutoGraph-R1 = 6782.8s 快 2.12×
Multi-hop QA 任务上精炼与重建的耗时对比 平均时间消耗(秒) DeepRefine = 3357.5s AutoGraph-R1 = 9780.4s 快 2.91×
Conversation QA(LOCOMO)上精炼与重建的耗时对比 平均时间消耗(秒) DeepRefine = 1115.8s AutoGraph-R1 = 2826.5s 快 2.53×
RL 训练消融(AR1 + HippoRAG2 总体均值) RAG F1(%) DeepRefine-8B(含 RL)= 45.27 DeepRefine-8B w/o RL = 44.49(原始 43.65) RL 训练相对 w/o RL 再提升 +0.78;w/o RL 已含额外推理增益 +0.84

局限与改进

作者承认的方法局限包括:1) 性能并非在所有 retriever × 构造函数组合上单调提升,例如 Naive + ToG + DeepRefine-4B 在 2WikiQA 上从 43.74 降到 42.88,Musique 上从 18.21 降到 16.91,说明小模型在精炼算子选择上不够稳定;2) 精炼效果高度依赖 retriever 与 KB 结构质量,Graphify 整体仍处于低分区间(最佳仅 16.35),提示模型难以"无中生有"修复严重缺失的结构性内容;3) 评测仅覆盖英文 5 个 OOD 任务,对中文及多模态知识库未做验证。基于自己的观察还可指出:a) 训练数据只来自 HotpotQA 一个源(5000 样本),策略可能在该数据集的多跳结构上有偏;b) 三种算子的表达力有限,对复杂模式(如 N 元关系、属性边、时间窗口约束)覆盖不足;c) GRPO 去 KL 项的简化虽省显存,但可能放大训练—推理分布漂移;d) 推理时需要多轮 LLM 调用,Latency 虽"异步",但实时性仍受限。

独立分析的弱点

独立审视的弱点与改进方向:(1) **缺陷定位的粒度仍偏粗**:当前只在 query-conditioned 子图上做修补,对跨 query 的全局冗余(如全图多处重复实体)识别困难,可考虑引入全局聚类或周期性 schema audit 步骤。(2) **GBD 奖励的方差问题**:单条 query 的下游 ACC 波动可能让 GRPO advantage 估计不稳定,建议加入批量 baseline 或使用更长 horizon 的 discounted reward 来平滑信号。(3) **动作空间有限**:仅 insert_edge/delete_edge/replace_node 三类,无法表达属性修改或条件删除,可扩展为类 Cypher 的受限 DSL 或引入 Program-of-Thought 风格的复合算子。(4) **冷启动依赖**:训练数据仅来自 HotpotQA,迁移到 GraphRAG、LightRAG 等不同 schema 的知识库时可能效果下降,可通过 prompt-level schema conditioning 或继续预训练缓解。(5) **缺乏人工审计接口**:尽管动作以代码形式产出,但未提供回滚与 diff 可视化工具,部署时若产生级联错误难以及时修复;建议结合 DBMS 事务或 git 式版本管理。

未来方向

作者在结论与附录中暗示的方向包括:(a) 把精炼范式推广到多模态 KB(含图像、表格、公式节点),并对非文本三元组重新设计 embedding;(b) 引入人类反馈(RLHF / RLAIF)做精炼动作的偏好对齐,缓解 GBD 奖励的过优化风险;(c) 把 DeepRefine 与 AutoGraph-R1 等构造函数联合训练,做"构建+精炼"端到端优化。基于结果可进一步延伸的方向:(d) **在线持续学习**——把每天新增的用户 query 流作为增量训练集,使策略随真实使用模式演化;(e) **跨域迁移**——评估在医疗、法律等结构化 KB 上 zero-shot 与 few-shot 精炼效果;(f) **反事实精炼评估**——构建包含可控缺陷注入的合成 KB(如随机删边 / 注入冲突断言),定量衡量 recall @ fix;(g) **与图数据库事务结合**——把每次精炼视为一次 ACID 事务,支持回放与冲突检测;(h) **压缩友好推理**——探索 4B 模型 + 知识蒸馏 / Q-LoRA 路径,降低部署门槛。

复现评估

复现性总体可控但门槛较高:作者使用 Qwen3-8B 与 Qwen3-4B-Instruct-2507en 作为 backbone(公开可下载),Qwen3-Embedding-0.6B 作 embedding、Qwen2.5-7B-Instruct 作冻结 reader,工具链与权重均可获取。训练数据是从 HotpotQA 训练集随机抽样 5000 条按 8:2 划分得到,但论文未直接公布此子集与具体的 prompt 模板(仅在 Appendix E 给出部分 tag 格式示例),需读者自行复现数据切分。GRPO 超参明确:8B 模型 batch size 64、mini-batch 16、group size 6、30 步、lr $1\times10^{-7}$;4B 模型 batch size 32、mini-batch 16、group size 6、60 步、lr $5\times10^{-7}$。最关键的复现成本是算力:8× NVIDIA A100-SXM4-80GB GPU,完整训练涉及多轮 LLM rollout 与下游 RAG 评估,单次实验耗资较高。代码与模型权重在论文首页给出的 Project / Models 链接中(未在文本中展开),但未声明开源许可证;评测涉及 5 个英文 OOD 数据集(NQ、PopQA、2WikiMultihopQA、Musique、LOCOMO),均可公开下载,整体复现难度为中等偏高。