arXiv Computation and Language

RIFT: KV Cache Compression for Draft Models with Complementary Compressibility Training

RIFT introduces selection‑based KV cache compression strategies that balance prefill information with recently generated tokens to mitigate decode staleness in speculative decoding. By refreshing the KV cache selection during token generation and employing an adaptive refresh strategy, RIFT improves drafter quality while managing decode latency. A compressibility‑driven drafter training phase further enhances performance, enabling Qwen3.5‑4B to retain over 98% of its full‑cache acceptance length within an 8K‑token KV cache budget on long‑context inputs.

Hugging Face Trending Papers
Oct 22

RIFT: KV Cache Compression for Draft Models with Complementary Compressibility Training

The paper introduces RIFT, a set of selection‑based KV cache compression strategies that balance prefill information with recently generated tokens to mitigate decode staleness in draft models. RIFT improves drafter quality by refreshing KV cache selection during token generation, and an adaptive refresh strategy reduces score calculations to lower latency. Additionally, a compressibility‑driven drafter training phase using sparse attention masks further enhances performance, enabling Qwen3.5‑4B to maintain over 98% of its full‑cache acceptance length within an 8K‑token KV cache budget on long‑context inputs.

arXiv Computation and Language
Sep 10

Fine-Tuning a KV Cache Concatenation-Aware Model or Recomputing KV Caches? Why Not Both?

The paper addresses the challenge of long input contexts in Retrieval-Augmented Generation (RAG) systems, where concatenating many retrieved chunks increases prefill workload and time to first token (TTFT). It proposes a dual strategy: fine‑tuning the model to be aware of KV cache concatenation and selectively recomputing only part of the KV caches. Experiments on the RULER benchmark show that for a 124k‑token input, this combined method boosts the RULER score by 9.7 points over a baseline that recomputes caches only, while cutting TTFT by 80% compared with full attention.

By Fumihiko Tachibana, Daisuke Miyashita, Jun Deguchi
arXiv AI
Jul 8

FreqDepthKV: Frequency-Guided Depth Sharing for Robust KV Cache Compression in Long-Context LLM Inference

arXiv:2607. 06519v1 Announce Type: new Abstract: Long-context LLM inference is increasingly limited by the memory and bandwidth cost of KV caches, yet aggressive compression can remove the layer-specific evidence needed for retrieval and multi-step reasoning.

By Anna C\'ordoba, Adam Puente Tercero, Nerea Angulo Hijo, Mar Linares Tercero, Julia Barrientos, Ainhoa Miranda, Jes\'us Olivera
arXiv AI
Jul 8

Benchmarking KV-Cache Optimizations across Task Quality and System Performance for Long-Context Serving

arXiv:2607. 05399v1 Announce Type: cross Abstract: Large language model serving is increasingly limited by KV-cache growth under long-context workloads, yet existing KV-cache compression techniques are difficult to compare because they were evaluated on different models, tasks, budgets, and serving stacks.

By Nikita Agrawal, Ruben Mayer