paper-with-me

Papers

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

2024-12-27 · Haoyang Li, Yiming Li, Anxin Tian, Tianhao Tang, Zhanchao Xu, Xuejia Chen, Nicole Hu, Wei Dong, Qing Li, Lei Chen

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}.

📄 PDF Abstract BibTeX arXiv:2412.19442

Code (1)

treeai-lab/awesome-kv-cache-management 공식 구현 pytorch

Tasks

Language ModelingLanguage ModellingLarge Language ModelLogical ReasoningManagementQuantizationScheduling

Methods 이 논문이 사용한 방법론

Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…
Attention 설명 없음
Focus 설명 없음

Similar Papers 제목 키워드 기반

Efficient Inference of Vision Instruction-Following Models with Elastic Cache

2024-07-25 · Zuyan Liu, Benlin Liu, Jiahui Wang, Yuhao Dong 외

In the field of instruction-following large vision-language models (LVLMs), the efficient deployment of these models faces challenges, notably due to the high memory demands of their key-value (KV) caches. Conventional c…

Instruction FollowingText Generation

PiKV: KV Cache Management System for Mixture of Experts

2025-08-02 · Dong Liu, Yanxuan Yu, Ben Lengerich, Ying Nian Wu arxiv

As large-scale language models continue to scale up in both size and context length, the memory and communication cost of key-value (KV) cache storage has become a major bottleneck in multi-GPU and multi-node inference. …

Improving the Serving Performance of Multi-LoRA Large Language Models via Efficient LoRA and KV Cache Management

2025-04-19 · Hang Zhang, Jiuchen Shi, Yixiao Wang, Quan Chen 외

Multiple Low-Rank Adapters (Multi-LoRAs) are gaining popularity for task-specific Large Language Model (LLM) applications. For multi-LoRA serving, caching hot KV caches and LoRA adapters in high bandwidth memory of accel…

Language ModelingLanguage ModellingLarge Language ModelManagement

From Tensor Buffer to Distributed Memory Hierarchy: A Survey of KV Cache Management for LLM Serving

2026-06-30 · Jie Li, Tongyang Wang, Yong Chen arxiv

The key-value (KV) cache has become a first-order memory object in LLM serving rather than a temporary per-request tensor. This survey classifies more than thirty KV-management systems and frameworks using four axes: loc…

LaCache: Exact Caching and Precision-Adaptive Inference for Diffusion Large Language Models

2026-07-16 · Xingru Chen, Zelang Liang, Yongjia Ma, Jiqing Zhan 외 arxiv

Diffusion-based Large Language Models(DLLMs) enable parallel generation via Semi-Autoregressive (SAR) decoding in text generation. However, current methods suffer from severe operator-level redundancy: they recompute the…

Text Generation