arXiv AI

Fractional Decay KV-Cache: Ownership-Aware Memory Management for Improved Inference Relevancy in Dialog Systems

Fractional Decay KV-Cache (FD‑KVC) is a new memory‑management algorithm for transformer‑based dialog systems that uses a dual‑channel scoring mechanism—cumulative attention and recency‑weighted relevance—to keep key‑value pairs relevant as conversation topics shift. The method incorporates an ownership‑loss‑driven adaptive learning rate, runs entirely on CPU with minimal overhead, and outperforms the heavy‑hitter baseline H2O by significant margins across multiple dialog scenarios, including faster topic adaptation and higher topic diversity. Ablation studies confirm that each component of FD‑KVC contributes to its performance gains.

arXiv Computation and Language
2d ago

Reformulating KV Cache Eviction Problem for Long-Context LLM Inference

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.

By Tho Mai, Joo-Young Kim
Hugging Face Trending Papers
Oct 5

DeferKV: Rethinking Eviction Timing for One-Shot KV Cache Compression

DeferKV rethinks when to evict key‑value (KV) cache entries in long‑context large language models. By delaying eviction until the first decoding step and combining prompt‑side and decode‑side attention signals, it aligns KV importance estimation with actual generation needs. The method requires no extra training or modules and consistently improves performance on benchmarks while keeping latency low.

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