English
Related papers

Related papers: HeatKV: Head-tuned KV-cache Compression for Visual…

200 papers

The expanding long-context capabilities of large language models are constrained by a significant memory bottleneck: the key-value (KV) cache required for autoregressive generation. This bottleneck is substantial; for instance, a…

Computation and Language · Computer Science 2026-04-20 Yuxuan Zhu , David H. Yang , Mohammad Mohammadi Amiri , Keerthiram Murugesan , Tejaswini Pedapati , Pin-Yu Chen

Graphical user interface (GUI) agents built on vision-language models have emerged as a promising approach to automate human-computer workflows. However, they also face the inefficiency challenge as they process long sequences of…

Computation and Language · Computer Science 2025-10-02 Kung-Hsiang Huang , Haoyi Qiu , Yutong Dai , Caiming Xiong , Chien-Sheng Wu

KV cache has become a de facto technique for the inference of large language models (LLMs), where tensors of shape (layer number, head number, sequence length, feature dimension) are introduced to cache historical information for…

Machine Learning · Computer Science 2025-05-19 Bokai Lin , Zihao Zeng , Zipeng Xiao , Siqi Kou , Tianqi Hou , Xiaofeng Gao , Hao Zhang , Zhijie Deng

Reasoning large language models exhibit complex reasoning behaviors via extended chain-of-thought generation that are highly fragile to information loss during decoding, creating critical challenges for KV cache compression. Existing…

Computation and Language · Computer Science 2026-05-28 Wenjie Du , Li Jiang , Keda Tao , Xue Liu , Huan Wang

Reasoning models have demonstrated impressive performance in self-reflection and chain-of-thought reasoning. However, they often produce excessively long outputs, leading to prohibitively large key-value (KV) caches during inference. While…

Key-Value cache (\texttt{KV} \texttt{cache}) compression has emerged as a promising technique to optimize Large Language Model (LLM) serving. It primarily decreases the memory consumption of \texttt{KV} \texttt{cache} to reduce the…

Machine Learning · Computer Science 2025-04-01 Wei Gao , Xinyu Zhou , Peng Sun , Tianwei Zhang , Yonggang Wen

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

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

A unified autoregressive model is a Transformer-based framework that addresses diverse multimodal tasks (e.g., text, image, video) as a single sequence modeling problem under a shared token space. Such models rely on the KV-cache mechanism…

Computer Vision and Pattern Recognition · Computer Science 2026-01-09 Kunyang Li , Mubarak Shah , Yuzhang Shang

Memory consumption of the Key-Value (KV) cache represents a major bottleneck for efficient large language model inference. While attention-score-based KV cache pruning shows promise, it faces critical practical limitations: attention scores…

Artificial Intelligence · Computer Science 2025-10-02 Alessio Devoto , Maximilian Jeblick , Simon Jégou

While vision transformers (ViTs) have continuously achieved new milestones in the field of computer vision, their sophisticated network architectures with high computation and memory costs have impeded their deployment on resource-limited…

Hardware Architecture · Computer Science 2023-02-28 Peiyan Dong , Mengshu Sun , Alec Lu , Yanyue Xie , Kenneth Liu , Zhenglun Kong , Xin Meng , Zhengang Li , Xue Lin , Zhenman Fang , Yanzhi Wang

The KV cache in self-attention has emerged as a major bottleneck in long-context and large-batch inference for LLMs. Existing approaches often treat sparsity prediction and compression as separate modules, relying on auxiliary index…

Machine Learning · Computer Science 2026-03-17 Xu Yang , Jiapeng Zhang , Dongyang Zhao , Guo Chen , Zhuo Tang

The advent of pre-trained large language models (LLMs) has revolutionized various natural language processing tasks. These models predominantly employ an auto-regressive decoding mechanism that utilizes Key-Value (KV) caches to eliminate…

Computation and Language · Computer Science 2024-06-12 Hao Yu , Zelan Yang , Shen Li , Yong Li , Jianxin Wu

As Large Language Models (LLMs) scale in size and context length, the memory requirements of the key value (KV) cache have emerged as a major bottleneck during autoregressive decoding. The KV cache grows with sequence length and embedding…

Machine Learning · Computer Science 2025-12-09 Sourjya Roy , Shrihari Sridharan , Surya Selvam , Anand Raghunathan

Key-Value (KV) cache facilitates efficient large language models (LLMs) inference by avoiding recomputation of past KVs. As the batch size and context length increase, the oversized KV caches become a significant memory bottleneck,…

Machine Learning · Computer Science 2025-02-04 Zunhai Su , Zhe Chen , Wang Shen , Hanyu Wei , Linge Li , Huangqi Yu , Kehong Yuan

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

Transformer-based large language models (LLMs) have demonstrated remarkable potential across a wide range of practical applications. However, long-context inference remains a significant challenge due to the substantial memory requirements…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-09 Bo Jiang , Taolue Yang , Youyuan Liu , Xubin He , Sheng Di , Sian Jin

Recent advancements in Large Visual Language Models (LVLMs) have gained significant attention due to their remarkable reasoning capabilities and proficiency in generalization. However, processing a large number of visual tokens and…

Computer Vision and Pattern Recognition · Computer Science 2025-07-24 Kai Huang , Hao Zou , Bochen Wang , Ye Xi , Zhen Xie , Hao Wang

Self-forcing video generation extends a short-horizon video model to longer rollouts by repeatedly feeding generated content back in as context. This scaling path immediately exposes a systems bottleneck: the key-value (KV) cache grows with…

Machine Learning · Computer Science 2026-03-31 Suraj Ranganath , Vaishak Menon , Anish Patnaik

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…