从 Attention 到 KV Cache:理解推理过程

先给你一句话版本:

KV Cache = 在自回归生成时,把历史 token 的 Key / Value 向量存起来,下一个 token 只算自己的 Q,然后复用过去的 K/V,避免重复算 Attention。

就这么简单。


一、先把 Attention 讲人话


Transformer 做的事本质就一句话:

每个 token 去“问”其他 token:我该关注谁?

怎么问?

每个 token 先变成三个向量:

  • Q(Query):我想找什么?
  • K(Key):我是什么标签?
  • V(Value):如果选我,我提供什么内容?

然后公式:

Attention(Q,K,V) = softmax(QK^T / sqrt(dk)) V

翻译成人话:

  1. 用 Q 和所有 K 算相似度
  2. 归一化成权重
  3. 用这些权重加权求和所有 V

得到一个新的向量 —— 这就是这个 token “看完全文之后”的理解。


二、关键问题来了:为什么推理会爆炸?


LLM 是自回归生成。

生成第 1 个 token
生成第 2 个 token
生成第 3 个 token

假设现在已经有 N 个 token。

第 N+1 个 token 的 Attention 需要:

  • 当前 token 的 Q
  • 前面所有 token 的 K
  • 前面所有 token 的 V

问题在哪?

如果你不缓存,

每生成一个新 token:

  • 所有历史 token 都要重新算 K
  • 所有历史 token 都要重新算 V
  • 然后再算 Attention

你等于在一遍遍重复做历史计算。

这不是优雅,这是浪费算力。


三、KV Cache 到底缓存了什么?


记住一个关键推导:

生成第 N+1 个 token 时:

  • 只需要当前 token 的 Q
  • 需要所有历史 token 的 K 和 V

历史 token 的 Q 没用了。
历史 token 的 K/V 不会变。

所以:

👉 K 和 V 可以缓存
👉 Q 没必要缓存(每次只用当前)

这就是 KV Cache 的本质。


四、没有 KV Cache 会发生什么?


复杂度会是:

O(N³)

有 KV Cache:

每步只算当前 Q/K/V,然后和缓存拼起来:

复杂度降到:

O(N²)

这不是小优化。

这是决定推理成本生死的东西。

所以业界才说:

KV Cache 用得好,模型跑得飞起。

这不是营销,是数学。


五、为什么它叫 Cache,但其实不一样?


传统 KV Cache(比如 Redis):

缓存“已有数据”。

LLM 的 KV Cache:

缓存“中间计算结果”。

它不是为了减少 IO。
它是为了避免重复 Attention 计算。

它更像:

计算缓存(Compute Cache)

这个区别很重要。


六、真正的工程问题在哪里?


理论很简单。

工程地狱在这里:

  • 每个 token 的 KV 很大
  • 模型层数很多(96层不是闹着玩的)
  • 每层都要存 KV
  • GPU 显存只有 80GB

举个真实级别的数字:

13B 模型
单 token KV ≈ 800KB
2048 tokens ≈ 1.6GB

一个请求就能吃掉 1~2GB 显存。

你要同时服务 50 个用户?

炸。

这就是 vLLM 出现的原因。


七、vLLM 干了什么?


一句话:

把 KV Cache 分页化。

它借鉴操作系统的“虚拟内存”。

不再要求一整段连续显存。

改成:

  • 每几个 token 一个 block
  • 逻辑连续
  • 物理可分散

结果:

  • 减少内存碎片
  • 支持共享 prefix
  • 支持 beam search
  • 提高显存利用率

这招不炫技。

但他妈的实用。


八、SGLang 干了什么?


它干了一件更聪明的事:

提高 KV Cache 的复用率。

核心思路:

  • 把所有请求组织成 Radix Tree
  • 共享最长前缀
  • 优先调度共享最多的请求

意思是:

“既然已经算过这个前缀,就多用几次。”

这比单纯分页更狠。


九、真正的底层认知


你要记住三个核心点:

1️⃣ Transformer 本质是矩阵运算机器

Attention 是加权信息混合。

2️⃣ 推理阶段的瓶颈是重复计算

KV Cache 就是砍掉重复。

3️⃣ 推理系统的胜负在显存管理

不是模型参数,而是 KV 的管理效率。

Scaling Law 时代靠堆参数。
现在靠榨显存。

时代变了。


十、你真正应该理解的哲学


Transformer 的三个灵魂:

  • 向量表示
  • Next-token 机制
  • Contextual Attention

KV Cache 是 Next-token 机制的副产品。

因为是“逐 token 生成”,才有缓存价值。

