arXiv Machine Learning

Kernel-based Interpolatory Methods for Context Compression of Attention Mechanisms

The paper introduces two new kernel‑based methods, GAIM and Q‑DAIM, for compressing the key‑value cache in attention mechanisms by formulating the problem in a Hilbert space and applying empirical interpolation techniques. To reduce the quadratic cost of these methods, the authors propose surrogate models—including Taylor series, Performer, and Random Fourier Features—that enable linear‑time compression. The work also extends to streaming attention, offering node‑selection penalties and interpolation updates, and demonstrates competitive performance on the LongBench‑E benchmark.

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
Jun 10

Blurry Window Attention

arXiv:2606. 09862v1 Announce Type: cross Abstract: The Softmax Attention operation in Transformer language models has a quadratic complexity in the sequence length and a growing state size in the form of KV cache, which becomes a bottleneck in long context scenarios.

By Axel Laborieux, Christos Sourmpis, Juan Gabriel Kostelec, Qinghai Guo
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.

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