

编订|Panda
遐想这么一个场景:你正盯着屏幕,看着你的自主 AI 智能体(比如 OpenClaw)浪漫地运作。

它正在自主审查一个包含数十万行代码的史诗级开源花样,穿梭于无数的文献、API 文档和调试日记之间。它透露得像一个不知疲钝的超等才能员,但在这「无所不成」的表象之下,藏匿着一个随时可能引爆的硬件梦魇 —— 跟着高下文变得越来越长,大模子的「使命顾虑」正在暴涨,像一个无底洞一样,冷凌弃地吞吃着不菲的 GPU 显存池!
这个令系数企业级 AI 开辟者闻风远扬的显存杀手,便是 KV Cache。
但当今,惩处决策来了,来自麻省理工学院(MIT)的商议团队(Adam Zweiger、Xinghong Fu 等东谈主)。他们开辟出了一种名为「注见识匹配」(Attention Matching)的全新潜在空间(Latent Space)压缩本领。

论文标题:Fast KV Compaction via Attention Matching
代码地址:https://github.com/adamzweiger/compaction
其或者在短短几秒钟内,将大型谈话模子的高下文内存浪漫压缩高达 50 倍,且险些莫得任何精度亏欠!

这意味着蓝本需要一通盘 H100 GPU 阵列才能勉强相沿的超长对话或巨型文档分析任务,当今可能只需要单张显卡就能孤高跑满并发。一场对于 AI 基础设施的戒指立异,似乎已悄然打响。
不菲的使命顾虑
大模子的阿喀琉斯之踵
要知晓这项本领有何等逆天,咱们必须先直视大模子的软肋。
LLM 是自回来的,它们生成回当令是逐 token 往外吐的。为了幸免在计议每一个新词时,都要把长达几万字的聊天纪录重新到尾从新打算一遍,模子必须将之前处理过的每一个 token 的「数学灵魂」缓存起来 —— 这些被索取出来的多维向量,便是「键(Key)」和「值(Value)」对,即 KV Cache。
跟着高下文的拉伸,这层使命顾虑会不可逆转地扩张。
在当代企业级应用中,比如分析成百上千页的法律协议、看护长达数月的私东谈主 AI 伴侣顾虑,或者运行 OpenClaw 这么的自治编码智能体,单单一个用户的苦求,其 KV Cache 就能短暂飙升到数十 GB。
正如论文第一作家 Adam Zweiger 所言:「在超长高下文做事中,KV Cache 是最大的物理瓶颈。它不仅死死锁住了并发量,将就你放松批处理界限,甚而逼着系统进行极其影响性能的正常卸载。」
靠近这个吞金兽,商议者们曾尝试过很多决策:
Token 丢弃与团结(如 H2O, SnapKV, PyramidKV 等):这些设施试图踢掉那些模子认为「不蹙迫」的 token。在轻度压缩时还能拼凑,但一朝将压缩率拉高(比如试图压缩 10 倍以上),模子的才能就会遇到断崖式下落。
文本选录:这是现时工业界最无奈的标配。当内存见底时,系统暂停,让模子我方写一段高下文总结,然后清空原有顾虑。这种设施尽头「有损」,会把极其要害的狭窄细节(比如医疗纪录里的一个荒僻方针)透彻抹除。
潜空间压缩(如 Cartridges):这是近期的前沿探索,证实了高比例压缩不仅可行,而且还能保握高精度。但它的代价极其抖擞:它需要通过极其逐步的端到端梯度下降来考验这些压缩后的顾虑。为了压缩一段高下文,哪怕动用不菲的 GPU,也需要耗尽数小时!这在条目「秒回」的及时企业应用中,简直是信口胡言。
咱们需要一种既有 Cartridges 的精度,又有传统设施速率的终极魔法。而 MIT 的「注见识匹配」,恰是为此而生。
冲突常理的数学魔法
「注见识匹配」的底层逻辑
MIT 的商议东谈主员莫得死磕逐步的机器学习考验,而是想出了一个绝妙的数学捷径。他们退后一步,问了一个极其本体的问题:当咱们压缩顾虑时,模子究竟在乎什么?
谜底是:模子根底不在乎你存了若干个 Key 和 Value,它只在乎当它抛出一个查询(Query,即 q)时,这堆顾虑能给它复返什么戒指!
为了完好行使 AI,让它以为「压缩后的顾虑和蓝本精深的顾虑一模一样」,压缩后的键值对 (C_k, C_v) 必须严格匹配原始顾虑的两个中枢数学属性:
注见识输出(Attention Output):这是 AI 索取到的实践信息向量。
注见识质地(Attention Mass):这是极其要害的小数。在拼接新 token 或旧顾虑时,一段顾虑的话语权取决于它的「质地」。
淌若你径直把 1000 个 token 压缩成 20 个,那么这 20 个 token 的「总质地」皆备拼不外蓝本的 1000 个,这会导致模子在后续推理时,尽头看轻这部分被压缩的顾虑。为了破解这个死局,商议团队引入了一个狭窄但号称神来之笔的变量:每 token 标量偏差 β。
这个 β 偏差就像是一个「杠杆权重」,它在注见识打算的指数层面上对保留住来的 Key 进行乘法重加权,让戋戋 1 个被保留的 Key,或者爆发出代表 50 个被移除 Key 的巨大「质地」!
淌若用严谨的数学谈话(如论文中的公式 1 和 2)来抒发,他们要优化的方向便是找到 (C_k,β, C_v),使得对于系数关连的查询 q:

