English
Related papers

Related papers: Predicting Future Utility: Global Combinatorial Op…

200 papers

Recently the generative Large Language Model (LLM) has achieved remarkable success in numerous applications. Notably its inference generates output tokens one-by-one, leading to many redundant computations. The widely-used KV-Cache…

Machine Learning · Computer Science 2024-12-10 Weizhuo Li , Zhigang Wang , Yu Gu , Ge Yu

Recent reasoning large language models (LLMs) excel in complex tasks but encounter significant computational and memory challenges due to long sequence lengths. KV cache compression has emerged as an effective approach to greatly enhance…

Computation and Language · Computer Science 2025-12-02 Mengqi Liao , Lu Wang , Chaoyun Zhang , Zekai Shen , Xiaowei Mao , Si Qin , Qingwei Lin , Saravan Rajmohan , Dongmei Zhang , Huaiyu Wan

This paper introduces MadaKV, a modality-adaptive key-value (KV) cache eviction strategy designed to enhance the efficiency of multimodal large language models (MLLMs) in long-context inference. In multimodal scenarios, attention heads…

Machine Learning · Computer Science 2025-06-23 Kunxi Li , Zhonghua Jiang , Zhouzhou Shen , Zhaode Wang , Chengfei Lv , Shengyu Zhang , Fan Wu , Fei Wu

In Large Language Model (LLM) inference, Key-Value (KV) caches (KV-caches) are essential for reducing time complexity. However, they result in a linear increase in GPU memory as the context length grows. While recent work explores KV-cache…

Machine Learning · Computer Science 2025-02-25 Ahmed Burak Gulhan , Krishna Teja Chitty-Venkata , Murali Emani , Mahmut Kandemir , Venkatram Vishwanath

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

Recently, large language models (LLMs) have shown remarkable reasoning abilities by producing long reasoning traces. However, as the sequence length grows, the key-value (KV) cache expands linearly, incurring significant memory and…

Computation and Language · Computer Science 2026-02-04 Zican Dong , Peiyu Liu , Junyi Li , Zhipeng Chen , Han Peng , Shuo Wang , Wayne Xin Zhao

KV cache eviction has emerged as an effective solution to alleviate resource constraints faced by LLMs in long-context scenarios. However, existing token-level eviction methods often overlook two critical aspects: (1) their irreversible…

Machine Learning · Computer Science 2026-01-21 Yi Zhao , Yajuan Peng , Cam-Tu Nguyen , Zuchao Li , Xiaoliang Wang , Hai Zhao , Xiaoming Fu

Large Language Models (LLMs) are increasingly expected to operate over long contexts, yet standard softmax attention incurs a KV cache that grows linearly with sequence length, quickly becoming the bottleneck for long context inference. A…

Computation and Language · Computer Science 2026-05-26 Xintong Yang , Hao Gu , Binxing Xu , Lujun Li , Bei Liu , Jiacheng Liu , Qiyuan Zhu , Sirui Han , Yike Guo

Optimizing the Key-Value (KV) cache of the Large Language Model (LLM) has been considered critical to saving the cost of inference. Most of the existing KV-cache compression algorithms attempted to sparsify the sequence of tokens by taking…

Machine Learning · Computer Science 2024-10-11 Zihao Wang , Bin Cui , Shaoduo Gan

Long-context inference increasingly operates over CPU-resident KV caches, either because decoding-time KV states exceed GPU memory capacity or because disaggregated prefill-decode systems place KV data in host memory. Although block-sparse…

Machine Learning · Computer Science 2026-05-11 Feiyu Yao , Zhixiong Niu , Xiaqing Li , Yongqiang Xiong , Juan Fang , Qian Wang

The long-output context generation of large reasoning models enables extended chain of thought (CoT) but also drives rapid growth of the key-value (KV) cache, quickly overwhelming GPU memory. To address this challenge, we propose ThinKV, a…

Machine Learning · Computer Science 2026-05-11 Akshat Ramachandran , Marina Neseem , Charbel Sakr , Rangharajan Venkatesan , Brucek Khailany , Tushar Krishna

Withtherapid advancement of large language models (LLMs), the context length for inference has been continuously increasing, leading to an exponential growth in the demand for Key-Value (KV) caching. This has resulted in a significant…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-11 Yanyu Liu , Jingying Fu , Sixiang Liu , Yitian Zou , You Fu , Jiehan Zhou , Shouhua Zhang

To reduce memory consumption during LLM inference, prior works have proposed numerous methods that focus on KV cache pruning based on various criteria. While these techniques often accomplish lossless memory reduction on many datasets, they…

Computation and Language · Computer Science 2026-01-07 Xuanfan Ni , Liyan Xu , Chenyang Lyu , Longyue Wang , Mo Yu , Lemao Liu , Fandong Meng , Jie Zhou , Piji Li

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 (LLMs) have become a research hotspot. To accelerate the inference of LLMs, storing computed caches in memory has become the standard technique. However, as the inference length increases, growing KV caches might lead…

Computation and Language · Computer Science 2024-12-13 Meizhi Zhong , Xikai Liu , Chen Zhang , Yikun Lei , Yan Gao , Yao Hu , Kehai Chen , Min Zhang

Large language models (LLMs) rely on key-value (KV) caches for efficient autoregressive decoding; however, cache size grows linearly with context length and model depth, becoming a major bottleneck in long-context inference. Prior KV cache…

Machine Learning · Computer Science 2025-09-22 Dmitry Akulov , Mohamed Sana , Antonio De Domenico , Tareq Si Salem , Nicola Piovesan , Fadhel Ayed

Large Language Models (LLMs) have significantly advanced the field of Artificial Intelligence. However, their deployment is resource-intensive, not only due to the large number of model parameters but also because the (Key-Value) KV cache…

Computation and Language · Computer Science 2025-06-05 Yifeng Gu , Zicong Jiang , Jianxiu Jin , Kailing Guo , Ziyang Zhang , Xiangmin Xu

Large Language Models (LLMs) have revolutionized the field of natural language processing, achieving unprecedented performance across a variety of applications. However, their increased computational and memory demands present significant…

Computation and Language · Computer Science 2025-02-28 Yuhui Xu , Zhanming Jie , Hanze Dong , Lei Wang , Xudong Lu , Aojun Zhou , Amrita Saha , Caiming Xiong , Doyen Sahoo

Large language models (LLMs) have achieved remarkable success on various aspects of human life. However, one of the major challenges in deploying these models is the substantial memory consumption required to store key-value pairs (KV),…

Machine Learning · Computer Science 2025-02-26 Qiheng Sun , Hongwei Zhang , Haocheng Xia , Jiayao Zhang , Jinfei Liu , Kui Ren

Key-value (KV) caching is essential for large language model inference, yet its memory overhead poses a critical bottleneck for long-context generation. Existing eviction policies predominantly rely on empirical heuristics, lacking a…

Machine Learning · Computer Science 2026-04-30 Jiaming Yang , Chenwei Tang , Liangli Zhen , Jiancheng Lv