AI 快讯
编译自 marktechpost #KV cache压缩#模型优化#长上下文
KV Cache压缩竞赛:TurboQuant、OSCAR与EpiCache三大方案对比解析
长上下文LLM的KV cache内存瓶颈日益严重,TurboQuant、OSCAR和EpiCache分别从理论最优量化、可部署INT2压缩和对话记忆管理三个方向给出解决方案。本文详解三者原理、性能差异及互补性,帮助中文开发者选择适合自身场景的压缩策略。
一句话看懂
长上下文LLM的KV cache内存已超过模型权重,TurboQuant、OSCAR和EpiCache分别从理论最优量化、可部署INT2压缩和对话记忆管理三个方向破解瓶颈,三者互补而非竞争。
详细发生了什么
长上下文大语言模型面临一个与模型权重无关的内存瓶颈:解码时Transformer会缓存每一层每个token的key和value向量(KV cache),避免重复计算注意力。这个缓存随序列长度和batch size线性增长,在高并发长上下文场景下远超模型本身的内存占用。以Llama-3.1-70B BF16为例,每个token的KV cache约0.31 MB,128K上下文时约40 GB,1M上下文时超过300 GB——而模型权重仅140 GB。更糟的是,每个新解码的token都需要将整个缓存从高带宽内存(HBM)中流式读取,使解码受限于内存带宽而非计算。
2026年的三项工作从不同方向攻坚超低位宽量化:
- TurboQuant(Google & NYU,ICLR 2026):数据无关、理论最优。通过随机旋转使坐标近似独立高斯分布,再用预计算的Lloyd-Max标量量化器逐坐标量化,最后用1-bit QJL变换处理残差。在3-4 bit区间实现近乎无损的压缩,且无需校准数据。
- OSCAR(Together AI):注意力感知、可部署。通过离线校准计算注意力感知的旋转矩阵,将key旋转到query协方差的本征基,value旋转到分数加权协方差本征基。采用混合精度分页缓存(sink和近期token保留BF16,历史压缩至INT2),在128K上下文仅约0.24%的token保持BF16。在Qwen3-8B上有效2.28 bits下与BF16差距仅1.42点,在GLM-4.7-FP8上完全匹配BF16。
- EpiCache(Apple):面向多轮对话。通过分块预填充、语义片段聚类、片段匹配检索和自适应逐层预算分配,在LongMemEval等基准上实现高达40%的准确率提升,4-6倍压缩下接近全缓存准确率,峰值内存降低3.5倍。
中文圈视角
这三项技术对中文开发者有直接参考价值:
- OSCAR已预计算了Qwen3-4B/8B/32B、GLM-4.7-FP8等模型的旋转矩阵,并集成SGLang,国内使用Qwen系列或GLM的用户可直接部署,无需重新校准。对于部署在国产GPU(如昇腾、寒武纪)上的场景,OSCAR的INT2压缩可显著降低显存需求,但需注意其Triton kernel对国产硬件的兼容性。
- TurboQuant的模型无关特性使其适用于任何开源模型,包括中文社区常用的Qwen、Yi、DeepSeek等。其3-4 bit近无损压缩适合对精度要求高的场景(如金融、医疗文档分析),但需注意其“8倍注意力加速”仅针对微基准测试,实际收益需自行评估。
- EpiCache针对多轮对话的优化对中文聊天机器人、客服系统等场景意义重大。目前中文社区对对话级KV cache管理讨论较少,EpiCache的“语义片段”思路与中文对话的上下文切换特点(如话题跳跃)可能产生新的优化空间。
- 三者均可组合使用:例如先用OSCAR或TurboQuant压缩单次上下文的KV cache,再叠加EpiCache管理多轮对话历史,实现叠加收益。
几条值得记住的细节
- TurboQuant在3-4 bit区间实现近乎无损,理论失真仅比信息论下限高约2.7倍,无需校准数据。
- OSCAR在128K上下文下有效位宽2.28 bits,Qwen3-32B上精度与BF16仅差0.02点,GLM-4.7-FP8上完全匹配。
- OSCAR在100K上下文下实现高达7.83倍任务吞吐量和约8倍KV cache内存缩减,解码速度提升约3倍。
- EpiCache在LongMemEval等基准上相比token驱逐方法准确率提升40%,峰值内存降低3.5倍,延迟降低约2.4倍。
- 三者均可组合:TurboQuant或OSCAR负责量化压缩,EpiCache负责对话级缓存管理,实现叠加收益。
一句话总结
根据你的位宽预算、模型可移植性和对话长度需求,选择或组合TurboQuant、OSCAR和EpiCache,可大幅降低长上下文LLM的部署成本。