况兼匹配总质地:

更惊东谈主的是,由于这种精妙的框架构建,这个看似复杂的非线性优化问题,果然自但是然地解体了!商议东谈主员完全甩掉了吃算力的反向传播和梯度优化。
率先,锁定 C_k 后,质地匹配问题退化成了一个非负最小二乘法(NNLS)问题,短暂就能打算出偏差 β。
滚球中国官方网站入口随后,注见识输出匹配问题径直酿成了一个圭表的正常最小二乘法(OLS)问题,通过肤浅的代数矩阵运算,片刻就能求出压缩后的值 C_v!
这简直是降维打击。蓝本需要数小时的考验,被线性代数优化到了以「秒」为单元。

来自 VentureBeat,由 AI 生成
预判你的预判
何如索取「参锻练询」与挑选「金钥匙」?
有了数学火器,体球网2026世界杯赛事直播入口接下来的工程落地一样惊艳。为了让压缩算法知谈该保留什么,系统需要一批「参锻练询」(Q_ref),行为模子将来可能忽视的问题的「替身」。
商议团队遐想了极其机灵的「预演」机制:
换取预填充:暗暗在文档末尾加一句荫藏辅导:「换取前边的高下文」,然后拿获模子在试图复述时产生的里面 Query 向量。
自我学习:让模子对文档进行快速的合成任务,比如「索取系数中枢事实」或「把日历结构化为 JSON」,从而嗅探出模子在深度推理时会生成什么样的 Query。
手里攥着这些极具代表性的 Query 探针,系统运行从原始的茫茫 Key 海中挑选「金钥匙」(C_k)。论文中提供了两种设施:
最高注见识法(Highest Attention Keys):这是一种闪电般的启发式设施,径直挑出在参锻练询中被和顺度最高的 Keys。速率极快,性价比超高。
正交匹配跟踪(Orthogonal Matching Pursuit, OMP):这是一种愈加极客和狡计的算法。它像搭积木一样,每一步都精挑细选一个最能填补「质地舛误」残差的 Key,然后用 NNLS 从新校准权重。天然略微耗时(已经只是几分钟级别),但能将压实质地推向巅峰(AM-OMP)。

并非系数「注见识」生来对等
非均匀压缩计谋
这还不是要点,在深刻探索模子架构时,他们发现了一个道理的表象:在多头注见识机制中,并非系数的「头」都是使命狂。
有些 Head 尽头狡计,需要精深的 KV 容量才能保握性能(比如厚爱长程依赖的 Head);而另一些 Head 则极其佛系,哪怕你把它的顾虑砍掉 90%,它已经能完好运转(比如只和顺局部词法结构的 Head)。

基于这个瞻念察,团队开辟了非均匀压缩(Nonuniform Compaction)计谋:为每一个模子事先打算了一条「明锐度弧线」,就像是给每一个注见识头进行了一次体检。在实践压缩时,系统不再是一刀切,而是将极其贵重的显存预算,歪斜分拨给那些对信息最明锐的「中枢 Head」。这一计谋的引入,径直让压缩后的模子性能完毕了质的飞跃!
即使在像 Gemma-3-12B 这种多数使用了滑动窗口注见识的搀杂架构模子上,注见识匹配已经透暴露了惊东谈主的合乎性和鲁棒性。
压力测试
见证古迹的时刻
为了考证这项本领是否确切能在现实寰宇的绞肉机中存活,商议东谈主员遴荐了 Qwen3-4B、Llama3.1-8B 和 Gemma3-12B,并将它们扔进了两个截然相背的测试场。

