阅读研究雷达投资体系
登录 / 注册
← 返回阅读
资讯专题 · LLM推理内存优化1 分钟阅读·2026-10-17DisconfirmAI 编辑部

普林斯顿与微软研究团队提出AbridgeKV,自报长链推理缓存压缩后H100速度提升3.2倍

AbridgeKV通过训练模型将长链推理的KV缓存压缩至学习到的暂停标记中,在保持准确率的同时实现显著的速度和内存优化。

重要性实质性证据E2 未复现写法快讯

核验于 2026-10-17 01:26

普林斯顿大学与微软研究院的研究人员提出了AbridgeKV,一种用于压缩长链思维(Chain-of-Thought)生成过程中KV缓存的方法。该方法不依赖传统的缓存驱逐策略,而是训练模型将每块生成的文本压缩进几个“学习到的暂停标记”(learned pause tokens)的KV条目中。这些暂停标记是带有训练嵌入的额外输入标记,本身不产生输出,但能编码前文信息。一旦完成编码,原始文本块的缓存即被释放,后续标记转而关注这些暂停标记的KV条目。

在包含446道竞赛数学题(来自AIME、HMMT等)的测试中,使用32K token生成长度限制时,AbridgeKV仅保留四分之一的生成缓存即可匹配同等训练条件下全KV模型的准确率和输出长度。相比之下,最强的已发表驱逐方法在相同预算下会导致更长的生成过程,而最接近的训练型压缩器LightThinker则损失了10个百分点的准确率。

在实际部署方面,该研究指出在vLLM服务引擎中,AbridgeKV使每个序列的缓存大小减少了3.5倍,并在饱和的H100 GPU上实现了比全KV模式快3.2倍的单问题完成速度。这一结果基于Qwen3-4B-Base模型在OpenMathReasoning数据集上的微调实验,属于作者自测数据,尚未见独立复现报告。

来源

  • 原始出处github.com↗
专题

LLM推理内存优化

大语言模型(LLM)的长链推理正面临严峻的内存瓶颈:随着生成令牌数量增加,键值缓存(KV Cache)线性增长,导致显存耗尽或带宽受限。【背景】传统方法如稀疏注意力主要解决计算量,而近期研究转向通过算法压缩、分层卸载及异构硬件协同来突破容量…

2 节点2 篇更新于 2026-10-17 04:41

事情怎么走到这一步

  1. 2026-10-17
登录 / 注册
阅读研究雷达投资体系
普林斯顿与微软发布AbridgeKV,通过学习到的暂停标记压缩长链推理的KV缓存,自报在H100上速度提升3.2倍。
  • 2026-10-17SemiAnalysis播客讨论HiCache与HiSparse,指出稀疏注意力主要解决带宽问题,而HiCache通过卸载KV缓存至低带宽存储缓解容量压力。
  • 还没定论的部分

    • AbridgeKV的压缩策略在代码生成等非数学推理任务中的泛化能力如何?
    • GPU-FPGA异构系统在大规模生产环境中的部署成本与维护复杂度是否可行?
    • 不同内存优化技术(如量化、卸载、压缩)组合使用时的性能叠加效应是否存在边际递减?

    这条线上还有

    • SemiAnalysis播客:HiCache与HiSparse内存优化讨论2026-10-17
    读完整专题 →
    带入研究 →