张大妈

DeepSeek新作:Engram架构,用

源自今日头条:柠檬AI摄影

01-22 20:02

面对大模型发展中普遍存在的“堆参数”困境,DeepSeek提出的Engram架构提供了一种全新解法。它通过创新的“条件记忆”机制,将静态知识检索与动态推理分离,不仅显著降低了训练成本,更开辟了一条独立于计算之外的模型扩展新路径,为突破Transformer性能瓶颈带来了革命性思路。

DeepSeek新作:Engram架构,用智能速览

  • Engram引入“条件记忆”机制,分离静态知识与动态推理。

  • 实现训练计算量减少18%,性能不降反升。

  • 通过增大记忆容量可持续提升性能,无需增加计算量。

  • 在长上下文任务中表现卓越,RULER基准显著超越基线。

  • 为大模型提供了“记忆换容量”的高效扩展新路径。

DeepSeek新作:Engram架构,用精华内容

传统Transformer用昂贵的计算来模拟记忆,效率低下。Engram架构创新性地引入了“条件记忆”,将静态知识检索与动态推理分离开来,为突破性能瓶颈提供了全新思路。

何为条件记忆

Engram的核心思想是为Transformer装上一个“外接记忆库”,专门处理静态知识,如人名、地名、固定表述和专业术语。传统模型处理这类信息时,需要在每一层重新计算语义关系,效率极低,如同每次都重新计算1+1=2。

Engram架构则通过O(1)时间复杂度的稀疏查找操作,直接从巨大的静态记忆表中获取信息。这好比将一个纯计算器升级为计算器加查表器:简单问题直接查表,复杂问题才启动计算,实现了明确的分工,效率最大化。这种设计让模型可以专注于动态推理,而将事实性记忆交给更高效的专门模块。

关键技术拆解

Engram的实现依赖于几项关键技术。首先是分词器压缩,通过转为小写和Unicode规范化,将128K的词表压缩23%,有效减少了哈希冲突,提升了记忆效率。

其次是多头哈希记忆,用固定大小的参数表近似海量的N-gram组合空间,在控制参数量的同时极大扩展了记忆容量。再者是上下文感知门控,它作为“智能开关”,根据当前语境动态激活最相关的记忆,避免了信息混淆。最后是深度可分离因果卷积与多分支架构,前者用于扩大感受野,后者则通过并行计算显著提升了GPU利用率。

实验性能提升

实验数据有力地证明了Engram的有效性。在相同算力下,采用Engram的27B模型在训练上实现了计算量减少18%,性能却超越了同参数量的MOE模型。在长上下文任务中,经过微调的Engram模型在RULER基准测试的多个子任务上获得超过12个百分点的提升,如多查询大海捞针任务得分从84.2%提升至97.0%。

更令人振奋的是,研究发现增加记忆槽位数量能够持续稳定地降低验证损失,这意味着模型可以通过“加装内存条”的方式提升性能,而无需增加计算成本,为模型扩展提供了全新的维度。

技术对比分析

Engram与现有主流技术形成了鲜明对比。相较于MOE(混合专家模型)的“计算换容量”,Engram是“记忆换容量”,两者是互补而非替代关系。与RAG(检索增强生成)相比,Engram将知识内化为模型一部分,而非外挂数据库,因此延迟更低,知识一致性更强。

不同于Mamba等状态空间模型专注于替代注意力机制以处理全局依赖,Engram则与注意力机制互补,专注于处理局部静态模式,让注意力机制可以更专注于全局动态关系,二者结合能发挥更大效用。

Engram架构的价值在于开辟了记忆与计算独立扩展的新范式,挑战了单纯堆叠参数的传统路径。它预示着未来AI模型可能是神经网络与智能检索引擎的结合体,为解决效率瓶颈提供了极具想象力的方案,并为个性化AI和在线学习等领域指明了新的发展方向。

内容由AI生成
0
扫一下,分享更方便,购买更轻松
0评论

当前文章无评论,是时候发表评论了
提示信息

取消
确认
评论举报

最新文章 热门文章