中文
相关论文

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

200 篇论文

Existing key-value (KV) cache compression methods typically rely on heuristics, such as uniform cache allocation across layers or static eviction policies, however, they ignore the critical interplays among layer-specific feature patterns…

机器学习 · 计算机科学 2025-09-11 Bohan Yu , Yekun Chai

Despite the remarkable progress of Large Language Models (LLMs), the escalating memory footprint of the Key-Value (KV) cache remains a critical bottleneck for efficient inference. While dimensionality reduction offers a promising…

计算与语言 · 计算机科学 2026-03-06 Liming Lu , Kaixi Qiu , Jiayu Zhou , Jushi Kai , Haoyan Zhang , Huanyu Wang , Jingwen Leng , Ziwei He , Zhouhan Lin

In this study, we introduce adaptive KV cache compression, a plug-and-play method that reduces the memory footprint of generative inference for Large Language Models (LLMs). Different from the conventional KV cache that retains key and…

计算与语言 · 计算机科学 2024-10-31 Suyu Ge , Yunan Zhang , Liyuan Liu , Minjia Zhang , Jiawei Han , Jianfeng Gao

Context lengths of Large Language Models (LLMs) have exploded in recent years, with 128k-token context becoming a standard and million-token context becoming a reality. Efficiently supporting long-context inference remains challenging as…

计算与语言 · 计算机科学 2024-10-08 Isaac Rehg

Rotary Position Embedding (RoPE) enables each attention head to capture multi-frequency information along the sequence dimension and is widely applied in foundation models. However, the nonlinearity introduced by RoPE complicates…

机器学习 · 计算机科学 2025-03-04 Yuhao Zhou , Sirui Song , Boyang Liu , Zhiheng Xi , Senjie Jin , Xiaoran Fan , Zhihao Zhang , Wei Li , Xuanjing Huang

The memory and computational demands of Key-Value (KV) cache present significant challenges for deploying long-context language models. Previous approaches attempt to mitigate this issue by selectively dropping tokens, which irreversibly…

机器学习 · 计算机科学 2024-07-24 Hanlin Tang , Yang Lin , Jing Lin , Qingsen Han , Shikuan Hong , Yiwu Yao , Gongyi Wang

Multimodal Large Language Models face severe challenges in computational efficiency and memory consumption due to the substantial expansion of the visual KV cache when processing long visual contexts. Existing KV cache compression methods…

机器学习 · 计算机科学 2026-05-07 Sihao Liu , YuFan Xiong , Zhonghua Jiang , Zhaode Wang , chengfei lv Shengyu Zhang

The increasing size of the Key-Value (KV) cache during the Large Language Models long-context inference is the main obstacle for its balance between the deployment cost and task accuracy. To reduce the KV cache size in such scenarios, most…

机器学习 · 计算机科学 2025-07-25 Manlai Liang , JiaMing Zhang , Xiong Li , Jinlong Li

Recent advancements in Audio-Video Large Language Models (AV-LLMs) have enhanced their capabilities in tasks like audio-visual question answering and multimodal dialog systems. Video and audio introduce an extended temporal dimension,…

多媒体 · 计算机科学 2025-11-17 Zhonghua Jiang , Kui Chen , Kunxi Li , Keting Yin , Yiyun Zhou , Zhaode Wang , Chengfei Lv , Shengyu Zhang

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…

计算与语言 · 计算机科学 2026-04-21 Zhiyuan Shi , Qibo Qiu , Feng Xue , Zhonglin Jiang , Li Yu , Jian Jiang , Xiaofei He , Wenxiao Wang

Large Language Models (LLMs) are increasingly deployed in scenarios demanding ultra-long context reasoning, such as agentic workflows and deep research understanding. However, long-context inference is constrained by the KV cache, a…

硬件体系结构 · 计算机科学 2026-03-11 Jianlong Lei , Shashikant Ilager

Processing long-context inputs with large language models presents a significant challenge due to the enormous memory requirements of the Key-Value (KV) cache during inference. Existing KV cache compression methods exhibit noticeable…

计算与语言 · 计算机科学 2025-07-29 Dongquan Yang , Yifan Yang , Xiaotian Yu , Xianbiao Qi , Rong Xiao

A fundamental challenge in Visual Autoregressive models is the substantial memory overhead required during inference to store previously generated representations. Despite various attempts to mitigate this issue through compression…

机器学习 · 计算机科学 2026-01-26 Yang Cao , Xiaoyu Li , Yekun Ke , Yingyu Liang , Zhenmei Shi , Zhao Song

The growing context length of Large Language Models (LLMs) enlarges the Key-Value (KV) cache, limiting deployment in resource-limited environments. Prior training-free approaches for KV cache compression typically rely on low-rank…

计算与语言 · 计算机科学 2026-03-18 Yixuan Wang , Qingyu Shi , Jiayu Zhou , Dianbo Liu , Ziwei He , Zhouhan Lin

Key--value (KV) caching enables fast autoregressive decoding but at long contexts becomes a dominant bottleneck in High Bandwidth Memory (HBM) capacity and bandwidth. A common mitigation is to compress cached keys and values by projecting…

Visual Autoregressive (VAR) modeling has gained popularity for its shift towards next-scale prediction. However, existing VAR paradigms process the entire token map at each scale step, leading to the complexity and runtime scaling…

计算机视觉与模式识别 · 计算机科学 2025-07-09 Hang Guo , Yawei Li , Taolin Zhang , Jiangshan Wang , Tao Dai , Shu-Tao Xia , Luca Benini

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…

计算与语言 · 计算机科学 2024-12-13 Meizhi Zhong , Xikai Liu , Chen Zhang , Yikun Lei , Yan Gao , Yao Hu , Kehai Chen , Min Zhang

Recent advances in large language models (LLMs) have significantly boosted long-context processing. However, the increasing key-value (KV) cache size poses critical challenges to memory and execution efficiency. Most KV cache compression…

计算与语言 · 计算机科学 2025-08-05 Xiaolin Lin , Jingcun Wang , Olga Kondrateva , Yiyu Shi , Bing Li , Grace Li Zhang

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…

计算与语言 · 计算机科学 2025-03-28 Youhui Zuo , Sibo Wei , Chen Zhang , Zhuorui Liu , Wenpeng Lu , Dawei Song

KV cache is a widely used acceleration technique for large language models (LLMs) inference. However, its memory requirement grows rapidly with input length. Previous studies have reduced the size of KV cache by either removing the same…

计算与语言 · 计算机科学 2025-01-28 Xingyang He , Jie Liu , Shaowei Chen