无需复杂训练!通过 Q-Filters 实现 KV 缓存的高效压缩,从而显著提升推理性能

近年来,基于 Transformer 架构的大型语言模型(LLMs)取得了显著进展,诸如 Gemini-Pro1.5、Claude-3、GPT-4和 Llama-3.1等新模型能够处理成百上千的token。

然而,这些扩展的上下文长度带来了实际应用中的一些重大挑战。随着序列长度的增加,解码延迟上升,内存限制也成为了一个严重的瓶颈。KV 缓存在推理过程中储存上下文信息,随着上下文长度的增加,缓存的大小也呈比例增长,这导致内存饱和,严重影响了处理长输入序列的效率,因此迫切需要优化解决方案。

QQ_1741335458994.png

虽然市场上存在一些无训练的方法,但它们通常依赖于获取注意力权重来确定键值对的重要性,这使其与高效的注意力算法(如 FlashAttention)不兼容。这些方法往往需要对注意力矩阵进行部分重新计算,从而引入了时间和内存开销。因此,现有的压缩算法主要用于在生成答案之前压缩提示,而非优化在内存受限的生成过程中。这一局限性强调了需要开发既能保持模型性能又不需要架构修改的压缩技术。

来自索邦大学、法国国家信息与自动化研究所、罗马萨平扎大学、爱丁堡大学和 Miniml.AI 的研究团队提出了 Q-Filters,这是一种强大的无训练 KV 缓存压缩技术,利用基于查询的过滤方法来优化内存使用,同时保持模型性能。Q-Filters 通过评估与当前查询相关的键值对的重要性,而不是依赖于注意力权重。这种方法确保了与高效注意力算法的兼容性,且无需重新训练或修改架构。通过动态评估并保留最相关的上下文信息,Q-Filters 实现了显著的内存减少,同时维持了推理质量。

QQ_1741335465420.png

Q-Filters 在多个评估场景中表现出色,始终优于现有的 KV 缓存压缩方法。在对 Pile 数据集的语言建模测试中,该技术在所有压缩方案中实现了最低的困惑度。特别是在 Llama-3.1-70B 模型上,Q-Filters 在上下文保留至关重要的序列后半部分显示出显著的困惑度降低。

在 “针在干草堆” 任务中,Q-Filters 保持了91% 的准确率,成功地保存了极端上下文长度(从1K 到64K token)中的重要信息。综合评估还验证了该方法的优越性,尤其是在高压缩率下(32倍),Q-Filters 在长上下文建模基准测试中取得了最高分。

论文:https://arxiv.org/abs/2503.02812

huggingface:https://huggingface.co/collections/nthngdy/q-filters-67a4994dcb302a3d37f3d119

划重点:

🔍 Q-Filters 是一种无训练的 KV 缓存压缩技术,能够有效优化内存使用而不损失模型性能。

📊 该方法在多个评估中表现优越,特别是在语言建模和极端上下文任务上取得了最低的困惑度和最高的准确率。

🛠️ Q-Filters 与高效注意力算法兼容,且只需在模型训练后进行一次准备步骤,便可用于实际应用。

猜你喜欢

杨柳青年画与苏州桃花坞年画相映成辉:解读“南桃北柳”之美的艺术之旅
杨柳青年画与苏州桃花坞年画相映成辉:解读“南桃北柳”之美的艺术之旅

中国传统年画艺术瑰宝“南桃北柳”备受瞩目,其中北方的杨柳青年画与南方的苏州桃花坞年画更是享誉中外。杨柳青年画,作为北方年

平凡人来修仙最新版本攻略详解(v15.3)及玩法心得分享
平凡人来修仙最新版本攻略详解(v15.3)及玩法心得分享

在平凡人来修仙最新版(v15 3)中,角色创建时要根据自身喜好选择合适的初始属性。优先关注与修仙核心属性相关的选项,如灵

姚一携禅意画作开启全球巡展之旅,诚邀您共同领略禅美艺术的独特魅力!
姚一携禅意画作开启全球巡展之旅,诚邀您共同领略禅美艺术的独特魅力!

近期,中国著名禅意画家姚一因的艺术之旅开启了新的篇章,其个人艺术作品巡回展在中国美术馆盛大开幕,标志着这一系列展览的国内

如何通过选择日本Windows-Server-18来提高企业工作效率和优化运营流程
如何通过选择日本Windows-Server-18来提高企业工作效率和优化运营流程

什么是日本Windows Server 18?Windows Server 18 是微软推出的一款服务器操作系统,在日本

北京舞台绽放女性魅力,精彩女性题材剧目庆祝三八国际妇女节的盛宴
北京舞台绽放女性魅力,精彩女性题材剧目庆祝三八国际妇女节的盛宴

在北京的舞台上,随着“三八”国际妇女节的到来,一系列聚焦女性题材的精彩剧目纷纷亮相,为观众呈现了一场场视觉与心灵的盛宴。

如何顺利通过江湖高级棋艺132考核的全面指南与策略解析
如何顺利通过江湖高级棋艺132考核的全面指南与策略解析

在这充满刀光剑影与侠骨柔情的江湖中,有一项独特的考核——高级棋艺132考核,它犹如一面镜子,映照出江湖儿女们的智慧与谋略

YSL千人千色T9T9T9T9T9MBA系列:挑选最适合你个性与风格的独特时尚单品攻略
YSL千人千色T9T9T9T9T9MBA系列:挑选最适合你个性与风格的独特时尚单品攻略

随着时尚圈不断推陈出新,YSL千人千色T9T9T9T9T9MBA系列的推出再次引起了广泛关注。这一系列不仅在设计上极具独

“南桃北柳”之美:与杨柳青年画齐名的何方年画揭晓之谜
“南桃北柳”之美:与杨柳青年画齐名的何方年画揭晓之谜

中国民间艺术瑰宝:“南桃北柳”展现独特风情在中国丰富多彩的民间艺术中,天津杨柳青年画与苏州桃花坞年画以其独特的艺术风格,

产区划分新视角:精细划分一二三产区的标准与方法解析-图片附带解读
产区划分新视角:精细划分一二三产区的标准与方法解析-图片附带解读

产区划分新探:精产一二三产区在现代农业生产与经济发展的背景下,产区划分成为一个值得探讨的重要话题。产区划分的合理性,不仅

黑光生存进化是否支持不同平台间数据互通的详细解析与解答
黑光生存进化是否支持不同平台间数据互通的详细解析与解答

在黑光:生存进化这款游戏中,数据互通是许多玩家关心的重要问题。对于跨平台的数据互通情况,目前该游戏并没有明确的官方消息表