编者摘要:本文提出HGMEM(基于超图的记忆机制),旨在解决多步 RAG 在长上下文复杂关系建模中的局限性 —— 现有记忆机制仅被动存储孤立事实,忽视高阶关联导致推理碎片化;HGMEM 将记忆建模为超图(超边作为独立记忆单元),通过更新、插入、合并动态演化,支持高阶 n 元关系建模,并结合自适应证据检索策略(局部调查 + 全局探索)引导子查询生成与证据获取;实验验证显示,HGMEM 在 Longbench、NarrativeQA 等多个基准测试中表现优异,如 Longbench 综合得分65.73%、NoCha 准确率73.81%,持续超越传统 RAG 与多步 RAG 基线,为长上下文全局理解与复杂推理提供有效支撑。
一、引言
- 问题提出
非结构化记忆(纯文本摘要):无法回溯引用,步骤间操控精度低;
结构化记忆(知识图谱 / 表格):仅支持二元关系,难以建模高阶关联;
核心缺口:缺乏动态演化能力,无法整合孤立事实形成高阶知识,导致长上下文推理碎片化、全局理解弱。
单步 RAG 难以处理长上下文复杂查询,多步 RAG 成为主流,但现有记忆机制存在局限:
- 贡献
- 提出HGMEM:将记忆建模为超图,支持高阶 n 元关系(n>2)动态演化;
- 设计自适应证据检索策略:结合局部调查(深入现有记忆单元)与全局探索(拓展未覆盖领域);
- 多基准验证:在长上下文全局理解任务中持续超越强基线。
二、相关工作
| 研究方向 | 核心局限 |
|---|---|
| 多步 RAG 工作记忆 | 图谱增强方法仅支持二元关系,无法建模高阶关联(如 Liu et al., 2024; Li et al., 2025a) |
| 结构化知识索引 | 静态存储(离线构建),无动态演化能力,无法适配多步推理中的知识更新(如 GraphRAG、HypergraphRAG) |
三、方法论
3.1 问题定义
- 输入:长文档 D(分割为文本块)、文档衍生图 G(实体 + 二元关系)、目标查询(\hat{q});
- 输出:基于多步检索 - 推理的最终响应;
- 核心目标:通过动态超图记忆,整合高阶关联,引导精准检索与推理。
3.2 HGMEM 核心设计
- 超图记忆存储
- 结构:(\mathcal{M}=(\mathcal{V}_{\mathcal{M}}, \tilde{\mathcal{E}}_{\mathcal{M}})),(\mathcal{V}_{\mathcal{M}})(实体顶点,子集于 G 的节点),(\tilde{\mathcal{E}}_{\mathcal{M}})(超边 = 记忆单元,含关系描述 + 关联顶点);
- 优势:超边可关联任意数量顶点,天然支持高阶关系建模。
- 自适应证据检索策略
- 局部调查:针对现有超边记忆单元,以关联顶点为锚点,在图 G 邻域检索补充信息;
- 全局探索:针对未覆盖领域,在 G 的非记忆节点集中检索新实体 / 关系;
- 核心价值:平衡深度挖掘与广度拓展,避免检索冗余或遗漏。
- 记忆动态演化(三大操作)
- 更新:修改现有超边的关系描述,不改变关联顶点;
- 插入:将新检索到的高阶关联作为新超边加入记忆;
- 合并:将语义 / 逻辑连贯的多个超边整合为单个高阶记忆单元,强化高阶关联建模。
- 记忆增强响应生成
- 触发条件:达到最大交互步数或记忆足够支撑响应;
- 输入:超边记忆单元描述 + 关联文本块;
- 输出:整合高阶关联的结构化响应。
四、实验设计
数据集(关键数字)| 数据集 | 文档数量 | 平均 tokens / 文档 | 查询数量 | 任务类型 Longbench(金融) | 20 | 266k | 100 | 生成式全局理解 QA
|| Longbench(政府) | 22 | 256k | 98 | 生成式全局理解 QA
|| Longbench(法律) | 7 | 194k | 55 | 生成式全局理解 QA
|| NarrativeQA | 10 | 218k | 100 | 长叙事理解
|| NoCha | 4 | 139k | 126 | 长叙事理解(事实判别)
|| Prelude | 5 | 280k | 135 | 长叙事理解(一致性判断)|
基线方法
- 传统 RAG:NaiveRAG、GraphRAG、LightRAG、HippoRAG v2;
- 多步 RAG:DeepRAG、ComoRAG(均含工作记忆);
- 控制条件:HGMEM 变体(仅局部调查、仅全局探索、移除更新 / 合并操作)。
- 评估指标
- 生成式 QA:综合覆盖率(0-100)、多样性(0-100);
- 长叙事理解:准确率(Acc)。
五、实验结果与分析
整体性能(GPT-4o 作为骨干模型)| 方法 | Longbench 综合覆盖率 | Longbench 多样性 | NarrativeQA Acc (%) | NoCha Acc (%) | Prelude Acc (%)
NaiveRAG | 61.62 | 64.20 | 52.00 | 67.46 | 60.00
|| GraphRAG | 60.39 | 64.02 | 53.00 | 70.63 | 59.26
|| DeepRAG | 63.62 | 65.98 | 45.00 | 67.46 | 36.30
|| ComoRAG | 62.18 | 65.82 | 54.00 | 63.49 | 54.07
||HGMEM|65.73|69.74|55.00|73.81|62.96|
消融实验(Qwen2.5-32B-Instruct)| 变体类型 | 方法 | Longbench 综合覆盖率 | NoCha Acc (%) | Prelude Acc (%)
|| 检索策略 | HGMEM(自适应) | 64.18 | 70.63 | 62.22
|| | 仅局部调查 | 61.38 | 63.49 | 60.00
|| | 仅全局探索 | 59.25 | 68.25 | 59.26
|| 记忆演化操作 | HGMEM(完整) | 64.18 | 70.63 | 62.22
|| | 移除更新 | 62.48 | 68.25 | 60.00
|| | 移除合并 | 61.76 | 61.11 | 57.78 |
关键发现
- 记忆演化最优步数:t=3 时性能达峰,多步数无增益且增加成本;
- 查询类型适配:推理型查询中,HGMEM 超边平均实体数(Avg-Nv=7.07)显著高于移除合并的变体(4.10),准确率提升 10%;
- 成本可控:HGMEM 的平均 token 消耗(4436.43)与推理延迟(15.84s)与 DeepRAG、ComoRAG 处于同一水平。
六. 关键问题Q&A
问题 1:HGMEM 的核心创新是什么?如何解决长上下文复杂关系建模的核心痛点?
- 答:核心创新是将记忆建模为动态超图结构,并设计配套的演化机制与检索策略。针对长上下文复杂关系建模的核心痛点(现有记忆静态化、仅支持二元关系、推理碎片化),HGMEM 通过三方面解决:① 超图结构天然支持高阶 n 元关系(n>2),突破传统图谱的二元关系局限;② 记忆动态演化(更新 + 插入 + 合并),从孤立事实逐步构建整合的高阶知识结构;③ 自适应检索策略(局部调查 + 全局探索),精准引导子查询生成,平衡深度挖掘与广度拓展,避免检索冗余或关键信息遗漏,最终提升长上下文全局理解与复杂推理能力。
问题 2:HGMEM 与现有多步 RAG 的记忆机制(如图谱增强记忆)的关键区别是什么?
- 答:关键区别集中在三方面:① 关系建模能力:现有图谱增强记忆仅支持二元关系,无法捕捉多实体间的高阶关联;HGMEM 以超边为记忆单元,可关联任意数量实体,天然适配复杂 n 元关系;② 记忆动态性:现有机制多为静态存储(离线构建后固定),仅积累事实不演化;HGMEM 通过合并操作主动构建高阶关联,记忆随多步推理动态优化;③ 检索引导方式:现有方法多为单一检索模式(如全局检索),HGMEM 结合局部调查(深入现有记忆单元)与全局探索(拓展未覆盖领域),检索更具针对性。实验显示,仅移除合并操作后,NoCha 准确率下降 9.72%,验证高阶关联建模的关键价值。
问题 3:实验如何验证 HGMEM 的有效性?核心结果有哪些?
答:实验通过 “多基准覆盖 + 消融实验 + 查询类型分析” 三重验证 HGMEM 的有效性:① 多基准验证:在 Longbench(长文档 QA)、NarrativeQA 等 4 类长上下文任务中,HGMEM 持续超越传统 RAG 与多步 RAG 基线,如 Longbench 综合覆盖率达 65.73%(超 NaiveRAG 4.11 个百分点)、NoCha 准确率 73.81%(超 GraphRAG 3.18 个百分点);② 消融实验:验证自适应检索策略(双模式协同优于单一模式,性能提升≥2.96%)与合并操作(核心贡献,移除后性能下降≥3.33%)的必要性;③ 查询类型分析:在推理型查询中,HGMEM 超边平均实体数达 7.07,准确率比移除合并的变体高 10%,证明其在复杂关系建模中的优势。核心结果表明,HGMEM 在不显著增加成本的前提下,有效提升长上下文复杂推理的准确性与全面性。
## 如何学习大模型 AI ?
由于新岗位的生产效率,要优于被取代岗位的生产效率,所以实际上整个社会的生产效率是提升的。
但是具体到个人,只能说是:
“最先掌握AI的人,将会比较晚掌握AI的人有竞争优势”。
这句话,放在计算机、互联网、移动互联网的开局时期,都是一样的道理。
我在一线互联网企业工作十余年里,指导过不少同行后辈。帮助很多人得到了学习和成长。
我意识到有很多经验和知识值得分享给大家,也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑,所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限,很多互联网行业朋友无法获得正确的资料得到学习提升,故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。
第一阶段(10天):初阶应用
该阶段让大家对大模型 AI有一个最前沿的认识,对大模型 AI 的理解超过 95% 的人,可以在相关讨论时发表高级、不跟风、又接地气的见解,别人只会和 AI 聊天,而你能调教 AI,并能用代码将大模型和业务衔接。
- 大模型 AI 能干什么?
- 大模型是怎样获得「智能」的?
- 用好 AI 的核心心法
- 大模型应用业务架构
- 大模型应用技术架构
- 代码示例:向 GPT-3.5 灌入新知识
- 提示工程的意义和核心思想
- Prompt 典型构成
- 指令调优方法论
- 思维链和思维树
- Prompt 攻击和防范
- …
第二阶段(30天):高阶应用
该阶段我们正式进入大模型 AI 进阶实战学习,学会构造私有知识库,扩展 AI 的能力。快速开发一个完整的基于 agent 对话机器人。掌握功能最强的大模型开发框架,抓住最新的技术进展,适合 Python 和 JavaScript 程序员。
- 为什么要做 RAG
- 搭建一个简单的 ChatPDF
- 检索的基础概念
- 什么是向量表示(Embeddings)
- 向量数据库与向量检索
- 基于向量检索的 RAG
- 搭建 RAG 系统的扩展知识
- 混合检索与 RAG-Fusion 简介
- 向量模型本地部署
- …
第三阶段(30天):模型训练
恭喜你,如果学到这里,你基本可以找到一份大模型 AI相关的工作,自己也能训练 GPT 了!通过微调,训练自己的垂直大模型,能独立训练开源多模态大模型,掌握更多技术方案。
到此为止,大概2个月的时间。你已经成为了一名“AI小子”。那么你还想往下探索吗?
- 为什么要做 RAG
- 什么是模型
- 什么是模型训练
- 求解器 & 损失函数简介
- 小实验2:手写一个简单的神经网络并训练它
- 什么是训练/预训练/微调/轻量化微调
- Transformer结构简介
- 轻量化微调
- 实验数据集的构建
- …
第四阶段(20天):商业闭环
对全球大模型从性能、吞吐量、成本等方面有一定的认知,可以在云端和本地等多种环境下部署大模型,找到适合自己的项目/创业方向,做一名被 AI 武装的产品经理。
- 硬件选型
- 带你了解全球大模型
- 使用国产大模型服务
- 搭建 OpenAI 代理
- 热身:基于阿里云 PAI 部署 Stable Diffusion
- 在本地计算机运行大模型
- 大模型的私有化部署
- 基于 vLLM 部署大模型
- 案例:如何优雅地在阿里云私有部署开源大模型
- 部署一套开源 LLM 项目
- 内容安全
- 互联网信息服务算法备案
- …
学习是一个过程,只要学习就会有挑战。天道酬勤,你越努力,就会成为越优秀的自己。
如果你能在15天内完成所有的任务,那你堪称天才。然而,如果你能完成 60-70% 的内容,你就已经开始具备成为一名大模型 AI 的正确特征了。