LLM 记忆架构三维分类体系(清华《Memory for Large Language Models》综述)¶
Ch01.729 LLM 记忆架构三维分类体系(清华《Memory for Large Language Models》综述)¶
📊 Level ⭐⭐ | 8.5KB |
entities/memory-for-llms-tsinghua-taxonomy-2026.md
LLM 记忆架构三维分类体系¶
清华大学唐杰团队 + NUS + Bosch AI 的综述《Memory for Large Language Models》(20 页 pre-print,用户提供论文原文)提出架构中心的 LLM 记忆三维 taxonomy:representation(隐式 vs 显式)× update dynamics(离线 vs 在线)× persistence(短期 vs 长期),并形式化记忆写入、路由、状态转移与整合的细粒度机制。
核心定位与 Scope¶
Memory 已从计算的隐式副产品转变为一等架构设计维度:Titans/端到端 TTT 支持部署期动态参数更新,Engram 等 lookup 架构解耦存储与稠密计算,Nested 多时间尺度更新模糊训练/推理边界,MoE 构成条件参数记忆。
关键 Scope 限定:本文聚焦 model-level 记忆机制(在模型架构或推理时动态中实例化的记忆),排除 agent-level 或 prompt-based 记忆系统——与库内 agent 级记忆实体(四学派/six 存储学派/系统设计)构成互补维度。Taxonomy 不是刚性分类,而是揭示貌似不同方法间共同结构的概念框架。
三维分类框架¶
| 轴 | 判定问题 | 分类 |
|---|---|---|
| Representation(表征形式) | 记忆是否 endowed with 独立、可控的存储/检索接口 | 隐式(计算副产品,无 read/write/lookup 接口)vs 显式(独立存储组件,清晰 read/write 语义 + 可控更新策略) |
| Update Dynamics(更新机制) | 记忆何时能被修改 | 离线(仅预训练梯度更新)vs 在线(推理时实时更新) |
| Persistence(持久性) | 信息有效作用跨度 | 短期(单会话/局部窗口)vs 长期(跨文档跨对话留存) |
Explicit 判定的关键洞见:显式记忆不限于非参数结构——参数化模块也可构成显式记忆。区分标准不是参数化形式,而是是否被设计并作为记忆运行(清晰 read/write 语义 + 可控更新策略)。注意即使 KV cache 物化为显式张量,其访问语义由架构固定、缺乏独立寻址/更新控制——仍是隐式记忆。
隐式记忆(Computation Dynamics)¶
- 注意力类(KV Cache):标准 Transformer、StreamingLLM、Longformer——与原生计算无缝融合,存储随序列线性上涨、会话结束销毁。
- 稀疏/选择性/结构化:MoBA、NSA、RATTENTION——不改变 KV 存储本质,仅路由/门控筛选 token,属访问优化而非新增存储模块。
- 循环序列记忆(SSM/线性注意力):Mamba、RWKV、Kalman 线性注意力、Gated DeltaNet——全部历史压缩为固定尺寸隐藏状态,O(n) 线性复杂度;短板是压缩丢失细粒度细节、长期状态干扰。
- 共同局限:容量受硬件/窗口限制、无自主读写控制、无法跨会话留存,不适合长期个性化 Agent 场景。
显式记忆(Addressable and Persistent)¶
- 参数化外部记忆(TTT 路线):Titans、TTT-E2E、In-Place TTT、MEMORYLLM——骨干之外新增可更新参数,推理时梯度/误差信号实时更新,冻结主干规避灾难性遗忘。
- 查表/检索导向:kNN-LM、Engram、PlugLM、ExplicitLM——哈希槽/可编辑向量库/独立 KV 池,容量无限扩容、与主干算力解耦,RAG 与原生模型融合桥梁。
- 条件参数记忆(MoE):Switch Transformer、Mixtral、DeepSeek-MoE——专家子网存领域知识、路由选择性激活;仅在训练阶段更新(离线记忆)。
- 多时间尺度嵌套更新:Nested Learning——快慢两套参数更新周期,平衡可塑性(学新)与稳定性(不忘旧)。
5 种细粒度更新规则(Table II,不互斥可组合)¶
- Optimization-based writing(TTT/Titans 梯度目标更新)——风险:漂移、不稳定、目标不匹配
- State-transition updates(Mamba/Gated DeltaNet/Kalman 递推)——风险:压缩损失、状态干扰
- Signal-gated writing/routing(AMOR/HAM 熵/预测误差门控)——风险:校准、阈值敏感、延迟相关性
- Admission/eviction/consolidation(StreamingLLM/RATTENTION 缓存淘汰压缩)——风险:不可逆损失、检索偏差
- Objective-induced/structural updates(Engram/MoE 离线架构固定)——风险:刚性、目标不匹配、测试时适应性有限
此细化不是第四轴,而是暴露"更新机会出现后记忆如何变化"的机制。
混合记忆架构(下一代主流)¶
- 层间交替注意力+SSM:Samba、Jamba/Jamba-1.5、Kimi Linear(3:1 配比)、LightTransfer(层替换免重训)——静态固定比例、无动态调度。
- 自适应混合记忆路由:AMOR(熵超阈值才启动注意力精炼)、HAM(预测误差大才写入高保真 KV)——算力随输入自适应,但门控阈值靠人工调参。
- 隐式短时+显式长期双体系:Titans、Hydra——底层 KV 管本轮上下文、顶层独立参数/哈希存储池跨轮沉淀,实现计算与存储解耦。
- 多时间尺度时序混合:TTT-E2E 批量更新 vs Nested Learning 多嵌套循环——记忆层级(token 级/交互批次级/主干冻结)。
- 多组件模块化设计:Hydra(状态空间主干+稀疏全局注意力+MoE 路由+双工作区+事实记忆)、Expansion Span(跨度扩展注意力)。
显式记忆的结构风险(解读版缺失)¶
- Capacity Growth:显式记忆容量可扩展但收益递减,需索引/剪枝/压缩。
- Interference and Memory Drift:持久在线存储新旧共存,无规约则干扰/覆盖/放大噪声;stability-plasticity tradeoff 仍是核心未决张力。
- Optimization and Convergence:额外目标/更新循环不共享标准预训练优化保证,频繁测试时更新累积偏差。
评测方法论(四维)¶
- 长上下文检索:NIAH(distractor 找目标事实)、LongBench、RULER/L-Eval(退化曲线,揭示 attention dilution 与 recency bias)
- 结构化依赖与推理:SCROLLS、NarrativeQA——跨长文档整合信息;混合/SSM 模型更稳定但牺牲细粒度 token 召回
- 遗忘/干扰/稳定性:显式在线记忆的持续更新后遗忘程度
- 效率权衡:同召回精度下显存/延迟,区分"真记忆提升"与"单纯扩大上下文窗口"
六大开放挑战¶
统一 LLM 记忆理论(缺通用数学抽象量化容量/压缩损耗/干扰)→ 终身可塑参数记忆(无遗忘持续学习)→ 可解释可控更新规则(可逆可诊断写入)→ 自适应动态记忆调度(智能控制器分配注意力/循环/检索资源)→ 软硬件协同设计(芯片/加速器适配分层记忆)→ 标准化多维评测框架(容量/保真度/持久度/效率统一标准)。
与库内实体关系¶
- 与 三种 Agent Memory 方案对比实验(MSA/D2L/RAG 实证)互补:那是 agent 级记忆方案选型,本文是 model-level 架构分类
- 与 Agent 记忆系统四学派 / 六存储学派 维度不同:库内为产品/存储级,本文为架构级三维元分类
- 与 上下文工程 的记忆维度(L4 记忆系统)衔接:本文提供统一坐标
→ 论文原文存档