← 피드로
[Submitted on 14 Aug 2025 (v1), last revised 10 Jul 2026 (this version, v2)]
Abstract:The growing sequence length of large language models poses significant challenges for key-value (KV) caches. Existing state-of-the-art cache eviction methods primarily analyze the inference behavior of attention heads in successful retrieval-reasoning cases, often overlooking diverse behaviors in failure cases, such as bias and distraction. This oversight limits the potential to leverage heterogeneous head behaviors for improved eviction performance. Inspired by the confusion matrix, we introduce an Attention Behavior Matrix to comprehensively analyze attention head behaviors in both success and failure scenarios. By maximizing the signal-to-noise ratio — strengthening valid reasoning pathways in success cases while inhibiting noise from bias and distraction in failure cases — we propose REtrieval-reAsoning and Logic-constructed (REAL) KV cache eviction, the first method to leverage multi-behavior analysis. Comprehensive evaluations show that REAL achieves remarkable performance across various models and benchmarks; notably, on LongBench v2, it achieves comparable accuracy to the strongest baseline, HeadKV-R2, while requiring 32x less space (Figure 1). By offering a novel perspective on behavior analysis, we pave the way for a shift from success-only to comprehensive, failure-aware methods in long-context modeling. Our code is available at this https URL.
Submission history
From: Mengjie Li [view email]
[v1]
Thu, 14 Aug 2025 14:08:58 UTC (992 KB)
[v2]
Fri, 10 Jul 2026 13:38:08 UTC (1,079 KB)
추출 본문 · 출처: arxiv.org · https://arxiv.org/abs/2508.15806
답글 남기기