高效模型与推理加速 · CVPR2026 ·2026.06.03

重访多模态 KV 缓存压缩:一种频域引导的离群 KV 感知方法

Revisiting Multimodal KV Cache Compression: A Frequency-Domain-Guided Outlier-KV-Aware Approach

Yaoxin Yang、Peng Ye、Xudong Tan、Chongjun Tu、Maosen Zhao、Jia Hao 等 7 人

与你有关的部分

对做多模态大模型视觉侧推理的人来说,这篇给出了一个绕开注意力分数、直接对 KV 矩阵做频域主成分建模来识别并保留离群 KV 的压缩思路,且与 FlashAttention 等高效注意力核兼容,可直接用于视觉 token 极长的多图/视频输入场景降低解码延迟与显存。

原文依据 achieving up to 1.69x faster decoding with 80% lower KV memory usage while maintaining task performance
可迁移到 多模态大模型推理加速视觉 token 压缩长视觉序列建模

摘要译文

由模型翻译 · 术语统一 · 请对照原文核对关键结论

多模态大语言模型面临显著的推理开销,因为多模态 KV 缓存会随视觉输入长度成比例增长。现有多模态 KV 缓存压缩方法大多依赖注意力分数来缩减缓存规模,这使其与成熟的高效注意力核(如 FlashAttention)不兼容,并且忽略了值向量对注意力输出的贡献。本文从 KV 矩阵分布的角度重新审视多模态 KV 缓存压缩。首先,我们观察到多模态 KV 矩阵的频域能量主要集中于低频,并通过低通滤波器提取该主能量。进一步地,我们发现移除显著偏离该主能量的 KV 对会导致性能明显下降,我们将这类 KV 定义为离群 KV。考虑到离群 KV 更可能编码对推理至关重要的特征,我们提出 FlashCache,一个频域引导、离群 KV 感知的 KV 缓存压缩框架。首先,我们引入离群 KV 识别模块,在频域中对多模态 KV 矩阵的主成分建模,并优先保留显著偏离该主成分的 KV 对。此外,设计了动态预算分配模块,以自适应地确定每层 KV 缓存大小,从而保留更多离群 KV。在多个 MLLM 与基准上的实验表明,FlashCache 优于最优的多模态 KV 压缩方法,在保持任务性能的同时实现最高 1.69 倍的解码加速与 80% 的 KV 显存占用降低。

英文原文摘要 1654 字符

Multimodal large language models suffer from substantial inference overhead since multimodal KV Cache grows proportionally with the visual input length. Existing multimodal KV cache compression methods mostly rely on attention score to reduce cache size, which makes them are incompatible with established efficient attention kernels (e.g., FlashAttention) and ignores the contribution of value vectors to the attention output. In this work, we revisit multimodal KV cache compression from the perspective of the KV matrices' distribution. First, we observe that frequency-domain energy of multimodal KV matrices is predominantly concentrated in low-frequency and extract this principal energy via a low-pass filter. Further, we find that removing KV pairs that deviate substantially from this principal energy leads to a pronounced performance drop, which we define as Outlier KVs. Considering Outlier KVs are more likely to encode features critical for inference, we propose FlashCache, a frequency-domain-guided, Outlier-KV-aware KV Cache compression framework. First, we introduce an Outlier KV Recognition Module that models the principal component of multimodal KV matrices in the frequency domain and preferentially retains KV pairs that significantly deviate from it. Furthermore, Dynamic Budget Allocation Module is designed to adaptively determine the per-layer KV cache size to retain more Outlier KVs. Experiments on multiple MLLMs and benchmarks demonstrate that FlashCache outperforms state-of-the-art multimoal KV compression methods, achieving up to 1.69x faster decoding with 80% lower KV memory usage while maintaining task performance.

标签

会议 CVPR2026
页码 39550-39560
Bibkey
来源 CVF Open Access