中文

KV-RM: Regularizing KV-Cache Movement for Static-Graph LLM Serving

硬件体系结构 2026-05-12 v1 人工智能 分布式、并行与集群计算 操作系统

摘要

Static-graph LLM decoders provide predictable launches, fixed tensor shapes, and low submission overhead, but online decoding exposes highly irregular KV-cache behavior: request lengths differ, EOS events arrive asynchronously, and logical histories fragment over time. Dynamic runtimes recover flexibility through paged KV management and step-level scheduling, while static-graph executors often over-reserve memory and suffer burst-time latency outliers. This paper studies whether much of this variability can be absorbed below a fixed decode interface. We present KV-RM, a runtime design that regularizes KV-cache movement beneath a static-graph LLM decoder. KV-RM decouples logical KV histories from physical storage, tracks active KV state through a block pager, and materializes each decode step through a single committed descriptor. A merge-staged transport path coalesces non-contiguous KV mappings into a small number of large transfer groups before a fixed-shape attention kernel consumes them. Optional bounded far-history summaries can be enabled under the same interface, but the core design does not depend on them. On a 2-GPU NVIDIA A100 node, KV-RM improves mixed-length decoding throughput and tail latency relative to a static-graph baseline, reduces reserved KV memory across workload families, and removes severe burst-time latency spikes under production-trace replay. These results suggest that KV-cache movement, rather than kernel shape, can be an effective boundary for recovering runtime flexibility in static-graph LLM serving.

关键词

引用

@article{arxiv.2605.09735,
  title  = {KV-RM: Regularizing KV-Cache Movement for Static-Graph LLM Serving},
  author = {Zhiqing Zhong and Zhijing Ye and Jian Zhang and Weijian Zheng and Bolun Sun and Xiaodong Yu},
  journal= {arXiv preprint arXiv:2605.09735},
  year   = {2026}
}

备注

14 pages, 7 figures, 7 tables