arXiv Computation and Language By Tho Mai, Joo-Young Kim

Reformulating KV Cache Eviction Problem for Long-Context LLM Inference

Read the original on arXiv Computation and Language →

The paper proposes LaProx, a new KV cache eviction strategy for long‑context LLM inference that redefines the problem as an output‑aware, layer‑wise matrix multiplication approximation. Unlike prior methods that rely only on local attention weights, LaProx explicitly models the interaction between attention maps and projected value states, accounting for inter‑head dependencies to assign globally comparable token importance scores. Experiments on 19 datasets, including LongBench and Needle‑In‑A‑Haystack, show that LaProx retains model performance while using only 5% of the KV cache and reduces accuracy loss by up to 2× under extreme compression compared to state‑of‑the‑art baselines.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Computation and Language.

arXiv AI
Oct 16

DropKV: Decoupling Residual-Output Perturbation for Near-Optimal KV-Cache Eviction

DropKV introduces a new KV‑cache eviction framework that decouples residual‑output perturbation into independent per‑token scores, avoiding the NP‑hard combinatorial optimization of prior methods. The approach offers a provable constant‑factor approximation under a cone condition verified on three long‑context LLMs, and a high‑performance Triton‑based implementation that yields a 19.8× speedup in scoring kernels and up to 9.9% end‑to‑end prefill speedup. Benchmarks on RULER and LongBench show DropKV consistently outperforms state‑of‑the‑art baselines at the same cache budgets, achieving the lowest inference latency among eviction strategies while remaining faster than dynamic cache baselines.

By Aozhong Zhang, Selcuk Gurses, Yanxia Deng, Naigang Wang, Chi-Chun Liu, Davis Wertheimer, Derrick Liu, Xin Li, Zi Yang, Felix X. -F. Ye, Penghang Yin
arXiv AI
Jul 7

IndexMem: Learned KV-Cache Eviction with Latent Memory for Long-Context LLM Inference

arXiv:2605. 25475v2 Announce Type: replace-cross Abstract: Large Language Models (LLMs) are increasingly expected to operate over long contexts, yet standard softmax attention incurs a KV cache that grows linearly with sequence length, quickly becoming the bottleneck for long context inference.

By Xintong Yang, Hao Gu, Binxing Xu, Lujun Li, Bei Liu, Jiacheng Liu, Qiyuan Zhu, Yike Guo, Sirui Han
arXiv AI
Jun 24

CompressKV: Semantic-Retrieval-Guided KV-Cache Compression for Resource-Efficient Long-Context LLM Inference

arXiv:2606. 24467v1 Announce Type: new Abstract: Long-context large language model (LLM) inference is increasingly constrained by the memory footprint and decoding cost of key-value (KV) caches, limiting sustainable deployment on resource-constrained hardware.

By Xiaolin Lin, Jingcun Wang, Olga Kondrateva, Yiyu Shi, Bing Li, Grace Li Zhang
Hugging Face Trending Papers
Oct 15

DropKV: Decoupling Residual-Output Perturbation for Near-Optimal KV-Cache Eviction

DropKV introduces a new key-value cache eviction strategy for large language models that decouples the joint eviction decision into independent per-token scoring, thereby avoiding the NP-hard combinatorial problem of minimizing output perturbation. The method offers a provable constant-factor approximation guarantee under a cone condition verified on three long-context LLMs, and its high-performance implementation using fused Triton kernels achieves a 19.8× speedup in scoring and up to 9.9% end-to-end prefill speedup. Evaluations on RULER and LongBench show that DropKV outperforms state-of-the-art baselines at equivalent cache budgets, delivering the lowest inference latency among eviction methods while remaining faster than the dynamic cache baseline.