← 最新论文
💻 computer science

VeriCache: Turning Lossy KV Cache into Lossless LLM Inference

VeriCache 是一个推理框架,它通过使用压缩的 KV 缓存来起草令牌并针对完整缓存进行验证,从而在实现与完整 KV 缓存解码完全无损的 LLM 输出的同时显著提升吞吐量,其中完整缓存被保留在 GPU 内存之外,仅在必要时交换进入。

原作者: Jiayi Yao, Samuel Shen, Kuntai Du, Shaoting Feng, Dongjoo Seo, Rui Zhang, Yuyang Huang, Yuhan Liu, Shan Lu, Junchen Jiang

发布于 2026-05-19
📖 1 分钟阅读☕ 轻松阅读

原作者: Jiayi Yao, Samuel Shen, Kuntai Du, Shaoting Feng, Dongjoo Seo, Rui Zhang, Yuyang Huang, Yuhan Liu, Shan Lu, Junchen Jiang

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

以下是用通俗易懂的语言和日常类比对论文 VeriCache 的解释。

核心难题:“内存过载”

想象一位超级智能的 AI 助手(大型语言模型),它需要记住海量信息来回答你的问题。这些信息存储在一个特殊的“草稿纸”中,称为KV 缓存

随着对话变长(比如撰写整本书或分析庞大的代码库),这个草稿纸会变得如此巨大,以至于无法放入 AI 快速但昂贵的内存(GPU)中。

  • 旧方案:为了使其能容纳,工程师开始“压缩”草稿纸。他们丢弃了一些细节或简化了数值(就像将一份 100 页的报告总结成 10 页)。
  • 弊端:这种压缩是有损的。这就像试图用一张模糊的地图开车。对于短途旅行,这没问题。但对于长途旅行(如编写代码或调用工具),AI 开始犯小错误。这些错误会累积,最终导致 AI 写出崩溃的代码或给出错误的答案,即使这些文字看起来是正确的。

新方案:VeriCache

研究人员构建了VeriCache,这是一个允许你为了速度使用“模糊地图”(压缩缓存),但会检查“真实地图”(完整缓存)以确保你不会迷路的系统。

把它想象成一位快速起草员和一位严格编辑

工作原理(类比)

  1. 起草员(压缩缓存)
    AI 使用小型、快速、压缩的内存,一次性快速写出一大堆句子(令牌)。这非常快,因为内存很小,可以轻松放入计算机中。

    • 类比:一个速写学生,基于快速总结猜测接下来的几个词。
  2. 严格编辑(完整缓存)
    在学生的工作发送给你之前,一位“严格编辑”会进行检查。编辑拥有整个原始、完美的内存访问权(因为它太大,无法一直放在桌面上,所以存放在房间另一头的文件柜里)。

    • 技巧:编辑仅在需要检查一批单词时,才将沉重的文件柜搬到桌面上。
  3. “交错”舞蹈(秘诀)
    这里是 VeriCache 巧妙之处。通常,如果你必须停下来去取一份沉重的文件,一切都会停滞。

    • VeriCache 的做法:当编辑走向文件柜去取沉重文件时,学生继续写作!
    • 因为取文件(将数据从存储移动到内存)和写下一句句子(使用 GPU)在计算机中使用的是不同的“道路”,它们可以同时发生而互不干扰。
    • 等到编辑带着文件回来时,学生已经写好了整整一个新段落。编辑检查前一个段落,修正任何错误,并批准其余部分。

为何这很重要

  • 无损速度:以前的方法迫使你做出选择:快但错(压缩)或者慢但完美(完整)。VeriCache 让你既快又完美。最终输出与如果你全程使用缓慢的完整内存所获得的结果完全相同。
  • 不再有“静默失败”:在编写代码或发出特定指令等任务中,微小的错误就是灾难。VeriCache 会在这些错误到达你之前将其捕获。
  • 适用于任何情况:它不关心内存是如何被压缩的。无论他们是丢弃了单词还是简化了数值,VeriCache 都可以将那个压缩版本用作“起草员”,并针对“完整”版本进行验证。

结果

在他们的测试中,VeriCache 能够比使用完整、缓慢的内存快高达 4 倍地生成文本,同时产生完全相同且正确的结果

简而言之:VeriCache 让 AI 在一条快速、轻量级的轨道上运行,但在后台设置了安全护栏,检查那条沉重、完美的轨道,确保 AI 永远不会掉下边缘。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →