arXiv:2604.05887cs.AI2026-04ACL被引 4

通过分类型压缩策略,大幅降低多模态大模型推理内存占用

HybridKV: Hybrid KV Cache Compression for Efficient Multimodal Large Language Model Inference

论文配图:HybridKV: Hybrid KV Cache Compression for Efficient Multimodal Large Language Model Inference
图 1 · 摘自论文原文
  • 按注意力头特性分类,动态与静态头采用不同压缩方法
  • 最高减少7.9倍缓存内存,推理速度提升1.52倍
  • 适合高精度多模态推理场景的高效部署需求

多模态大语言模型在文本、图像和视频统一推理方面取得进展,但其推理受关键值(KV)缓存快速增长制约。每个视觉输入生成数千个标记,导致缓存随上下文长度线性增长,并在整个解码过程中驻留于GPU内存,造成极高内存开销和延迟,即使在高端GPU上亦然。常见解决方案是在不同粒度下以固定预算压缩缓存:标记级均匀丢弃不重要标记,层级跨层调整保留率,头级在注意力头间重新分配预算。然而这些方法仅关注分配,忽视了注意力头异构行为所需的差异化压缩策略。我们提出HybridKV,一种三阶段混合缓存压缩框架:首先利用以文本为中心的注意力将头分类为静态或动态;然后采用自顶向下的预算分配方案,分层分配KV预算;最后对静态头使用以文本优先的剪枝,对动态头采用分块检索压缩。在11个多模态基准测试中,基于Qwen2.5-VL-7B的实验表明,HybridKV将KV缓存内存最多减少7.9倍,解码速度提升1.52倍,性能几乎无损失,甚至相对全缓存模型更高。

原文摘要 · Abstract (English)

Multimodal Large Language Models (MLLMs) have advanced unified reasoning over text, images, and videos, but their inference is hindered by the rapid growth of key-value (KV) caches. Each visual input expands into thousands of tokens, causing caches to scale linearly with context length and remain resident in GPU memory throughout decoding, which leads to prohibitive memory overhead and latency even on high-end GPUs. A common solution is to compress caches under a fixed allocated budget at different granularities: token-level uniformly discards less important tokens, layer-level varies retention across layers, and head-level redistributes budgets across heads. Yet these approaches stop at allocation and overlook the heterogeneous behaviors of attention heads that require distinct compression strategies. We propose HybridKV, a hybrid KV cache compression framework that integrates complementary strategies in three stages: heads are first classified into static or dynamic types using text-centric attention; then a top-down budget allocation scheme hierarchically assigns KV budgets; finally, static heads are compressed by text-prior pruning and dynamic heads by chunk-wise retrieval. Experiments on 11 multimodal benchmarks with Qwen2.5-VL-7B show that HybridKV reduces KV cache memory by up to $7.9\times$ and achieves $1.52\times$ faster decoding, with almost no performance drop or even higher relative to the full-cache MLLM.

多模态缓存压缩推理加速KV缓存

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。