1. QuALITY 基准测试:秒杀全场
在这个包含 5000 到 8000 词的圭表阅读知晓测试中,Attention Matching 在 50 倍的极限压缩比下,只是耗时几秒到一分钟(取决于是否使用 OMP 算法),就透彻打爆了 H2O+、SnapKV、KVzip 等系数基于 token 编订的前辈。它的准确率弧线牢牢咬住了耗时数小时的 Cartridges,证实了什么是「快、准、狠」。
2. LongHealth 医疗卷宗:传统决策的茔苑
这是一个代表的确企业级挑战的数据集。整整 60,000 个 token,塞满了多个患者复杂的病历、化验单和用药纪录,信息密度极高。
在这个测试中,工业界最爱用的「文本选录」透彻沦为笑柄 —— 它的准确率跌到了和「不提供任何高下文(No-Context)」一模一样的底线,意味着模子看了选录等于没看。
而 Attention Matching 则犹如战神附体,大幅罕见了系数传统权宜之策。
天然,Zweiger 也坦诚地给出了工程建议:「对于这种极高信息密度的任务,淌若你想保留系数细节,建议将压缩比调得蔼然一些(比如 10 倍或 20 倍),以换取皆备的精准度。」
3. AIME 2025 在线动态压缩:遨游中换引擎
最让东谈主推动激越的,是针对在线压缩的见解考证。靠近 AIME 顶级数学推理题,商议东谈主员锁死了物理内存上限。模子就像是在一个忐忑的笼子里进行尽头消耗脑力的打算。

每当内存爆满,系统就会短暂按下暂停键,用 Attention Matching 将其使命顾虑暴力压缩 50%,然后让模子接续念念考!即使在一次解题历程中,连气儿六次「切除」一半的顾虑,模子最终已经得胜找到了正确谜底,其透露与领有无尽内存的模子完全一致。
这对于 OpenClaw 这么需要万古刻运行、抑遏产生冗长器具调用日记的 Agent 来说,简直是救命稻草!
甚而,对于那些追求压缩率、对精度条目稍优容的场景,商议东谈主员还玩出了一种「200 倍压缩」的组合技:先让模子生成文本选录,然后再对选录的 KV Cache 进行 Attention Matching 压缩!最终在聊胜于无的显存占用下,达到了与纯选录一样的准确率。

结语
从开辟者自救到大厂标配的范式升沉?
天然,莫得任何魔法是莫得代价的。
必须指出的是,淌若你靠近的是极其复杂的数据,况兼非要追求 100 倍以上压缩,那么逐步的、基于梯度优化的 Cartridges 已经能在精度上险胜一筹,因为它能在更广博的潜空间中搜索最优解,而不受限于「从原始 Key 中挑选」的设定。
此外,这套神技现时还不是一个不错「无脑装配」的插件软件。正如 Zweiger 解释的那样:「潜空间压缩是一种模子层的本领。你必须领有走访模子权重的权限。」这意味着,淌若你完全依赖闭源的 API(比如径直调用 GPT-4 接口),你是无法我方完毕这套魔法的。企业要想享受这种显存摆脱,必须拥抱开源权重模子(如 Llama 3、Qwen 3)。
而且,要将这种潜空间 KV 压缩本领编织进当代极其复杂的商用推理引擎(那些早已布满了前缀缓存、变长内存打包等复杂妙技的系统)中,已经需要工程师们掉光不少头发。
但趋势已无可反抗。正如 Zweiger 所预言的:「咱们正在见证高下文压缩发生根人性的范式升沉 —— 它正从『企业我方拼凑的粗拙工程』,进化为『底层模子提供商内置的核火器』。比如 OpenAI 最近推出的黑盒压缩端点,复返的便是一个不透明的对象,而不是纯文本选录。」
当「注见识匹配」透彻融入 AI 基础设施的血液中时,显存瓶颈将被透彻击碎。到当时,像 OpenClaw 这么的智能体,也许确切或者以单机之躯,斟酌通盘寰宇的学问。
参考勾通体球网