Files
myWiki/concepts/multi-head-latent-attention.md

1.3 KiB
Raw Blame History

Multi-head Latent Attention (MLA)

多潜在头注意力DeepSeek 团队在 DeepSeek-V2 (2024) 中提出的革命性注意力机制。

核心思想

MLA 通过低秩联合压缩将 KV 映射到远小于原始维度的潜在空间,推理时只需缓存压缩后的潜在向量,解码时再解压重建 KV。这与 MQA/GQA 的"共享头"策略不同——MLA 是信息论意义上的压缩,而非简单的共享。

关键机制

  1. 低秩压缩: KV 先映射到 d_latent << d 的潜在空间
  2. 潜在缓存: 推理时只缓存压缩后的潜在向量,而非完整 KV
  3. 按需重建: 解码时从潜在向量高效重建完整 KV

里程碑意义

  • 缓存缩减: KV 缓存可减少至 MHA 的 1/10 ~ 1/20
  • 质量保持: 压缩不显著影响模型性能
  • 工业落地: DeepSeek-V2/V3 的核心推理效率引擎

与 HCA 关系

MLA 可视为 heavily-compressed-attention 的泛化形式——HCA 是 DeepSeek-V4 中 MLA 的增强版,融合了流形约束。

相关概念