arXiv:2606.02964cs.ARcs.CL2026-06

通过感知计算延迟优化缓存管理,显著提升大模型推理速度。

Multi-Segment Attention: Enabling Efficient KV-Cache Management for Faster Large Language Model Serving

  • 根据GPU注意力核性能动态决定缓存保留,避免低效计算。
  • 减少首令牌时间(TTFT)1.90-2.03倍,每令牌耗时(TPOT)降低1.62-1.71倍。
  • 适合高并发大模型服务,尤其在智能体系统中表现优异。

大语言模型推理依赖键值(KV)缓存避免重复注意力计算。现有无损缓存管理方法主要基于访问频率或位置启发式决定淘汰策略,未考虑不同缓存块对GPU注意力核执行效率的影响。本文提出AsymCache,一种计算延迟感知的缓存管理框架,包含三个核心组件:支持高效非连续上下文处理的多段注意力(MSA)、联合优化命中率与重构成本的位置感知淘汰策略、以及提升硬件利用率的自适应分块调度器。实验表明,AsymCache在常见工作负载下将首令牌时间(TTFT)降低1.90-2.03倍,每令牌耗时(TPOT)减少1.62-1.71倍,验证了其在计算效率与缓存命中率间的平衡能力。此外,其底层设计可无缝集成至如Continuum等智能体服务系统,平均任务延迟进一步降低18.1%。

原文摘要 · Abstract (English)

Large Language Model (LLM) inference relies on key-value (KV) caches to avoid redundant attention computation. While approximate KV cache retention techniques reduce memory usage by sacrificing model accuracy, lossless approaches instead evict KV cache blocks from GPU memory and reconstruct them on demand to preserve exact outputs. Existing lossless KV cache management systems primarily base eviction decisions on access frequency or positional heuristics, without considering how different KV cache blocks affect the execution efficiency of GPU attention kernels. In this paper, we propose AsymCache, a computation-latency-aware KV cache management system for LLM inference that explicitly aligns cache residency decisions with GPU attention kernel performance, including three key components: Multi-Segment Attention (MSA) for efficient non-contiguous KV context processing, a cache eviction policy that jointly optimizes hit rate and position-aware recomputation cost, and an adaptive chunking scheduler for high hardware utilization. Experiments show that AsymCache reduces TTFT by up to 1.90-2.03x and time-per-output-token (TPOT) by 1.62-1.71x over latest baselines, confirming the effectiveness of the method in common workloads and validating its design goal of balancing computational efficiency with cache hit rate. Moreover, the low-level design of AsymCache allows seamless integration into agent serving systems such as Continuum, where it further reduces average job latency by up to 18.1%.

缓存管理大模型推理GPU优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。