Skip to content
AI News HubLIVE
Original source3 min read

Sparse Prefix Caching for Hybrid and Recurrent LLM Serving

Summary

A new paper introduces sparse prefix caching that leverages recurrent layers' ability to resume from a single state, optimizing LLM inference by storing exact recurrent states at sparse checkpoints. An O(NM) dynamic program determines optimal checkpoint placement, improving latency over standard heuristics on real-world tasks.

SourcearXiv Machine LearningAuthor: Mikhail Shirokikh, Sergey Nikolenko
Sparse Prefix Caching for Hybrid and Recurrent LLM Serving
Report an error

The correction channel is not available yet. You can copy the article reference below for later.

Correction instructions
Read article

[2605.05219] Sparse Prefix Caching for Hybrid and Recurrent LLM Serving

Skip to main content

Learn about arXiv becoming an independent nonprofit.

We gratefully acknowledge support from the Simons Foundation, member institutions, and all contributors. Donate

> cs > arXiv:2605.05219

Help | Advanced Search

All fields Title Author Abstract Comments Journal reference ACM classification MSC classification Report number arXiv identifier DOI ORCID arXiv author ID Help pages Full text

Search

GO

quick links

Login

Help Pages

About

-->

Computer Science > Machine Learning

arXiv:2605.05219 (cs)

[Submitted on 17 Apr 2026]

Title:Sparse Prefix Caching for Hybrid and Recurrent LLM Serving

Authors:Mikhail Shirokikh, Sergey Nikolenko

View a PDF of the paper titled Sparse Prefix Caching for Hybrid and Recurrent LLM Serving, by Mikhail Shirokikh and Sergey Nikolenko

View PDF HTML (experimental)

Abstract:Prefix caching is a key latency optimization for autoregressive LLM serving, yet existing systems assume dense per-token key/value reuse. State-space models change the structure of the problem: a recurrent layer can resume from a single stored state rather than requiring the entire token history. This asymmetry opens a new design point between no reuse and dense caching: store exact recurrent states at a sparse set of checkpoint positions and, on a cache hit, resume from the deepest stored checkpoint and recompute the remaining suffix exactly.

We formalize sparse prefix caching as checkpoint placement under a distribution over overlap depths, yielding an exact O(NM) dynamic program. For use cases where requests share a non-trivial prefix (e.g. asking different questions about a single long document), we show that our method consistently improves the Pareto frontier traced by standard heuristics on real-world data. Across QuALITY and System Prompts, distribution-aware placement dominates every fixed-budget baseline on the measured layer-group Pareto frontier and matches or outperforms the strongest heuristic (block caching) while typically using substantially fewer checkpoints, with the largest gains at low checkpoint budgets where the overlap distribution is most non-uniform. The method is most relevant when many requests share a substantial but not identical prefix within a retained cache entry. It preserves exact outputs, does not change the recurrent computation itself or require new recurrent update kernels, applies to recurrent/SSM layers whose hidden state can be extracted and restored exactly, and for hybrid models can be combined with existing KV-cache compression techniques.

Subjects:

Machine Learning (cs.LG); Artificial Intelligence (cs.AI)

Cite as: arXiv:2605.05219 [cs.LG]

(or arXiv:2605.05219v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2605.05219

Focus to learn more

arXiv-issued DOI via DataCite

Submission history

From: Mikhail Shirokikh [view email] [v1] Fri, 17 Apr 2026 09:24:58 UTC (744 KB)

Full-text links:

Access Paper:

View a PDF of the paper titled Sparse Prefix Caching for Hybrid and Recurrent LLM Serving, by Mikhail Shirokikh and Sergey Nikolenko

View PDF

HTML (experimental)

TeX Source

view license

Current browse context:

cs.LG

new | recent | 2026-05

Change to browse by:

cs cs.AI

References & Citations

NASA ADS

Google Scholar

Semantic Scholar

export BibTeX citation Loading...

BibTeX formatted citation

×

loading...

Data provided by:

Bookmark

Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle

Bibliographic Explorer (What is the Explorer?)

Connected Papers Toggle

Connected Papers (What is Connected Papers?)

Litmaps Toggle

Litmaps (What is Litmaps?)

scite.ai Toggle

scite Smart Citations (What are Smart Citations?)

Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle

alphaXiv (What is alphaXiv?)

Links to Code Toggle

CatalyzeX Code Finder for Papers (What is CatalyzeX?)

DagsHub Toggle

DagsHub (What is DagsHub?)

GotitPub Toggle

Gotit.pub (What is GotitPub?)

Huggingface Toggle

Hugging Face (What is Huggingface?)

ScienceCast Toggle

ScienceCast (What is ScienceCast?)

Demos

Demos

Replicate Toggle

Replicate (What is Replicate?)

Spaces Toggle

Hugging Face Spaces (What is Spaces?)

Spaces Toggle

TXYZ.AI (What is TXYZ.AI?)

Related Papers

Recommenders and Search Tools

Link to Influence Flower

Influence Flower (What are Influence Flowers?)

Core recommender toggle

CORE Recommender (What is CORE?)

IArxiv recommender toggle

IArxiv Recommender (What is IArxiv?)

Author

Venue

Institution

Topic

About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.

Which authors of this paper are endorsers? | Disable MathJax (What is MathJax?)

About

Help

contact arXivClick here to contact arXiv Contact

subscribe to arXiv mailingsClick here to subscribe Subscribe

Copyright

Privacy Policy

Web Accessibility Assistance

arXiv Operational Status

Key points and analysis

Article intelligence

EngineersAdvanced

Key points

  • Existing prefix caching assumes dense key/value reuse, but state-space models can resume from a single state.
  • Sparse caching stores recurrent states at selected checkpoints, recomputing the suffix on a cache hit.
  • Distribution-aware checkpoint placement via dynamic programming outperforms fixed-budget baselines.
  • Method preserves exact outputs and is compatible with existing KV-cache compression.

Highlights and analysis are generated automatically and may contain errors. Check the original source.

Sparse Prefix Caching for Hybrid and Recurrent LLM Serving | AI News Hub