中文
相关论文

相关论文: HeatKV: Head-tuned KV-cache Compression for Visual…

200 篇论文

Serving transformer language models with high throughput requires caching Key-Values (KVs) to avoid redundant computation during autoregressive generation. The memory footprint of KV caching is significant and heavily impacts serving costs.…

机器学习 · 计算机科学 2026-04-28 Anastasiia Filippova , David Grangier , Marco Cuturi , João Monteiro

Visual AutoRegressive (VAR) modeling has garnered significant attention for its innovative next-scale prediction paradigm. However, mainstream VAR paradigms attend to all tokens across historical scales at each autoregressive step. As the…

计算机视觉与模式识别 · 计算机科学 2026-03-31 Zekun Li , Ning Wang , Tongxin Bai , Changwang Mei , Peisong Wang , Shuang Qiu , Jian Cheng

Key-value (KV) cache memory management is the primary bottleneck limiting throughput and cost-efficiency in large-scale GPU inference serving. Current systems suffer from three compounding inefficiencies: (1) the absence of unified KV cache…

硬件体系结构 · 计算机科学 2026-05-01 Sanjeev Rao Ganjihal

The integration of visual information into Large Language Models (LLMs) has enabled Multimodal LLMs (MLLMs), but the quadratic memory and computational costs of Transformer architectures remain a bottleneck. Existing KV cache eviction…

机器学习 · 计算机科学 2026-02-03 Xindian Ma , Yidi Lu , Peng Zhang , Jing Zhang

Large language models (LLMs) inference relies heavily on KV-caches to accelerate autoregressive decoding, but the resulting memory footprint grows rapidly with sequence length, posing significant efficiency challenges. Current KV-cache…

数据库 · 计算机科学 2025-09-04 Chenxia Tang , Jianchun Liu , Hongli Xu , Liusheng Huang

Vision-Language Models (VLMs) face a critical memory bottleneck when processing long-form video content due to the linear growth of the Key-Value (KV) cache with sequence length. Existing solutions predominantly employ reactive eviction…

计算机视觉与模式识别 · 计算机科学 2026-02-17 Vishnu Sai , Dheeraj Sai , Srinath B , Girish Varma , Priyesh Shukla

While Large Language Models (LLMs) can theoretically support extensive context windows, their actual deployment is constrained by the linear growth of Key-Value (KV) cache memory. Prevailing compression strategies mitigate this through…

人工智能 · 计算机科学 2026-02-03 Aryan Sood , Tanvi Sharma , Vansh Agrawal

The Key-Value (KV) cache in generative large language models (LLMs) introduces substantial memory overhead. Existing works mitigate this burden by offloading or compressing the KV cache. However, loading the entire cache incurs significant…

计算与语言 · 计算机科学 2025-05-28 Dingyu Yao , Bowen Shen , Zheng Lin , Wei Liu , Jian Luan , Bin Wang , Weiping Wang

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…

计算与语言 · 计算机科学 2024-10-22 Zhen Yang , J. N. Han , Kan Wu , Ruobing Xie , An Wang , Xingwu Sun , Zhanhui Kang

Key-Value (KV) cache plays a crucial role in accelerating inference in large language models (LLMs) by storing intermediate attention states and avoiding redundant computation during autoregressive generation. However, its memory footprint…

机器学习 · 计算机科学 2026-04-14 Yuzhen Mao , Qitong Wang , Martin Ester , Ke Li

Large language models cache all previously computed key-value (KV) pairs during generation, and this KV cache grows linearly with sequence length, making it a primary memory bottleneck for serving. Quantizing the KV cache to fewer bits…

机器学习 · 计算机科学 2026-05-11 Fei Zuo , Zikang Zhou , Hao Cong , Xiaoyan Xi , Ho Fai Leung

Vision Transformer (ViT)-based models have shown state-of-the-art performance (e.g., accuracy) in vision-based AI tasks. However, realizing their capability in resource-constrained embedded AI systems is challenging due to their inherent…

神经与进化计算 · 计算机科学 2026-01-06 Rachmad Vidya Wicaksana Putra , Saad Iftikhar , Muhammad Shafique

Talking head video compression has advanced with neural rendering and keypoint-based methods, but challenges remain, especially at low bit rates, including handling large head movements, suboptimal lip synchronization, and distorted facial…

图像与视频处理 · 电气工程与系统科学 2025-06-17 Riku Takahashi , Ryugo Morita , Jinjia Zhou

Autoregressive video diffusion models enable streaming generation, opening the door to long-form synthesis, video world models, and interactive neural game engines. However, their core attention layers become a major bottleneck at inference…

计算机视觉与模式识别 · 计算机科学 2026-02-03 Dvir Samuel , Issar Tzachor , Matan Levy , Micahel Green , Gal Chechik , Rami Ben-Ari

Large Language Models (LLMs) are increasingly used in applications requiring long context lengths, but the key-value (KV) cache often becomes a memory bottleneck on GPUs as context grows. To address this, we propose Commutative Vector…

Large language models (LLMs) have shown strong performance across diverse tasks, but their inference with long input contexts is bottlenecked by memory size and bandwidth. The Key-Value (KV) cache size grows linearly with sequence length…

机器学习 · 计算机科学 2026-05-12 Junkai Zhang , Hang Guo , Luca Benini , Yawei Li

While Transformer self-attention offers strong parallelism, the Key-Value (KV) cache grows linearly with sequence length and becomes a bottleneck for inference efficiency. Multi-head latent attention was recently developed to compress the…

机器学习 · 计算机科学 2025-11-04 Keqi Deng , Philip C. Woodland

Recent research has shown a strong theoretical connection between variational autoencoders (VAEs) and the rate-distortion theory. Motivated by this, we consider the problem of lossy image compression from the perspective of generative…

图像与视频处理 · 电气工程与系统科学 2023-03-28 Zhihao Duan , Ming Lu , Zhan Ma , Fengqing Zhu

In this study, we investigate whether attention-based information flow inside large language models (LLMs) is aggregated through noticeable patterns for long context processing. Our observations reveal that LLMs aggregate information…

计算与语言 · 计算机科学 2025-05-16 Zefan Cai , Yichi Zhang , Bofei Gao , Yuliang Liu , Yucheng Li , Tianyu Liu , Keming Lu , Wayne Xiong , Yue Dong , Junjie Hu , Wen Xiao

KV cache compression is critical for efficient long-context LLM inference. Approaches that reduce the per-pair footprint -- quantization and low-rank decomposition -- are orthogonal to those that reduce the sequence length of the cache.…

机器学习 · 计算机科学 2026-03-31 Bo Jiang , Sian Jin