MIT发布KV压缩新技术:推理提速百倍,将数小时的任务缩短至秒级
这项研究介绍了一种名为注意力匹配(Attention Matching)的新技术,旨在解决现代语言模型在大规模长文本处理中的KV缓存内存瓶颈。该方法通过引入标量偏置和最小二乘法优化,在显著缩减缓存大小的同时,能够精准模拟原始模型的注意力行为。与现有的令牌修剪或文本摘要等基线相比,该技术在极高压缩率下仍能保持更优的下游任务准确率。此外,该方案通过非均匀头部预算分配进一步提升效率,且无需对模型进行重新训练或微调。实验证明,该方法在处理长文档理解和复杂推理任务时,不仅速度比传统的梯度优化方法快两个数量级,且展现出极佳的推理稳定性。总之,这项研究为提升长周期智能体的运行效率提供了一种高性能且低成本的上下文压缩方案。