[控场AI]
产品

galahad-kv

开源KV状态持久化工具,将Transformer推理过程中的KV Cache保存至加密本地NVMe磁盘,支持按字节精确重新加载,避免重复计算,显著提升长上下文推理速度并降低GPU能耗

时间轴 (近 90 天)

10月9日

开源工具galahad-kv将每个约1.6万token块的KV状态保存到加密的本地NVMe磁盘,之后按字节精确地重新加载,完全跳过重新计算

待验证50%
10月9日

在0到5000万token的各个深度上,研究者探测的每个块都成功从加密存储中加载回来,100次探测100次命中,两个模型皆是如此

待验证50%
10月9日

加载一个块比重新计算它快2.8到4.3倍,GPU能耗减少8.8到12.3倍

待验证50%
10月9日

在整个5000万token的流处理过程中,GPU显存占用始终保持平稳,因为任意时刻只需加载一个块到显存

待验证50%
10月9日

galahad-kv是对存储状态的复用而非更宽的注意力窗口,系统一次只加载一个块,并没有让模型同时看到5000万token

待验证50%
10月9日

写入记忆是一次性成本,首次处理文本仍需完整计算KV状态;5000万token的KV状态会占用数TB的本地NVMe磁盘空间

待验证50%
10月9日

galahad-kv方法接近RAG思路,差异在于RAG检索原始文本片段并重新编码,而本方法直接复用已编码的KV状态,但跨块的直接注意力交互不存在

待验证50%
10月9日

研究提供单GPU的可复现方案,使用公开软件,且galahad-kv包采用免费许可

待验证50%
10月9日

该研究的测试协议被设计为能够抵抗常见的长上下文基准刷分手段,如将关键信息放序列首尾、使用训练集文档、设计过于简单的问题

待验证50%

全部知识事实 (9)

来源文章