AI & ML
HiState: Hierarchical Prefix Cache Management for Hybrid LLM Serving in Agentic Workloads
arXiv cs.DC0 views
arXiv:2610.41508v1 Announce Type: new Abstract: Hybrid large language models combine full-attention and recurrent linear layers to reduce inference costs. Prefix caching can further reduce redundant computation across requests, but a recurrent checkpoint cannot recover an earlier state corresponding to a shorter prefix. Existing hybrid prefix caching systems retain checkpoints at selected boundaries to support common reuse patterns in conventional conversations. However, agent workloads modify and revisit earlier contexts, exposing reusable prefixes whose checkpoints were never saved or have already been evicted. A straightforward approach is to retain more checkpoints to reduce recomputation, but their cumulative footprint can displace reusable prefixes from other sessions and cause additional cache misses. To address these problems, we present HiState, a prefix cache management system for hybrid models serving agent workloads. Our key observation is that prefix reuse accesses all matching key-value (KV) entries but only one recurrent checkpoint, regardless of how many checkpoints are retained. This asymmetry allows HiState to retain historical checkpoints on SSD while prioritizing fast memory for KV entries: each hit retrieves only one checkpoint whose size is independent of context length. Within this hierarchy, we propose adaptive interval control to balance checkpoint storage against recomputation, and unified eviction to coordinate checkpoint sparsification with reclamation of stale trajectories. We implement HiState on top of vLLM and evaluate it on two public coding-agent trace datasets and controlled microbenchmarks. On real agent traces, HiState achieves up to 5.73x and 5.44x speedups in p99 TTFT over vLLM-Marconi and SGLang-HiCache, respectively.
Source:https://arxiv.org/abs/2610.41508PreviousRecIS: Unifying Sparse and Dense Computation for Efficient Recommendation TrainingNextMemX: Memory-Aware Automatic Parallelism for Cross-Vendor Heterogeneous GPU Training
Related posts
How to Calculate In-Hand Salary from CTC in India (Budget FY 2026-27 Breakdown)
0dev.to
Amp Spaces: One Agent to Coordinate Them All — What Changes When You Stop Managing Threads
0dev.to
AI Agents Now Design and Run Your Robot-Powered Lab. The Molecules They Engineer Deserve Better Than a Flat Screen.
0dev.to

When AI Does All the Work, What Is the Meaning of Human Life? Beyond the Wage-Labor Identity
0dev.to
Three traps when an AI assistant reads congressional stock trade disclosures
0dev.to
Students Can Get Zed Pro Free for 1 Year
0dev.to