DeepSeek提出的条件记忆模块并非外界误解的外挂记忆,而是一种架构优化。该模块通过查表方式存储静态实体关系,使Transformer参数从重建知识的任务中解放,转而专注于推理,显著提升了代码生成与长上下文处理能力。
智能速览
该模块本质是预训练关系的查表系统,非外部记忆单元
通过并行处理保持推理速度,不增加计算延迟
在代码和数学推理任务中表现优于传统MoE架构
长上下文检索能力大幅提升,准确率从84.2%增至97%
模块参数与总参数的比值在70%-80%时效果最佳
精华内容
探究DeepSeek Engram模块如何通过分离静态知识与动态推理,在不牺牲推理速度的前提下提升模型智能水平。
澄清记忆概念
该模块常被误读为给大模型增加了无限外部记忆。实际上,它是一个存储token关系范式的数据库。模型在预训练阶段将“太阳从东边升起”等高度刻板的静态关系存储其中。推理时直接调用,无需重建,从而节省了用于识别实体关系的参数,让Transformer的核心优势回归到逻辑推理上。
技术实现机制
Engram模块在预训练时同步学习2-gram或3-gram等词语关联习惯。当输入Token进入Transformer Block时,并行从Engram中提取关键实体嵌入。通过词源压缩和哈希匹配,不仅降低了模型大小,还通过上下文门控机制缓解了语义冲突,确保检索内容的高准确率。
推理效率保障
引入新模块通常担心速度下降,但Engram采用O(1)时间复杂度的检索方式,与模型计算并行处理。这意味着模型检索速度与参数大小无关,且该模块可部署在CPU上。实测显示,推理时间未增加,成本反而因GPU算力释放而降低。
性能实测数据
在DeepSeek-MoE 27B和40B模型上的实验表明,Engram架构全面优于传统MoE。知识型任务表现符合预期,令人意外的是代码和数学等推理任务也大幅提升。长上下文检索能力从84.2%跃升至97%,证明其在处理长文本时具有更强的信息捕捉能力。
架构参数配比
研究发现,Engram模块参数与模型总参数的比值存在最优区间。当比值在70%至80%时,Loss曲线表现最佳且持续下降。这种架构设计等同于增加了模型深度,且具有规模效应,能够平滑扩展至大模型而不损失性能。
DeepSeek Engram模块通过架构级的创新,成功将静态知识存储与动态推理分离。虽然未直接节省算力,但显著提升了模型在代码、数学及长文本上的表现,这种“术业有专攻”的设计思路为大模型进化提供了新方向。