English
Related papers

Related papers: Q-Filters: Leveraging QK Geometry for Efficient KV…

200 papers

Large language models have revolutionized data processing in numerous domains, with their ability to handle extended context reasoning receiving notable recognition. To speed up inference, maintaining a key-value (KV) cache memory is…

Computation and Language · Computer Science 2024-10-22 Zhen Yang , J. N. Han , Kan Wu , Ruobing Xie , An Wang , Xingwu Sun , Zhanhui Kang

The efficacy of Large Language Models (LLMs) in long-context tasks is often hampered by the substantial memory footprint and computational demands of the Key-Value (KV) cache. Current compression strategies, including token eviction and…

Computation and Language · Computer Science 2025-07-29 Runchao Li , Yao Fu , Mu Sheng , Xianxuan Long , Haotian Yu , Pan Li

The linear growth of key-value (KV) cache memory and quadratic computational in attention mechanisms complexity pose significant bottlenecks for large language models (LLMs) in long-context processing. While existing KV cache optimization…

Computation and Language · Computer Science 2025-10-07 Xin Liu , Xudong Wang , Pei Liu , Guoming Tang

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

KV cache pruning has emerged as a promising technique for reducing memory and computation costs in long-context auto-regressive generation. Existing methods for vision-language models (VLMs) typically rely on self-attention scores from…

Computer Vision and Pattern Recognition · Computer Science 2024-12-09 Xiaohuan Pei , Tao Huang , Chang Xu

The linear memory growth of the KV cache poses a significant bottleneck for LLM inference in long-context tasks. Existing static compression methods often fail to preserve globally important information. Although recent dynamic retrieval…

Computation and Language · Computer Science 2026-04-21 Zhiyuan Shi , Qibo Qiu , Feng Xue , Zhonglin Jiang , Li Yu , Jian Jiang , Xiaofei He , Wenxiao Wang

KV cache compression methods have mainly relied on scalar quantization techniques to reduce the memory requirements during decoding. In this work, we apply residual vector quantization, which has been widely used for high fidelity audio…

Machine Learning · Computer Science 2024-10-22 Ankur Kumar

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

Transformer-based Large Language Models rely critically on the KV cache to efficiently handle extended contexts during the decode phase. Yet, the size of the KV cache grows proportionally with the input length, burdening both memory…

Computation and Language · Computer Science 2025-08-14 Payman Behnam , Yaosheng Fu , Ritchie Zhao , Po-An Tsai , Zhiding Yu , Alexey Tumanov

Key-value (KV) caching has emerged as a crucial optimization technique for accelerating inference in large language models (LLMs). By allowing the attention operation to scale linearly rather than quadratically with the total sequence…

Computation and Language · Computer Science 2026-01-06 Gopi Krishna Jha , Sameh Gobriel , Liubov Talamanova , Nilesh Jain

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 struggle with memory demands from the growing Key-Value (KV) cache as context lengths increase. Existing compression methods homogenize head dimensions or rely on attention-guided token pruning, often sacrificing…

Computation and Language · Computer Science 2025-06-16 Xiaoran Liu , Siyang He , Qiqi Wang , Ruixiao Li , Yuerong Song , Zhigeng Liu , Linlin Li , Qun Liu , Zengfeng Huang , Qipeng Guo , Ziwei He , Xipeng Qiu

Large Language Models (LLMs) require significant GPU memory when processing long texts, with the key value (KV) cache consuming up to 70\% of total memory during inference. Although existing compression methods reduce memory by evaluating…

Computation and Language · Computer Science 2025-10-15 Xiang Liu , Zhenheng Tang , Peijie Dong , Zeyu Li , Yue Liu , Bo Li , Xuming Hu , Xiaowen Chu

The development of large language models (LLMs) has significantly expanded model sizes, resulting in substantial GPU memory requirements during inference. The key and value storage of the attention map in the KV (key-value) cache accounts…

Machine Learning · Computer Science 2024-10-25 Yifei Yang , Zouying Cao , Qiguang Chen , Libo Qin , Dongjie Yang , Hai Zhao , Zhi Chen

Efficiently serving large language models (LLMs) requires batching of many requests to reduce the cost per request. Yet, with larger batch sizes and longer context lengths, the key-value (KV) cache, which stores attention keys and values to…

Computation and Language · Computer Science 2024-07-26 Zirui Liu , Jiayi Yuan , Hongye Jin , Shaochen Zhong , Zhaozhuo Xu , Vladimir Braverman , Beidi Chen , Xia Hu

With the advancements in long-context inference capabilities of large language models (LLMs), the KV cache has become one of the foundational components. However, its substantial GPU memory consumption makes KV cache compression a key…

Computation and Language · Computer Science 2025-03-28 Youhui Zuo , Sibo Wei , Chen Zhang , Zhuorui Liu , Wenpeng Lu , Dawei Song

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

Scaling language models to long contexts is often bottlenecked by the size of the key-value (KV) cache. In deployed settings, long contexts are typically managed through compaction in token space via summarization. However, summarization…

Machine Learning · Computer Science 2026-05-28 Adam Zweiger , Xinghong Fu , Han Guo , Yoon Kim

Long streaming video QA remains challenging due to growing visual tokens and limited reasoning length of large language models (LLMs). KV-caching stores the Key-Value (KV) of the historical tokens via LLM prefill and enables more efficient…

Computer Vision and Pattern Recognition · Computer Science 2026-05-22 Junbin Xiao , Jiajun Chen , Tianxiang Sun , Xun Yang , Angela Yao

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