Year: 2025
Type: conference-paper
Abstract: Efficient KV cache management in LLMs is crucial for long-context tasks like RAG and summarization.Existing KV cache compression methods enforce a fixed pattern, neglecting task-specific characteristics and reducing the retention of essential information.However, we observe distinct activation patterns across layers in various tasks, highlighting the need for adaptive strategies tailored to each task's unique demands.Based on this insight, we propose DynamicKV, a method that dynamically optimizes token retention by adjusting the number of tokens retained at each layer to adapt to the specific task.DynamicKV establishes global and per-layer maximum KV cache budgets, temporarily retaining the maximum budget for the current layer, and periodically updating the KV cache sizes of all preceding layers during inference.Our method retains only 1.7% of the KV cache size while achieving ∼ 90% of the Full KV cache performance on LongBench.Notably, even under extreme compression (0.9%), DynamicKV surpasses state-of-the-art (SOTA) methods by 11% in the Needle-in-a-Haystack test using Mistral-7B-Instruct-v0.2.The code is available at repository https://github.com/DreamMr/ DynamicKV. KV Cache SizeLayer Step a. FullKV b.StreamingLLM/H2O/SnapKV c. PyramidKV d.
Language: English
Topic: Topic Modeling
Subfield: Artificial Intelligence
Field: Computer Science
Domain: Physical Sciences
Open Access status: hybrid
OpenAlex ID: W4416034925