这是论文清单仓库而非可运行工具,没有安装步骤;新增论文需填 generate.py 后运行 python generate.py 生成表格行;部分条目只给论文链接,代码仓库未必存在
Awesome-KV-Cache-Optimization
jjiantong活跃维护系统梳理大模型推理中 KV Cache 优化的综述仓库,按系统与架构视角归纳显存占用和吞吐瓶颈的各类优化方法,适合做推理加速选型时查阅。
[ACL 2026] Towards Efficient Large Language Model Serving: A Survey on System-Aware KV Cache Optimization
README 里没有找到可直接复制的安装命令,请到 README 或官方文档查看安装方式。
README 未说明
ACL 2026 综述配套的 KV Cache 优化论文库,按时间、空间、结构三类整理,帮你快速定位相关工作
系统梳理大模型推理中 KV Cache 优化的综述仓库,按系统与架构视角归纳显存占用和吞吐瓶颈的各类优化方法,适合做推理加速选型时查阅。
核心能力
- 按 Temporal/Spatial/Structural 三维分类组织 KV Cache 优化文献
- 覆盖调度、流水线重叠、硬件感知执行等时间维度方法
- 收录内存层级与计算设备间的 KV 编排与迁移工作
- 汇总 KV 压缩与保留管理等结构维度技术
- 附工具、模拟器与基准测试资源章节
容易踩的坑
暂无 release 数据(可能没有发版,或尚未同步)。
用户点评
登录用户的真实使用记录,可按版本和硬件筛选还没有人点评。第一个说出真实体验的人最有价值。
[ACL Anthology] [arXiv] [Website] [LinkedIn] [机器之心]
This repository accompanies our survey paper (ACL 2026 Findings):
Towards Efficient Large Language Model Serving: A Survey on System-Aware KV Cache Optimization
*Jiantong Jiang1, Peiyu Yang1*, Rui Zhang2, Feng Liu1
1The University of Melbourne, 2Huazhong University of Science and Technology
This repository maintains a continuously updated collection of system-aware, serving-time, KV-centric optimization methods that improve system metrics without retraining or architecture modification, a scope we call sKis. We organize the literature using the system behavior-oriented taxonomy introduced in our survey:
🔷 Temporal — when is KV cache accessed or computed?
🔷 Spatial — where is KV cache placed and migrated?
🔷 Structural — how is KV cache represented and managed?
📚 The taxonomy serves as a stable organizing framework, while the paper collection below is continuously updated as new work appears.
🧠 Grounded in this taxonomy, we analyze cross-behavior co-design affinity and behavior–objective effects, revealing overlooked regions and concrete open challenges.
News
- 2026-10: 🔥 63 new papers were added.
- 2026-09: 🧰 Added a new Tools, Simulators & Benchmarking Resources section. Many thanks to @Naloam for the contribution!
- 2026-08: 🌐 Launched our new website with an interactive paper explorer. Huge thanks to @ryan-utopia for making this possible!
- 2026-06: 🔥 51 new papers were added.
- 2026-04: 🎉 Our survey was accepted to ACL 2026.
- 2026-01: 📰 Our survey was featured by 机器之心.
- 2025-11: ⭐ 100-star milestone! Thanks to the amazing support from the community!
- 2025-10: 📝 The survey was available on TechRxiv.
Contributing
The repository is still under active development and will be updated regularly.
🙋 If you would like to include your paper in this survey and repository, please feel free to submit a pull request. You can generate the markdown row for each paper by filling in the first part of generate.py and running python generate.py. Alternatively, you can open an issue with the paper's title and a brief summary highlighting its key techniques. You can also contact us via email.
🙋🏻♀️ Please let us know if you find any metadata, venue, classification, or link errors, or have suggestions for the repository. We greatly appreciate your feedback!
🌟 If you find this resource helpful for your work, please consider giving us a star and citing our research.
Quick Index
- Temporal — Execution & Scheduling
- Spatial — Placement & Migration
- Structural — Representation & Retention
- Insights from the ACL 2026 Survey
- Tools, Simulators & Benchmarking Resources
Temporal — Execution & Scheduling
These methods act on when KV data is executed, computed, or scheduled to improve latency and throughput. We divide these methods into three categories: KV-centric scheduling, pipelining & overlapping, and hardware-aware execution.
KV-Centric Scheduling
KV-centric scheduling methods explicitly integrate KV characteristics into runtime scheduling decisions.
| Paper | Type | Code |
|---|---|---|
| Agent-aware KV scheduling under tool-call stalls; Also belongs to memory hierarchy KV orchestration | ||
| Mixed-fidelity KV execution and runtime scheduling | ||
| Fathom: Per-Query Read Depth for Sparse Decoding over Offloaded KV Caches [Link] Vivek Kalyanarangan | Token-level attention compute scheduling; per-query, per-channel read depth over a bit-plane K store; Also belongs to KV cache compression, quantization (structural) | |
| Prefix-affinity request routing with load-aware cache locality | ||
| KV reuse- and load-aware request scheduling via multiplicative routing score | ||
| Cache-loading-aware request scheduling; Also belongs to Memory Hierarchy KV Orchestration (spatial) | ||
| Draft- and target-guided token-level KV selection for sparse speculative verification | ||
| Binary-quantized proxy scoring for adaptive Top-p KV retrieval in sparse attention | ||
| KV block- and compression-state-aware hybrid request scheduling; Also belongs to KV cache retention management (structural) | ||
| KV-length-aware request batching and batch-level scheduling | ||
| Shared KV access scheduling across requests | ||
| Hotness-aware KV cache scheduling for prefix sharing; Also belongs to KV cache retention management | ||
| PRISM: Fast Online LLM Serving via Scheduling-Memory Co-design [Link] Xingyu Qu, Tianhao Lin, Yiqi Li, Zhiyu Chen, Sheng Wang | KV reuse-aware request scheduling; Also belongs to KV cache retention management (structural) | |
| KV-loading-aware request scheduling; Also belongs to Memory hierarchy KV orchestration (spatial) | ||
| KV usage-aware request-level scheduling | ||
| Token-level attention compute scheduling | ||
| Query-aware page-level KV selection for sparse attention computation | ||
| Token-level attention compute scheduling | ||
| Token-level attention compute scheduling | ||
| Kernel-level workload scheduling across CUDA thread blocks; Also belongs to allocation & reuse (structural) | ||
| KV reuse-aware request-level scheduling | ||
| KV reuse-aware request-level scheduling; Also belongs to HW-aware execution | ||
| Token-level attention compute scheduling | ||
| KV reuse-aware request-level scheduling; Also belongs to allocation & reuse (structural) | ||
| KV usage-aware request-level scheduling | ||
| LayerKV: Optimizing Large Language Model Serving with Layer-wise KV Cache Management [Link] Yi Xiong, Hao Wu, Changxu Shao, Ziqing Wang, Rui Zhang, Yuhong Guo, Junping Zhao, Ke Zhang, Zhenxuan Pan | KV usage-aware request-level scheduling; Also belongs to memory hierarchy KV orchestration (spatial) | |
| KV-state-aware request scheduling over preserve, swap, and recompute decisions; Also belongs to memory hierarchy KV orchestration (spatial) | ||
| Token-level attention compute scheduling | ||
| Token-level attention compute scheduling | ||
| KV usage-aware request-level scheduling; Also belongs to HW-aware execution | ||
| Inference without interference: Disaggregate LLM inference for mixed downstream workloads [Link] Cunchen Hu, Heyang Huang, Liangliang Xu, Xusheng Chen, Jiang Xu, Shuang Chen, Hao Feng, Chenxi Wang, Sa Wang, Yungang Bao, Ninghui Sun, Yizhou Shan | KV usage-aware request-level scheduling; Also belongs to HW-aware execution |
Pipelining & Overlapping
Pipelining and overlapping methods hide latency by concurrently executing KV-related compute, communication, and I/O. Though often embedded in the broader systems, we highlight the following methods where pipelining and overlapping forms the core technical contribution.
| Paper | Type | Code |
|---|---|---|
| KV prefetch (CPU→GPU) ↔ GPU decode compute; Also belongs to memory hierarchy KV orchestration (spatial) | ||
| Layer-ahead CPU attention precompute ↔ GPU decode; Also belongs to HW-aware execution | ||
| KV prefetch ↔ sparse-attention indexer computation; Also belongs to memory hierarchy KV orchestration (spatial) | ||
| Asynchronous proxy-index construction and attention execution with lazy updates; Also belongs to KV-centric scheduling (temporal) | ||
| ObjectCache: Layerwise Object-Storage Retrieval for KV Cache Reuse [Link] Yu Zhu, Aditya Dhakal, Yunming Xiao, Dejan Milojicic, Gustavo Alonso | Object-storage KV retrieval ↔ GPU layerwise consumption; Also belongs to memory hierarchy KV orchestration (spatial) | |
| Critical-layer KV recomputation ↔ remote reused-KV loading; Also belongs to compute device KV orchestration (spatial) | ||
| Uncached-token KV recomputation ↔ decoding/communication via fine-grained adaptive caching; Also belongs to KV cache retention management (structural) | ||
| GPU compute ↔ GPU KV prefetch (CXL memory→L2); Also belongs to memory hierarchy KV orchestration (spatial) | ||
| KV prefetch/offload (CPU↔GPU) ↔ FFN & QKV generation; Also belongs to memory hierarchy KV orchestration (spatial) | ||
| GPU KV prefetch (HBM→L2) ↔ GPU attention compute; Also belongs to memory hierarchy KV orchestration (spatial) | ||
| GPU KV recompute ↔ KV transfer (CPU↔GPU) | ||
| CPU attention compute ↔ GPU linear ops; Also belongs to HW-aware execution | ||
| Selective KV recomputation ↔ cached-KV retrieval for non-prefix RAG chunks; Also belongs to KV cache retention management (structural) | ||
| KV state restoration compute ↔ host-storage I/O via bubble-free scheduling | ||
| PRESERVE: Prefetching Model Weights and KV-Cache in Distributed LLM Serving [Link] Ahmet Caner Yüzügüler, Jiawei Zhuang, Lukas Cavigelli | GPU KV prefetch (HBM→L2) ↔ GPU collective communication; Also belongs to memory hierarchy KV orchestration (spatial) | |
| CPU MHSA compute ↔ FFN data transfer (CPU→GPU); Also belongs to HW-aware execution | ||
| CPU R-part compute ↔ GPU S-part compute; Also belongs to HW-aware execution | ||
| KV load/store (CPU↔GPU) ↔ GPU compute; Also belongs to memory hierarchy KV orchestration (spatial) |
Hardware-aware Execution
Hardware-aware execution methods adapt KV cache-related operations to the underlying heterogeneous hardware.
Disaggregated Inference
Disaggregated inference separates heterogeneous computation in LLM inference and maps them to distinct hardware resources to reduce interference and improve utilization.
| Paper | Type | Code |
|---|---|---|
| CPU/GPU collaborative attention; Also belongs to pipelining & overlapping | ||
| HillInfer: Efficient Long-Context LLM Inference on the Edge with Hierarchical KV Eviction using SmartSSD [Link] He Sun, Shinan Liu, Li Li, Mingjun Xiao | SmartSSD-assisted KV offloading; Also belongs to memory hierarchy KV orchestration (spatial) | |
| PD disaggregation; Also belongs to KV-centric scheduling | ||
| Decoupling prefill and decode to different GPUs; Also belongs to memory hierarchy KV orchestration (spatial) | ||
| Colocating PD jobs of multiple LLMs within each GPU via SM partitioning; Also belongs to KV-centric scheduling | ||
| Decoupling prefill and decode to different GPUs; Also belongs to compute device KV orchestration (spatial) | ||
| Decoupling prefill and decode to different GPUs; Also belongs to compute device KV orchestration (spatial) | Splitwise (integrated into vLLM) | |
| Inference without interference: Disaggregate LLM inference for mixed downstream workloads [Link] Cunchen Hu, Heyang Huang, Liangliang Xu, Xusheng Chen, Jiang Xu, Shuang Chen, Hao Feng, Chenxi Wang, Sa Wang, Yungang Bao, Ninghui Sun, Yizhou Shan | Decoupling prefill and decode to different GPUs; Also belongs to KV-centric scheduling | |
| Infinite-LLM: Efficient LLM Service for Long Context with DistAttention and Distributed KVCache [Link] Bin Lin, Chen Zhang, Tao Peng, Hanyu Zhao, Wencong Xiao, Minmin Sun, Anmin Liu, Zhipeng Zhang, Lanbo Li, Xiafei Qiu, Shen Li, Zhigang Ji, Tao Xie, Yong Li, Wei Lin | Disaggregating at the operator level; Also belongs to compute device KV orchestration (spatial) |
Compute Offloading
Compute offloading relocates partial compute to auxiliary devices to reduce GPU bottlenecks, utilizing hardware heterogeneity and workload features.
| Paper | Type | Code |
|---|---|---|
| In-network query broadcast and attention aggregation offloaded to programmable switches/NICs; Also belongs to compute device KV orchestration (spatial) | ||
| Near-storage attention offloading to SmartSSD accelerators; Also belongs to compute device KV orchestration (spatial) | ||
| CPU offloading (attention and KV caches); Also belongs to pipelining & overlapping | ||
| CPU offloading (attention and retrieval) | ||
| PIM-based offloading | ||
| CPU offloading with dynamic GPU-CPU division; Also belongs to pipelining & overlapping | ||
| CPU offloading | ||
| InstInfer: In-Storage Attention Offloading for Cost-Effective Long-Context LLM Inference [Link] Xiurui Pan, Endian Li, Qiao Li, Shengwen Liang, Yizhou Shan, Ke Zhou, Yingwei Luo, Xiaolin Wang, Jie Zhang | CSD Offloading; Also belongs to compute device KV orchestration (spatial) | |
| CPU offloading (attention and KV caches); Also belongs to pipelining & overlapping | ||
| PIM-based offloading; Also belongs to compute device KV orchestration (spatial) |
Hardware-Specialized Execution
[Newly added] These methods specialize KV-related kernels, data layouts, or dedicated execution engines to exploit hardware-native compute primitives and access patterns.
| Paper | Type | Code |
|---|---|---|
| Prefix-aware multi-tile attention kernel that co-schedules shared-prefix queries to reduce redundant KV reads | ||
| SW-HW co-designed dynamic KV retrieval engine for streaming video LLM inference; Also belongs to memory hierarchy KV orchestration | ||
| CUDA/Tensor-Core cooperative execution for efficient low-bit KV cache decoding |
Spatial — Placement & Migration
These works optimize where KV data is stored or transferred to balance memory and I/O pressure. We divide these methods into two categories: memory hierarchy KV orchestration, and compute device KV orchestration.
Memory Hierarchy KV Orchestration
Memory hierarchy KV orchestration methods distribute KV caches across memory hierarchies.
Cross-device Memory Hierarchy
These methods manage KV caches across fast but limited GPU HBM, and larger but slower alternatives like CPU DRAM or SSD.
| Paper | Type | Code |
|---|---|---|
| Proactive KV offload/upload for tool-call stalls; Also belongs to KV-centric scheduling (temporal) | ||
| Fine-grained adaptive GPU–CPU KV placement reconfiguration under latency SLOs | ||
| Dual-path KV loading from external storage; Also belongs to compute device KV orchestration | ||
| Predictive host-to-GPU KV prefetching; Also belongs to pipelining & overlapping (temporal) | ||
| CXL-hybrid-memory KV tiering with request-level prefix prefetching | ||
| Dynamic host–GPU KV offloading with lossless prefetching; Also belongs to pipelining & overlapping (temporal) | ||
| Zero-copy CPU-resident KV offloading over NVLink-C2C | ||
| Hierarchical KV caching across GPU, CPU, and SSD with cache-aware request scheduling; Also belongs to KV-centric scheduling (temporal) | ||
GPU-native top-k KV retrieval from CPU-offloaded caches | ||
| Asynchronous hierarchical KV retrieval with head heterogeneity | ||
| NVMe-direct KV offloading on memory-constrained edge devices | ||
| CPU/DRAM/SSD multi-tier KV placement and migration | ||
| System cost-driven decision | ||
| Head-stability-aware GPU–CPU KV placement with periodic top-page promotion | ||
| ObjectCache: Layerwise Object-Storage Retrieval for KV Cache Reuse [Link] Yu Zhu, Aditya Dhakal, Yunming Xiao, Dejan Milojicic, Gustavo Alonso | Layerwise KV placement and retrieval from object storage; Also belongs to pipelining & overlapping (temporal) | |
| Tutti: Making SSD-Backed KV Cache Practical for Long-Context LLM Serving [Link] Shi Qiu, Yifan Hu, Xintao Wang, Wenhao Zhu, Jianqin Yan, Hao Chen, Kaiqiang Xu, Kai Chen, Yiming Zhang | SSD-backed KV placement with GPU-friendly retrieval layout | |
| Disaggregated KV cache storage with advisory prefetching and cooperative GPU-memory management | ||
| System cost-driven decision; Also belongs to compute device KV orchestration | ||
| Importance-aware methods | ||
Speculative top-k KV retrieval | ||
| TTKV: Temporal-Tiered KV Cache for Long-Context LLM Inference [Link] Gradwell Dzikanyanga, Weihao Yang, Hao Huang, Donglei Wu, Shihao Wang, Wen Xia, Sanjeeb K C | Temporal-tiered KV placement with heterogeneous precision/retention | |
| Host-memory/SSD KV caching; Also belongs to KV-centric scheduling (temporal) | ||
| SSD-resident KV management with sparse selective access and speculative prefetching | ||
| System cost-driven decision; Also belongs to pipelining & overlapping (temporal) | ||
| HillInfer: Efficient Long-Context LLM Inference on the Edge with Hierarchical KV Eviction using SmartSSD [Link] He Sun, Shinan Liu, Li Li, Mingjun Xiao | SmartSSD-backed hierarchical KV offloading; Also belongs to HW-aware execution (temporal) | |
| Hierarchical KV cache management with selective retrieval for streaming video contexts; Also belongs to HW-aware execution | ||
| Importance-aware methods; Also belongs to pipelining & overlapping (temporal) | ||
| TraCT: Disaggregated LLM Serving with CXL Shared Memory KV Cache at Rack-Scale [Link] Dongha Yoon, Younghoon Min, Hoshik Kim, Sam H. Noh, Jongryool Kim | System cost-driven decision; Also belongs to compute device KV orchestration | |