English
Related papers

Related papers: EntropyCache: Decoded Token Entropy Guided KV Cach…

200 papers

Large language models (LLMs) with extended context windows have become increasingly prevalent for tackling complex tasks. However, the substantial Key-Value (KV) cache required for long-context LLMs poses significant deployment challenges.…

Computation and Language · Computer Science 2025-06-16 Jie Hu , Shengnan Wang , Yutong He , Ping Gong , Jiawei Yi , Juncheng Zhang , Youhui Bai , Renhai Chen , Gong Zhang , Cheng Li , Kun Yuan

Key-value (KV) caching is critical for efficient inference in large language models (LLMs), yet its memory footprint scales linearly with context length, resulting in a severe scalability bottleneck. Existing approaches largely treat KV…

Computation and Language · Computer Science 2026-04-23 Gradwell Dzikanyanga , Weihao Yang , Hao Huang , Donglei Wu , Shihao Wang , Wen Xia , Sanjeeb K C

Large Language Models (LLMs) have been widely adopted to process long-context tasks. However, the large memory overhead of the key-value (KV) cache poses significant challenges in long-context scenarios. Existing training-free KV cache…

Machine Learning · Computer Science 2024-10-22 Luning Wang , Shiyao Li , Xuefei Ning , Zhihang Yuan , Shengen Yan , Guohao Dai , Yu Wang

Large language models (LLMs) have demonstrated exceptional capabilities in generating text, images, and video content. However, as context length grows, the computational cost of attention increases quadratically with the number of tokens,…

Computation and Language · Computer Science 2025-04-23 Neusha Javidnia , Bita Darvish Rouhani , Farinaz Koushanfar

Reducing the key-value (KV) cache burden in Large Language Models (LLMs) significantly accelerates inference. Dynamically selecting critical KV caches during decoding helps maintain performance. Existing methods use random linear hashing to…

Computation and Language · Computer Science 2025-10-10 Wenhao Li , Yuxin Zhang , Gen Luo , Haiyuan Wan , Ziyang Gong , Fei Chao , Rongrong Ji

Key-Value (KV) caching is a common technique to enhance the computational efficiency of Large Language Models (LLMs), but its memory overhead grows rapidly with input length. Prior work has shown that not all tokens are equally important…

Computation and Language · Computer Science 2025-10-24 Yu Fu , Zefan Cai , Abedelkadir Asi , Wayne Xiong , Yue Dong , Wen Xiao

Diffusion language models (DLMs) have emerged as a promising alternative to autoregressive language generation due to their potential for parallel decoding and global refinement of the entire sequence. To unlock this potential, DLM…

Machine Learning · Computer Science 2026-04-20 Xiang Xia , Wuyang Zhang , Jiazheng Liu , Cheng Yan , Yanyong Zhang

Transformer-based large language models (LLMs) cache context as key-value (KV) pairs during inference. As context length grows, KV cache sizes expand, leading to substantial memory overhead and increased attention latency. This paper…

Databases · Computer Science 2025-10-01 Jang-Hyun Kim , Jinuk Kim , Sangwoo Kwon , Jae W. Lee , Sangdoo Yun , Hyun Oh Song

The growing size of Large Language Models (LLMs) makes efficient inference challenging, primarily due to the memory demands of the autoregressive Key-Value (KV) cache. Existing eviction or compression methods reduce cost but rely on…

Computation and Language · Computer Science 2026-02-12 Luca Moschella , Laura Manduchi , Ozan Sener

The Key-Value (KV) cache of Large Language Models (LLMs) is prefix-based, making it highly inefficient for processing contexts retrieved in arbitrary order. Position-Independent Caching (PIC) has been proposed to enable KV reuse without…

Machine Learning · Computer Science 2026-02-03 Shiju Zhao , Junhao Hu , Jiaqi Zheng , Guihai Chen

The Key-Value (KV) cache is central to the efficiency of transformer-based large language models (LLMs), storing previously computed vectors to accelerate inference. Yet, as sequence length and batch size grow, the cache becomes a major…

Machine Learning · Computer Science 2025-12-08 Damien Lesens , Beheshteh T. Rakhshan , Guillaume Rabusseau

Diffusion large language models (dLLMs) represent a significant advancement in text generation, offering parallel token decoding capabilities. However, existing open-source implementations suffer from quality-speed trade-offs that impede…

Computation and Language · Computer Science 2025-10-09 Fanheng Kong , Jingyuan Zhang , Yahui Liu , Zirui Wu , Yu Tian , Victoria W. , Guorui Zhou

While Key-Value (KV) cache compression is essential for efficient LLM inference, current evaluations disproportionately focus on sparse retrieval tasks, potentially masking the degradation of High-Density Reasoning where Chain-of-Thought…

Computation and Language · Computer Science 2026-05-13 Xiang Liu , Zhenheng Tang , Hong Chen , Peijie Dong , Zeyu Li , Xiuze Zhou , Bo Li , Xuming Hu , Xiaowen Chu

Key-Value (KV) cache has become a de facto component of modern Large Vision-Language Models (LVLMs) for inference. While it enhances decoding efficiency in Large Language Models (LLMs), its direct adoption in LVLMs introduces substantial…

Computer Vision and Pattern Recognition · Computer Science 2026-05-04 Xihao Chen , Yangyang Guo , Roger Zimmermann

Large language models (LLMs) are typically served from clusters of GPUs/NPUs that consist of large number of devices. Unfortunately, communication between these devices incurs significant overhead, increasing the inference latency and cost…

Artificial Intelligence · Computer Science 2025-05-27 Ahmet Caner Yüzügüler , Jiawei Zhuang , Lukas Cavigelli

Large Language Models (LLMs) have become the new foundation for many applications, reshaping human society like a storm. Disaggregated inference, which separates prefill and decode stages, is a promising approach to improving hardware…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-01-28 Shiyang Chen , Rain Jiang , Dezhi Yu , Jinlai Xu , Mengyuan Chao , Fanlong Meng , Chenyu Jiang , Wei Xu , Hang Liu

Transformers have emerged as the underpinning architecture for Large Language Models (LLMs). In generative language models, the inference process involves two primary phases: prompt processing and token generation. Token generation, which…

Machine Learning · Computer Science 2024-04-09 Muhammad Adnan , Akhil Arunkumar , Gaurav Jain , Prashant J. Nair , Ilya Soloveychik , Purushotham Kamath

Long-context LLMs demand accurate inference at low latency, yet decoding becomes primarily constrained by KV cache as context grows. Prior pruning methods are largely context-agnostic: their token selection ignores step-wise relevance and…

Artificial Intelligence · Computer Science 2026-02-25 Chao Fei , Guozhong Li , Chenxi Liu , Panos Kalnis

KV cache restoration has emerged as a dominant bottleneck in serving long-context LLM workloads, including multi-turn conversations, retrieval-augmented generation, and agentic pipelines. Existing approaches treat restoration as a…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-04-29 Sean Nian , Jiahao Fang , Qilong Feng , Zhiyu Wu , Fan Lai

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