English
Related papers

Related papers: SPHERICAL KV: Angle-Domain Attention and Rate-Dist…

200 papers

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 key-value (KV) cache accelerates LLMs decoding by storing KV tensors from previously generated tokens. It reduces redundant computation at the cost of increased memory usage. To mitigate this overhead, existing approaches compress KV…

Machine Learning · Computer Science 2025-07-30 Hao Wang , Ligong Han , Kai Xu , Akash Srivastava

Speculative decoding accelerates LLM inference, but SOTA hidden-state-based drafters suffer from long-range decay: draft accuracy degrades as the speculative step increases. Existing work attributes this decay to train-inference mismatch…

Computation and Language · Computer Science 2026-05-12 Tianyu Liu , Yuhao Shen , Xinyi Hu , Baolin Zhang , Hengxin Zhang , Jun Dai , Jun Zhang , Shuang Ge , Lei Chen , Yue Li , MingCheng Wan

Large language models face significant computational and memory challenges when processing long contexts. During inference, efficient management of the key-value (KV) cache, which stores intermediate activations for autoregressive…

Computation and Language · Computer Science 2025-09-30 Yuxuan Zhu , Ali Falahati , David H. Yang , Mohammad Mohammadi Amiri

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

Vision-Language-Action (VLA) models offer a unified framework for robotic perception and control, but their ability to scale to real-world, long-horizon tasks is limited by the high computational cost of attention and the large memory…

Computer Vision and Pattern Recognition · Computer Science 2025-11-25 Wanshun Xu , Long Zhuang , Lianlei Shan

Under modern test-time compute and agentic paradigms, language models process ever-longer sequences. Efficient text generation with transformer architectures is increasingly constrained by the Key-Value cache memory footprint and bandwidth.…

Long-context inference in large language models is bottlenecked by Key--Value (KV) cache loading during the decoding stage, where the sequential nature of generation requires repeatedly transferring the KV cache from off-chip High-Bandwidth…

Machine Learning · Computer Science 2026-03-03 Songtao Liu , Hongwu Peng , Zhiwei Zhang , Zhengyu Chen , Yue Guo

The linear growth of the Key-Value (KV) cache is a critical bottleneck in long-form LLM inference. Existing KV compression methods mitigate this by evicting tokens based on importance scores. However, we show that their reliance on global…

Machine Learning · Computer Science 2026-05-25 Junzhe Yang , Xiaoyu Shen

Key-value (KV) cache compression has emerged as a critical technique for reducing the memory and latency overhead of autoregressive language models during inference. Prior approaches predominantly rely on query-key attention scores to rank…

Computation and Language · Computer Science 2025-09-19 Ayan Sengupta , Siddhant Chaudhary , Tanmoy Chakraborty

In large-scale recommender systems, ultra-long user behavior sequences encode rich signals of evolving interests. Extending sequence length generally improves accuracy, but directly modeling such sequences in production is infeasible due to…

Information Retrieval · Computer Science 2025-08-26 Kaiyuan Li , Yongxiang Tang , Yanhua Cheng , Yong Bai , Yanxiang Zeng , Chao Wang , Xialong Liu , Peng Jiang

Retrieval-augmented generation (RAG) for long-context question answering is bottlenecked by inference-time prefilling over large retrieved contexts. A common strategy is to precompute key-value (KV) caches for individual documents and…

Machine Learning · Computer Science 2026-03-06 Xin Teng , Canyu Zhang , Shaoyi Zheng , Danyang Zhuo , Tianyi Zhou , Shengjie Wang

The deployment of large language models (LLMs) is often hindered by the extensive memory requirements of the Key-Value (KV) cache, especially as context lengths increase. Existing approaches to reduce the KV cache size involve either…

Computation and Language · Computer Science 2024-11-05 Alessio Devoto , Yu Zhao , Simone Scardapane , Pasquale Minervini

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

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

Visual Autoregressive (VAR) modeling has garnered significant attention for its innovative next-scale prediction approach, which yields substantial improvements in efficiency, scalability, and zero-shot generalization. Nevertheless, the…

Machine Learning · Computer Science 2025-05-27 Kunjun Li , Zigeng Chen , Cheng-Yen Yang , Jenq-Neng Hwang

Recent advances in large language models (LLMs) have showcased exceptional performance in long-context tasks, while facing significant inference efficiency challenges with limited GPU memory. Existing solutions first proposed the…

Computation and Language · Computer Science 2025-02-20 Qingfa Xiao , Jiachuan Wang , Haoyang Li , Cheng Deng , Jiaqi Tang , Shuangyin Li , Yongqi Zhang , Jun Wang , Lei Chen

With the development of large language models (LLMs), efficient inference through Key-Value (KV) cache compression has attracted considerable attention, especially for long-context generation. To compress the KV cache, recent methods…

Computation and Language · Computer Science 2025-10-28 Qingyue Yang , Jie Wang , Xing Li , Zhihai Wang , Chen Chen , Lei Chen , Xianzhi Yu , Wulong Liu , Jianye Hao , Mingxuan Yuan , Bin Li

As the context length of current large language models (LLMs) rapidly increases, the memory demand for the Key-Value (KV) cache is becoming a bottleneck for LLM deployment and batch processing. Traditional KV cache compression methods…

Computation and Language · Computer Science 2025-12-23 Aomufei Yuan , Zhiming Wang , Ruijie Miao , Dayu Wang , Yuxuan Tian , Zihan Wang , Yebo Peng , Yuhan Wu , Bairen Yi , Xin Liu , Tong Yang

Large language models have revolutionized natural language processing but face significant challenges of high storage and runtime costs, due to the transformer architecture's reliance on self-attention, particularly the large KV cache for…

Computation and Language · Computer Science 2026-05-29 Yuan Feng , Junlin Lv , Haoyu Guo , Yukun Cao , S Kevin Zhou , Xike Xie