察元AI教程库 · 文章 · 部署

KVPR:通过I/O感知的KV缓存部分重计算,实现高效的LLM推理

大语言模型(LLMs)的推理过程计算需求很高。为了降低自回归解码的开销,采用键值(KV)缓存来存储中间激活值,显著降低了生成每个标记的计算开销。然而,KV缓存所需的内存却成为一个瓶颈。

作者:Chaoyi Jiang, Lei Gao, Hossein Entezari Zarch · 语言:en

前往原址学习

简介

大语言模型(LLMs)的推理过程计算需求很高。为了降低自回归解码的开销,采用键值(KV)缓存来存储中间激活值,显著降低了生成每个标记的计算开销。然而,KV缓存所需的内存却成为一个瓶颈。

返回教程库 · 察元AI产品家族