如果不是自回归生成,
KV Cache 根本不存在。


最后一句总结


KV Cache = 用空间换时间,用显存换算力,用工程换成本。

它不优雅。

但它是现实世界里最重要的优化。

普通人如何抓住AI大模型的风口?

领取方式在文末

为什么要学习大模型?

目前AI大模型的技术岗位与能力培养随着人工智能技术的迅速发展和应用 , 大模型作为其中的重要组成部分 , 正逐渐成为推动人工智能发展的重要引擎 。大模型以其强大的数据处理和模式识别能力, 广泛应用于自然语言处理 、计算机视觉 、 智能推荐等领域 ,为各行各业带来了革命性的改变和机遇 。

目前,开源人工智能大模型已应用于医疗、政务、法律、汽车、娱乐、金融、互联网、教育、制造业、企业服务等多个场景,其中,应用于金融、企业服务、制造业和法律领域的大模型在本次调研中占比超过 30%。
在这里插入图片描述

随着AI大模型技术的迅速发展,相关岗位的需求也日益增加。大模型产业链催生了一批高薪新职业:
在这里插入图片描述

人工智能大潮已来,不加入就可能被淘汰。如果你是技术人,尤其是互联网从业者,现在就开始学习AI大模型技术,真的是给你的人生一个重要建议!

最后

只要你真心想学习AI大模型技术,这份精心整理的学习资料我愿意无偿分享给你,但是想学技术去乱搞的人别来找我!

在当前这个人工智能高速发展的时代,AI大模型正在深刻改变各行各业。我国对高水平AI人才的需求也日益增长,真正懂技术、能落地的人才依旧紧缺。我也希望通过这份资料,能够帮助更多有志于AI领域的朋友入门并深入学习。

真诚无偿分享!!!
vx扫描下方二维码即可
加上后会一个个给大家发

【附赠一节免费的直播讲座,技术大佬带你学习大模型的相关知识、学习思路、就业前景以及怎么结合当前的工作发展方向等,欢迎大家~】
在这里插入图片描述

大模型全套学习资料展示

自我们与MoPaaS魔泊云合作以来,我们不断打磨课程体系与技术内容,在细节上精益求精,同时在技术层面也新增了许多前沿且实用的内容,力求为大家带来更系统、更实战、更落地的大模型学习体验。

图片

希望这份系统、实用的大模型学习路径,能够帮助你从零入门,进阶到实战,真正掌握AI时代的核心技能!

01 教学内容

图片

  • 从零到精通完整闭环:【基础理论 →RAG开发 → Agent设计 → 模型微调与私有化部署调→热门技术】5大模块,内容比传统教材更贴近企业实战!

  • 大量真实项目案例: 带你亲自上手搞数据清洗、模型调优这些硬核操作,把课本知识变成真本事‌!

02适学人群

应届毕业生‌: 无工作经验但想要系统学习AI大模型技术,期待通过实战项目掌握核心技术。

零基础转型‌: 非技术背景但关注AI应用场景,计划通过低代码工具实现“AI+行业”跨界‌。

业务赋能突破瓶颈: 传统开发者(Java/前端等)学习Transformer架构与LangChain框架,向AI全栈工程师转型‌。

image.png

vx扫描下方二维码即可
【附赠一节免费的直播讲座,技术大佬带你学习大模型的相关知识、学习思路、就业前景以及怎么结合当前的工作发展方向等,欢迎大家~】
在这里插入图片描述

本教程比较珍贵,仅限大家自行学习,不要传播!更严禁商用!

03 入门到进阶学习路线图

大模型学习路线图,整体分为5个大的阶段:
图片

04 视频和书籍PDF合集

图片

从0到掌握主流大模型技术视频教程(涵盖模型训练、微调、RAG、LangChain、Agent开发等实战方向)

图片

新手必备的大模型学习PDF书单来了!全是硬核知识,帮你少走弯路(不吹牛,真有用)
图片

05 行业报告+白皮书合集

收集70+报告与白皮书,了解行业最新动态!
图片

06 90+份面试题/经验

AI大模型岗位面试经验总结(谁学技术不是为了赚$呢,找个好的岗位很重要)图片
在这里插入图片描述

07 deepseek部署包+技巧大全

在这里插入图片描述

由于篇幅有限

只展示部分资料

并且还在持续更新中…

真诚无偿分享!!!
vx扫描下方二维码即可
加上后会一个个给大家发

【附赠一节免费的直播讲座,技术大佬带你学习大模型的相关知识、学习思路、就业前景以及怎么结合当前的工作发展方向等,欢迎大家~】
在这里插入图片描述

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