Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

RetentiveKV: State-Space Memory for Uncertainty-Aware Multimodal KV Cache Eviction

About

Multimodal Large Language Models face severe challenges in computational efficiency and memory consumption due to the substantial expansion of the visual KV cache when processing long visual contexts. Existing KV cache compression methods typically rely on the "persistence of importance" hypothesis to prune tokens. However, this approach proves fragile in multimodal settings due to two key issues: 1) Visual tokens display "deferred importance," initially exhibiting low salience but becoming pivotal during later decoding, which can lead to premature eviction. 2) Discrete pruning disrupts the inherent spatial continuity of visual cues. To address these challenges, we propose RetentiveKV, an entropy-driven KV cache optimization method that reformulates KV eviction from "discrete context truncation" to "continuous memory evolution" based on State Space Models. Our method leverages information entropy to quantify the information potential of low-attention tokens and integrates tokens scheduled for eviction into a continuous state space through entropy-guided state transitions, enabling their dynamic reactivation when semantic relevance arises during subsequent decoding. Extensive experiments on multimodal benchmarks demonstrate that RetentiveKV achieves 5.0 times KV cache compression and 1.5 times decoding acceleration.

Sihao Liu, YuFan Xiong, Zhonghua Jiang, Zhaode Wang, chengfei lv Shengyu Zhang• 2026

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringTextVQA
Accuracy61.4
1453
Massive Multi-discipline Multimodal UnderstandingMMMU
Accuracy42.45
216
Document Visual Question AnsweringDocVQA
Accuracy62.21
203
Multimodal EvaluationMMStar
Accuracy62.58
139
Mathematical Visual Question AnsweringMathVista
Accuracy62.6
87
Instruction FollowingALFRED
Accuracy16.34
57
Multimodal Conversational Question AnsweringMMCoQA
ROUGE-L33.1
21
Multimodal PerceptionBLINK
Accuracy64.22
21
Showing 8 of 8 rows

Other info

Follow for update