🧬

多头潜在注意力(MLA)

Multi-head Latent Attention (MLA)
技术原理
注意力机制KV缓存DeepSeek

多头潜在注意力(MLA)是 DeepSeek-V2 论文引入的注意力变体,要解决的问题是:推理阶段 KV 缓存随上下文长度线性膨胀,成为长文本推理的显存瓶颈。

机制

与标准多头注意力为每个注意力头独立保存键值不同,MLA 把 KV 缓存显著压缩进一个低秩的潜在向量,推理时再从该向量还原所需表示。论文摘要明确写道:MLA 通过把 KV 缓存大幅压缩成潜在向量来保证高效推理。在 DeepSeek-V2 与 DeepSeek 67B 的对比中,论文报告 KV 缓存减少 93.3%,最大生成吞吐提升至 5.76 倍,同时训练成本节省 42.5%。

与其他注意力变体的边界

多查询注意力(MQA)和分组查询注意力(GQA)走的是「减少 KV 头数量」的路线,MLA 走的是「把 KV 压缩进潜在空间」的路线,两者目标一致(省显存、提吞吐)但手段不同。MLA 的代价是实现更复杂,且与部分依赖标准 KV 缓存布局的推理优化(如某些 KV 缓存复用策略)的兼容性需要额外适配。

适用边界

MLA 是模型架构层面的设计,用户无法在已有模型上后补;它的价值主要出现在自托管长上下文推理、或评估「为什么这个模型的推理显存占用更低」时。

🔗 探索更多 AI 工具板块

除了本板块,AI工具宝箱还有这些自有内容板块,帮你从不同角度发现好工具。

☆ 0