arXiv:2412.19442cs.AIcs.DC2024-12中稿 · TMLR 2025综述被引 158

系统梳理大模型推理中键值缓存优化方法,助力高效部署。

A Survey on Large Language Model Acceleration based on KV Cache Management

论文配图:A Survey on Large Language Model Acceleration based on KV Cache Management
图 1 · 摘自论文原文
  • 按粒度分层归纳缓存管理策略,涵盖令牌、模型与系统级优化
  • 提出多类技术路径,如缓存合并、量化与低秩分解,提升复用率
  • 适合关注大模型推理加速的科研与工程人员参考

大型语言模型(LLMs)在自然语言处理、计算机视觉及多模态任务中因具备上下文理解与逻辑推理能力而引发变革。然而,其推理阶段的计算与内存开销极大,限制了在长上下文、实时应用中的扩展。键值(KV)缓存管理已成为关键优化手段,通过减少冗余计算和提升内存利用率来加速推理。本综述全面梳理了基于KV缓存管理的大模型加速策略,将其分为令牌级、模型级与系统级三类优化。令牌级方法包括缓存选择、预算分配、合并、量化与低秩分解;模型级优化聚焦架构创新与注意力机制以增强缓存复用;系统级方案则涉及内存管理、调度与硬件感知设计,提升异构环境下的效率。此外,综述还介绍了用于评估这些策略的文本与多模态数据集及基准测试。通过详尽分类与对比分析,本文为研究者与实践者提供实用洞见,推动高效可扩展的缓存管理技术发展,助力大模型在真实场景落地。相关论文列表见:https://github.com/TreeAI-Lab/Awesome-KV-Cache-Management。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have revolutionized a wide range of domains such as natural language processing, computer vision, and multi-modal tasks due to their ability to comprehend context and perform logical reasoning. However, the computational and memory demands of LLMs, particularly during inference, pose significant challenges when scaling them to real-world, long-context, and real-time applications. Key-Value (KV) cache management has emerged as a critical optimization technique for accelerating LLM inference by reducing redundant computations and improving memory utilization. This survey provides a comprehensive overview of KV cache management strategies for LLM acceleration, categorizing them into token-level, model-level, and system-level optimizations. Token-level strategies include KV cache selection, budget allocation, merging, quantization, and low-rank decomposition, while model-level optimizations focus on architectural innovations and attention mechanisms to enhance KV reuse. System-level approaches address memory management, scheduling, and hardware-aware designs to improve efficiency across diverse computing environments. Additionally, the survey provides an overview of both text and multimodal datasets and benchmarks used to evaluate these strategies. By presenting detailed taxonomies and comparative analyses, this work aims to offer useful insights for researchers and practitioners to support the development of efficient and scalable KV cache management techniques, contributing to the practical deployment of LLMs in real-world applications. The curated paper list for KV cache management is in: \href{https://github.com/TreeAI-Lab/Awesome-KV-Cache-Management}{https://github.com/TreeAI-Lab/Awesome-KV-Cache-Management}.

大模型加速缓存管理推理优化

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