KV Cache 工程详解:单序列 8K 上下文即占 1 GiB,服务性能胜负手

techNmak · x · 2026-10-31

techNmak 发长文梳理 KV Cache 工程——LLM 推理服务性能的关键战场,附一张信息量很大的图解。

基本原理:自回归生成时不必每步重算历史 token 的 K/V,每层只追加新的 key/value,query 对保留的 K/V 做注意力。过去的 query 无需缓存。

内存账:全注意力下 KV 存储约为 2×层数×缓存 token 数×KV 头数×头维度×每元素字节数。以 32 层、8 KV 头、head dim 128、BF16 计,每位置约 128 KiB;8192 位置的单序列约 1 GiB,32 条不共享序列即 32 GiB——这还没算权重、激活与运行时开销。

技术手段与权衡:

生产监控:KV 用量要与活跃序列长度、并发、空闲缓存块、前缀缓存命中率、抢占、排队时间、TTFT 和 ITL 一起看。KV 短缺可能表现为排队/抢占而非 GPU OOM。

结论:KV 优化最终同时是内存分配、带宽、调度、缓存复用问题,也是高并发下 GPU 产出的上限之一。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →