前缀缓存为什么能加速 LLM 推理:从 KV Cache 说起

举报
yd_241388977 发表于 2026/10/28 16:41:44 2026/10/28
【摘要】 大模型推理分两个阶段:prefill 把输入提示词整体算一遍,decode 再一个 token 一个 token 往外生成。慢和贵的大头在 prefill,提示词越长越明显。先说 KV Cache。自注意力要用到每个 token 的 Key 和 Value,decode 时如果每生成一个词都把前面的全部重算,计算量会爆炸,所以推理引擎会把已算出的 KV 存下来,后面直接复用。这是单次请求内的...

大模型推理分两个阶段:prefill 把输入提示词整体算一遍,decode 再一个 token 一个 token 往外生成。慢和贵的大头在 prefill,提示词越长越明显。

先说 KV Cache。自注意力要用到每个 token 的 Key 和 Value,decode 时如果每生成一个词都把前面的全部重算,计算量会爆炸,所以推理引擎会把已算出的 KV 存下来,后面直接复用。这是单次请求内的优化。

前缀缓存是把它推到跨请求:很多请求的开头是完全一样的——同一段系统提示词、同一批 few-shot 示例、同一份长文档。在 token 层面逐字一致的前提下,这段公共前缀的 KV 只算一次、存下来,后续请求命中就直接跳过这段 prefill,只算后面变化的部分。效果是首 token 延迟明显下降,算力成本也跟着降,前缀越长、复用次数越多,收益越大。

用好它有三条经验。一是提示词要「稳定在前、变化在后」:前缀是按顺序逐 token 匹配的,开头改一个字,后面全失效,用户输入、时间戳这类动态内容务必放最后。二是缓存不是永久的,各家实现有容量和过期策略,命中率要实测,不能假设。三是它只省 prefill,decode 的逐字生成一点不省,长输出场景别指望它解决一切。

一句话总结:前缀缓存省的是重复劳动——同一段话,别让模型一遍遍重读。

【声明】本内容来自华为云开发者社区博主,不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息,否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0)

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。