English
Related papers

Related papers: Self-Indexing KVCache: Predicting Sparse Attention…

200 papers

Chunk-wise autoregressive video diffusion models rely on a KV cache of previously generated chunks to avoid redundant computation, but this cache quickly becomes a memory bottleneck as videos grow longer. Methods that quantize the KV cache…

Machine Learning · Computer Science 2026-05-27 Tuna Tuncer , Felix Becker , Thomas Pfeil

Large Language Models (LLMs) face a significant bottleneck during autoregressive inference due to the massive memory footprint of the Key-Value (KV) cache. Existing compression techniques like token eviction, quantization, or other low-rank…

Machine Learning · Computer Science 2025-11-25 Santhosh G S , Saurav Prakash , Balaraman Ravindran

Long-context inference in large language models (LLMs) is increasingly constrained by the KV cache bottleneck: memory usage grows linearly with sequence length, while attention computation scales quadratically. Existing approaches address…

Computation and Language · Computer Science 2025-11-13 Huanxuan Liao , Yixing Xu , Shizhu He , Guanchen Li , Xuanwu Yin , Dong Li , Emad Barsoum , Jun Zhao , Kang Liu

Recent advances in large language models (LLMs) have significantly boosted long-context processing. However, the increasing key-value (KV) cache size poses critical challenges to memory and execution efficiency. Most KV cache compression…

Computation and Language · Computer Science 2025-08-05 Xiaolin Lin , Jingcun Wang , Olga Kondrateva , Yiyu Shi , Bing Li , Grace Li Zhang

Large language models have shown exceptional capabilities in a wide range of tasks, such as text generation and video generation, among others. However, due to their massive parameter count, these models often require substantial storage…

Machine Learning · Computer Science 2024-10-18 Qian Tao , Wenyuan Yu , Jingren Zhou

The Key-Value (KV) cache is crucial for efficient Large Language Models (LLMs) inference, but excessively long contexts drastically increase KV cache memory footprint. Existing KV cache compression methods typically rely on input-side…

Computation and Language · Computer Science 2026-03-13 Zhenxu Tian , Yi Su , Juntao Li , Min Zhang

The Key-Value (KV) cache, which stores intermediate attention computations (Key and Value pairs) to avoid redundant calculations, is a fundamental mechanism for accelerating Large Language Model (LLM) inference. However, this efficiency…

Cryptography and Security · Computer Science 2026-03-25 Zhifan Luo , Shuo Shao , Su Zhang , Lijing Zhou , Yuke Hu , Chenxu Zhao , Zhihao Liu , Zhan Qin

KV cache compression is critical for efficient long-context LLM inference. Approaches that reduce the per-pair footprint -- quantization and low-rank decomposition -- are orthogonal to those that reduce the sequence length of the cache.…

Machine Learning · Computer Science 2026-03-31 Bo Jiang , Sian Jin

Visual Autoregressive (VAR) models have recently demonstrated impressive image generation quality while maintaining low latency. However, they suffer from severe KV-cache memory constraints, often requiring gigabytes of memory per generated…

Computer Vision and Pattern Recognition · Computer Science 2026-05-15 Jonathan Cederlund , Axel Berg , Durmus Alp Emre Acar , Chuteng Zhou , Pontus Giselsson

Attention is the dominant source of latency during long-context LLM inference, an increasingly popular workload with reasoning models and RAG. We propose Kascade, a training-free sparse attention method that leverages known observations…

Machine Learning · Computer Science 2025-12-19 Dhruv Deshmukh , Saurabh Goyal , Nipun Kwatra , Ramachandran Ramjee

The rapid expansion of context window sizes in Large Language Models~(LLMs) has enabled them to tackle increasingly complex tasks involving lengthy documents. However, this progress comes at the cost of a substantial increase in memory…

Computation and Language · Computer Science 2025-08-05 Da Ma , Lu Chen , Situo Zhang , Yuxun Miao , Su Zhu , Zhi Chen , Hongshen Xu , Hanqi Li , Shuai Fan , Lei Pan , Kai Yu

Despite the significant success of large language models (LLMs), their extensive memory requirements pose challenges for deploying them in long-context token generation. The substantial memory footprint of LLM decoders arises from the…

Machine Learning · Computer Science 2024-02-12 Amir Zandieh , Insu Han , Vahab Mirrokni , Amin Karbasi

The Key-Value (KV) cache is a crucial component in serving transformer-based autoregressive large language models (LLMs), enabling faster inference by storing previously computed KV vectors. However, its memory consumption scales linearly…

Machine Learning · Computer Science 2024-10-07 Rongzhi Zhang , Kuang Wang , Liyuan Liu , Shuohang Wang , Hao Cheng , Chao Zhang , Yelong Shen

As large language models (LLMs) process increasing context windows, the memory usage of KV cache has become a critical bottleneck during inference. The mainstream KV compression methods, including KV pruning and KV quantization, primarily…

Computation and Language · Computer Science 2025-02-21 Jiebin Zhang , Dawei Zhu , Yifan Song , Wenhao Wu , Chuqiao Kuang , Xiaoguang Li , Lifeng Shang , Qun Liu , Sujian Li

Large language models (LLMs) inference relies heavily on KV-caches to accelerate autoregressive decoding, but the resulting memory footprint grows rapidly with sequence length, posing significant efficiency challenges. Current KV-cache…

Databases · Computer Science 2025-09-04 Chenxia Tang , Jianchun Liu , Hongli Xu , Liusheng Huang

KV-cache compression at small budgets is a crowded design space spanning cache representation, head-wise routing, compression cadence, decoding behavior, and within-budget scoring. We study seven mechanisms across these five families under…

Machine Learning · Computer Science 2026-05-19 Libo Sun , Po-wei Harn , Peixiong He , Xiao Qin

The increasing size of the Key-Value (KV) cache during the Large Language Models long-context inference is the main obstacle for its balance between the deployment cost and task accuracy. To reduce the KV cache size in such scenarios, most…

Machine Learning · Computer Science 2025-07-25 Manlai Liang , JiaMing Zhang , Xiong Li , Jinlong Li

As the demand for long-context large language models (LLMs) increases, models with context windows of up to 128K or 1M tokens are becoming increasingly prevalent. However, long-context LLM inference is challenging since the inference speed…

Computation and Language · Computer Science 2024-08-28 Jiaming Tang , Yilong Zhao , Kan Zhu , Guangxuan Xiao , Baris Kasikci , Song Han

Recently, large language models (LLMs) have been able to handle longer and longer contexts. However, a context that is too long may cause intolerant inference latency and GPU memory usage. Existing methods propose mixed-precision…

Computation and Language · Computer Science 2025-04-01 Wei Tao , Bin Zhang , Xiaoyang Qu , Jiguang Wan , Jianzong Wang

Agentic code tasks such as fault localization and patch generation require processing long codebases under tight memory constraints, where the Key-Value (KV) cache becomes the primary inference bottleneck. Existing compression methods rely…

Computation and Language · Computer Science 2026-04-14 Qiujiang Chen , Jing Xiong , Chenyang Zhao , Sidi Yang , Ngai Wong