LinearKV Optimizes Caching for Hybrid LLMs

Yirui Liu, Ruoling Qi, Longwen Wang, Xuaner Wu, Jian Chen, Yuxin Jin, Jiawei Shao, Xuelong Li· August 13, 2026 View original

Key takeaways

  • LinearKV enables position-independent caching for hybrid LLMs.
  • A single cached state is sufficient for linear layers, simplifying caching.
  • LinearKV significantly reduces time-to-first-token for hybrid LLMs.
  • It outperforms more complex exact composition methods on certain architectures.

Who benefits

Cloud ComputingAI InfrastructureSoftware DevelopmentTelecommunicationsFintech

Summary

LinearKV is a new training-free framework that enables position-independent caching (PIC) in hybrid LLMs, which combine attention layers with linear recurrences. It demonstrates that a single cached state is sufficient for linear layers, significantly cutting time-to-first-token and outperforming more complex exact composition methods on certain architectures.

Large Language Model (LLM) serving performance is increasingly enhanced by position-independent caching (PIC), which reuses token chunks. However, existing PIC methods are designed for full-attention models and struggle with "hybrid LLMs" that incorporate linear recurrences, as these models expose only a fixed-size state rather than a token-indexed KV cache. This paper introduces LinearKV, a novel training-free framework that brings PIC benefits to hybrid models. The core innovation of LinearKV is "decoupled initialization." For linear layers, it maps multiple matched local states to a single initial state, while full-attention layers continue to concatenate their KV as before. Crucially, the research finds that a *single cached state* is sufficient as the linear layer's initializer. This simpler approach is not only more efficient but also, on some architectures like Mamba-2, significantly outperforms the algebraically principled but more complex method of composing all cached states into an exact full-prefix state. LinearKV significantly cuts the time-to-first-token, achieving 0.46 times the full prefill time, compared to a 5-17% overhead for exact composition. These performance gains and quality preservation were demonstrated across various hybrid models and PIC selectors on benchmarks like LongBench QA and RULER, highlighting its practical value for accelerating hybrid LLM serving.

Why it matters

For professionals involved in deploying and serving LLMs, optimizing inference speed and efficiency is paramount. LinearKV offers a significant advancement for hybrid LLMs, enabling faster response times and reduced computational costs without requiring retraining, directly impacting user experience and operational expenses.

How to implement this in your domain

  1. 1Evaluate existing LLM serving infrastructure for opportunities to integrate position-independent caching (PIC) for hybrid models.
  2. 2Explore implementing the LinearKV framework to leverage single-cached-state initialization for linear layers in hybrid LLMs.
  3. 3Benchmark the performance of LinearKV against current caching strategies to quantify improvements in time-to-first-token and overall throughput.
  4. 4Consider adopting hybrid LLM architectures that can benefit from LinearKV for applications requiring low-latency inference.
  5. 5Collaborate with research teams or open-source communities to integrate and optimize LinearKV in production environments.

Original post by Yirui Liu, Ruoling Qi, Longwen Wang, Xuaner Wu, Jian Chen, Yuxin Jin, Jiawei Shao, Xuelong Li

"arXiv:2608.11231v1 Announce Type: new Abstract: LLM serving is increasingly accelerated by position-independent caching (PIC). Existing PIC methods, however, are built for full-attention models, where a token-indexed KV cache underlies its core operations: matching reusable token…"

View on X

Originally posted by Yirui Liu, Ruoling Qi, Longwen Wang, Xuaner Wu, Jian Chen, Yuxin Jin, Jiawei Shao, Xuelong Li on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses