English
Related papers

Related papers: LAVa: Layer-wise KV Cache Eviction with Dynamic Bu…

200 papers

Large Language Models (LLMs) face limitations due to the high demand on GPU memory and computational resources when handling long contexts. While sparsify the Key-Value (KV) cache of transformer model is a typical strategy to alleviate…

Machine Learning · Computer Science 2024-02-06 Yumeng Wang , Zhenyang Xiao

The integration of visual information into Large Language Models (LLMs) has enabled Multimodal LLMs (MLLMs), but the quadratic memory and computational costs of Transformer architectures remain a bottleneck. Existing KV cache eviction…

Machine Learning · Computer Science 2026-02-03 Xindian Ma , Yidi Lu , Peng Zhang , Jing Zhang

Large language models (LLMs) demonstrate remarkable capabilities but face substantial serving costs due to their high memory demands, with the key-value (KV) cache being a primary bottleneck. State-of-the-art KV cache compression…

Machine Learning · Computer Science 2025-09-03 Yanqi Zhang , Yuwei Hu , Runyuan Zhao , John C. S. Lui , Haibo Chen

Large Language Models (LLMs) excel in natural language processing tasks but pose significant computational and memory challenges for edge deployment due to their intensive resource demands. This work addresses the efficiency of LLM…

Hardware Architecture · Computer Science 2025-07-02 Zhican Wang , Hongxiang Fan , Haroon Waris , Gang Wang , Zhenyu Li , Jianfei Jiang , Yanan Sun , Guanghui He

With the growing demand for long-context LLMs across a wide range of applications, the key-value (KV) cache has become a critical bottleneck for both latency and memory usage. Recently, KV-cache offloading has emerged as a promising…

Machine Learning · Computer Science 2026-05-18 Andrey Bocharnikov , Ivan Ermakov , Denis Kuznedelev , Vyacheslav Zhdanovskiy , Yegor Yershov

Generative inference in Large Language Models (LLMs) is impeded by the growing memory demands of Key-Value (KV) cache, especially for longer sequences. Traditional KV cache eviction strategies, which discard less critical KV pairs based on…

Computation and Language · Computer Science 2025-03-14 Zhongwei Wan , Xinjian Wu , Yu Zhang , Yi Xin , Chaofan Tao , Zhihong Zhu , Xin Wang , Siqi Luo , Jing Xiong , Longyue Wang , Mi Zhang

Reusing KV cache is essential for high efficiency of Large Language Model (LLM) inference systems. With more LLM users, the KV cache footprint can easily exceed GPU memory capacity, so prior work has proposed to either evict KV cache to…

Transformer-based large language models (LLMs) rely on key-value (KV) caching to avoid redundant computation during autoregressive inference. While this mechanism greatly improves efficiency, the cache size grows linearly with the input…

Machine Learning · Computer Science 2026-03-12 Jinwoo Ahn , Ingyu Seong , Akhil Kedia , Junhan Kim , Hyemi Jang , Kangwook Lee , Yongkweon Jeon

Long-context Large Language Models (LLMs) enable powerful applications but incur high memory costs due to the key-value states (KV-Cache). Recent studies attempt to share KV-Cache across layers, but these approaches either require expensive…

Generative reasoning with large language models (LLMs) often involves long decoding sequences, leading to substantial memory and latency overheads from accumulating key-value (KV) caches. While existing KV compression methods primarily…

Machine Learning · Computer Science 2025-12-16 Hui Zeng , Daming Zhao , Pengfei Yang , WenXuan Hou , Tianyang Zheng , Hui Li , Weiye Ji , Jidong Zhai

Large Language Models (LLMs), epitomized by ChatGPT's release in late 2022, have revolutionized various industries with their advanced language comprehension. However, their efficiency is challenged by the Transformer architecture's…

Computation and Language · Computer Science 2024-11-21 Luohe Shi , Hongyi Zhang , Yao Yao , Zuchao Li , Hai Zhao

Large language models (LLMs) have demonstrated remarkable performance, but their long-context reasoning remains constrained by the excessive memory required for the Key-Value (KV) cache. This makes KV cache compression a critical step…

Machine Learning · Computer Science 2025-09-30 Xianglong Yan , Zhiteng Li , Tianao Zhang , Haotong Qin , Linghe Kong , Yulun Zhang , Xiaokang Yang

Recently, large vision-language models (LVLMs) have rapidly gained popularity for their strong generation and reasoning capabilities given diverse multimodal inputs. However, these models incur significant computational and memory overhead…

Computer Vision and Pattern Recognition · Computer Science 2025-10-20 Ao Wang , Hui Chen , Jiaxin Li , Jianchao Tan , Kefeng Zhang , Xunliang Cai , Zijia Lin , Jungong Han , Guiguang Ding

A critical approach for efficiently deploying computationally demanding large language models (LLMs) is Key-Value (KV) caching. The KV cache stores key-value states of previously generated tokens, significantly reducing the need for…

Computation and Language · Computer Science 2024-09-10 Akide Liu , Jing Liu , Zizheng Pan , Yefei He , Gholamreza Haffari , Bohan Zhuang

While large language models (LLMs) excel at handling long-context sequences, they require substantial prefill computation and key-value (KV) cache, which can heavily burden computational efficiency and memory usage in both prefill and…

Machine Learning · Computer Science 2026-04-21 Dongwon Jo , Jiwon Song , Yulhwa Kim , Jae-Joon Kim

Vision-Language Models (VLMs) have demonstrated impressive performance across a versatile set of tasks. A key challenge in accelerating VLMs is storing and accessing the large Key-Value (KV) cache that encodes long visual contexts, such as…

Computer Vision and Pattern Recognition · Computer Science 2024-11-01 Dezhan Tu , Danylo Vashchilenko , Yuzhe Lu , Panpan Xu

The memory and computational demands of Key-Value (KV) cache present significant challenges for deploying long-context language models. Previous approaches attempt to mitigate this issue by selectively dropping tokens, which irreversibly…

Machine Learning · Computer Science 2024-07-24 Hanlin Tang , Yang Lin , Jing Lin , Qingsen Han , Shikuan Hong , Yiwu Yao , Gongyi Wang

The high memory demands of the Key-Value (KV) Cache during the inference of Large Language Models (LLMs) severely restrict their deployment in resource-constrained platforms. Quantization can effectively alleviate the memory pressure caused…

Machine Learning · Computer Science 2026-02-03 Fei Li , Song Liu , Weiguo Wu , Shiqiang Nie , Jinyu Wang

Scaling the input context length of a large language model (LLM) incurs a significant increase in computation cost and memory footprint to maintain the attention key-value (KV) cache. Existing KV cache compression methods suffer from…

Computation and Language · Computer Science 2025-01-31 Yuxiang Huang , Binhang Yuan , Xu Han , Chaojun Xiao , Zhiyuan Liu

Key-value (KV) caching is widely used to accelerate transformer inference, but its memory cost grows linearly with input length, limiting long-context deployment. Existing token eviction methods reduce memory by discarding less important…

Machine Learning · Computer Science 2026-03-24 Ruijie Miao , Zhiming Wang , Wang Li , Shiwei Wu , Shufan Liu , Yanbing Jiang , Tong Yang