工程实践更新于 2026-10-24
KV 缓存
推理时缓存已处理 token 的 Key/Value,避免每生成一个新词都重算历史。
#推理#显存#性能
通俗解释
自回归生成每一步都要对全部历史做注意力。KV 缓存把历史 token 的 Key 和 Value 保存下来,新 token 只需计算自己的 Query 并拼接缓存,显著降低延迟。代价是显存占用随上下文长度和并发数线性增长,长上下文服务因此常配合分页缓存(PagedAttention)、量化缓存或多轮会话裁剪。缓存命中率直接影响推理成本和首 token 延迟。
例子
多轮对话中保留 KV 缓存可以避免每轮重算前文,但上下文越长,显存占用越高。