语言人工智能指南

多头潜在注意力

多头潜在注意力(MLA)是 DeepSeek-V2 中引入的一种注意力机制,它将占用内存的键值缓存压缩为一个小的共享潜在向量。

听听这个指南AI叙述 · ~2 min

在本页阅读时间:2分钟
  1. 概述
  2. 深入探讨
  3. 现实世界的实施
  4. 多头潜在注意力的未来
  5. 付诸实践
  6. 不断探索
  7. 常见问题

概述

它允许大型语言模型以少得多的 GPU 内存运行,同时保持质量接近标准关注度。

深入探讨

当转换器生成文本时,它会将每个过去标记的键和值向量存储在“KV 缓存”中。该缓存随着上下文长度而增长,并在推理期间主导内存使用。 MLA 将许多全尺寸键/值向量替换为每个令牌的单个低秩潜在向量,然后将其动态投影回每个头的键和值。由于仅缓存紧凑的潜在变量,DeepSeek-V2 报告称,与标准多头注意力机制相比,KV 缓存内存减少了 90% 以上,从而实现了更长的上下文和更大的批量大小。至关重要的是,上投影矩阵可以折叠成其他权重,因此 MLA 可以实现这种压缩,而建模质量几乎没有或没有可测量的损失。

现实世界的实施

为 DeepSeek-V2/V3 聊天模型提供服务,每个请求的 GPU 内存占用量显着减少

运行长文档问答,否则大型 KV 缓存会耗尽 VRAM

在固定 GPU 上增加推理批量大小,因为每个序列仅存储一个微小的潜在向量

在商用硬件上为检索增强助理启用更长的上下文窗口

多头潜在注意力的未来

MLA 帮助 DeepSeek-V2 和 V3 实现大规模服务的经济性,并且随着团队追求更便宜的长上下文推理,该技术正在传播。预计 MLA 风格的潜在压缩将与稀疏专家混合层、量化缓存和未来开放模型中的推测解码相结合。研究人员还在探索在质量下降之前潜在维度可以缩小多少,以及相同的低等级想法是否可以在训练期间压缩注意力,而不仅仅是推理。

付诸实践

为什么这很重要

  • 速度与规模. 语言工作流程可以在不牺牲一致性的情况下更快地移动。
  • 交通与覆盖范围. 它扩展了跨语言和沟通方式的访问。
  • 更清晰的判决. 团队可以花更多时间进行判断,而自动化则可以处理重复。

留意

  • 幻觉的事实可以悄悄地进入报告、支持流程或研究成果。
  • 及时的敏感性可能会在类似的请求中产生不一致的结果。
  • 如果访问控制薄弱,敏感文本数据可能会暴露。

后续步骤

  1. 在推出之前定义输出格式、语气和质量标准。
  2. 当准确性很重要时,请使用可信来源进行地面响应。
  3. 为高风险输出保留人工审查检查点。
  4. 跟踪故障模式并定期重新训练提示或工作流程。

不断探索

免费通讯

获取每日AI简报

每个工作日早晨提供三条经过验证的AI故事,使用简明英文撰写。永久免费,无广告。

每个工作日发送一封邮件。一键退订。我们永不出售或分享您的地址。

测试一下自己

参加多头潜在注意力测验

每个答案即时反馈,课程通过后可获得可验证ID的可分享证书。

开始测验

支持免费AI教育。AI Understanding是一家501(c)(3)非营利组织——无广告,无付费墙,永不收费。 捐款

常见问题

什么是多头潜在注意力?

多头潜在注意力(MLA)是 DeepSeek-V2 中引入的一种注意力机制,它将占用内存的键值缓存压缩为一个小的共享潜在向量。它允许大型语言模型以少得多的 GPU 内存运行,同时保持质量接近标准关注度。

多头潜在注意力旨在减少的主要问题是什么?

MLA 的目标是 KV 缓存,它随着上下文长度的增长而增长,并在文本生成期间占据内存。

MLA如何收缩KV缓存?

MLA 为每个标记缓存一个紧凑的潜在向量,并通过上投影从中重建键和值。

哪个模型首先引入了多头潜在注意力?

MLA由DeepSeek在其DeepSeek-V2模型中引入,并被引入到DeepSeek-V3中。

为什么 MLA 需要一个单独的“解耦”路径来进行旋转位置嵌入?

旋转变换不能被吸收到其他权重矩阵中,因此MLA保留了一个小的解耦关键组件来携带位置信息。

与标准多头注意力相比,DeepSeek-V2 从 MLA 报告的 KV 缓存内存减少了大约多少?

DeepSeek-V2 报告称,KV 缓存内存减少了 90% 以上,从而实现了更长的上下文和更大的批次。