Skip to content
简体中文

切换主题

// 阶段 6 · 推理部署与效率

解码、采样与 KV Cache

预填充与解码、温度/top-p、KV Cache、内存带宽瓶颈。

约 6 小时4 个算法2 篇论文已有正文

01学习目标

建议先学:现代 LLM 架构:位置编码、GQA/MLA 与长上下文

02站内正文

03要掌握的算法

每张卡都有原始论文、前置和学习资料。

04必读论文

05课程与视频

06动手:项目、代码与练习

全部是官方或作者公开的免费页面,多数能在浏览器或免费 Colab 里直接运行。

动手项目

给自己的小 GPT 加上 KV Cache,测量生成 1000 token 的提速倍数。

  1. 1
    Hugging Face Transformers:生成策略(贪心、束搜索、采样、推测解码) ↗官方文档

    用同一个模型对比贪心、束搜索、top-p 采样和推测解码的输出与速度。

  2. 2
    karpathy/nanoGPT(参考实现) ↗参考实现

    给 nanoGPT 的生成函数加上 KV 缓存,测量提速。

07自测题

先自己想,再点开看答案。

KV Cache 大小与哪些量成正比?

层数×KV 头数×头维度×序列长度×batch×2。

top-p=0.9 是什么意思?

只在累计概率达 90% 的最小候选集中采样。

为何解码阶段 GPU 算力利用率低?

每步只算一个 token,主要时间在读权重和缓存。

所在人群路线:应用开发者

数据更新于:AI 算法图谱与课程体系(北京时间)