01学习目标
- 估算一个模型的 KV Cache 大小
- 解释为何解码受带宽限制
- 会调采样参数
02站内正文
03要掌握的算法
每张卡都有原始论文、前置和学习资料。
温度与 Top-p 采样Temperature & Nucleus (Top-p) Sampling 控制生成随机性:只从累计概率达到 p 的候选里抽样。 2019核心 KV Cache 把已生成 token 的 Key/Value 存起来,避免每步重复计算。 核心 注意力汇聚点 StreamingLLMAttention Sinks (StreamingLLM) 保留开头几个“汇聚”token 的 KV,就能无限长流式生成。 2023扩展 分组查询注意力 GQAGrouped-Query Attention (GQA) 把注意力头分组,每组共享一份 K/V,在 MHA 与 MQA 间折中。 2023核心
04必读论文
05课程与视频
06动手:项目、代码与练习
全部是官方或作者公开的免费页面,多数能在浏览器或免费 Colab 里直接运行。
动手项目
给自己的小 GPT 加上 KV Cache,测量生成 1000 token 的提速倍数。
- 1
- 2
07自测题
先自己想,再点开看答案。
KV Cache 大小与哪些量成正比?
层数×KV 头数×头维度×序列长度×batch×2。
top-p=0.9 是什么意思?
只在累计概率达 90% 的最小候选集中采样。
为何解码阶段 GPU 算力利用率低?
每步只算一个 token,主要时间在读权重和缓存。
所在人群路线:应用开发者