中文
相关论文

相关论文: MLKV: Multi-Layer Key-Value Heads for Memory Effic…

200 篇论文

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…

机器学习 · 计算机科学 2025-12-08 Damien Lesens , Beheshteh T. Rakhshan , Guillaume Rabusseau

Efficient key-value (KV) cache compression is critical for scaling transformer-based Large Language Models (LLMs) in long sequences and resource-limited settings. Existing methods evict tokens based on their positions or importance scores,…

计算与语言 · 计算机科学 2025-05-19 Ziwei He , Jian Yuan , Haoli Bai , Jingwen Leng , Bo Jiang

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

The increasing memory demand of the Key-Value (KV) cache poses a significant bottleneck for Large Language Models (LLMs) in long-context applications. Existing low-rank KV compression methods reduce this footprint by modifying model…

计算与语言 · 计算机科学 2026-05-14 Shiyu Ji , Yixuan Wang , Yijun Liu , Qingfu Zhu , Wanxiang Che

Transformers have emerged as the prevailing standard solution for various AI tasks, including computer vision and natural language processing. The widely adopted Query, Key, and Value formulation (QKV) has played a significant role in this.…

计算机视觉与模式识别 · 计算机科学 2023-05-31 Ali Borji

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

Large language models (LLMs) demonstrate remarkable capabilities but face substantial serving costs due to their high memory demands, with the key-value (KV) cache being a primary bottleneck. State-of-the-art KV cache compression…

机器学习 · 计算机科学 2025-09-03 Yanqi Zhang , Yuwei Hu , Runyuan Zhao , John C. S. Lui , Haibo Chen

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…

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…

机器学习 · 计算机科学 2025-05-27 Kunjun Li , Zigeng Chen , Cheng-Yen Yang , Jenq-Neng Hwang

Large Language Models (LLMs) use key-value (KV) cache to reduce redundant computation in autoregressive generation. However, the KV cache size increases linearly during generation, leading to excessive memory usage, especially for long…

计算与语言 · 计算机科学 2025-03-04 Jian Yuan , Ziwei He , Haoli Bai , Jingwen Leng , Bo Jiang

While Key-Value (KV) cache succeeds in reducing redundant computations in auto-regressive models, it introduces significant memory overhead, limiting its practical deployment in long-sequence scenarios. Existing KV retrieval methods…

机器学习 · 计算机科学 2025-10-14 Wenbo Wu , Qingyi Si , Xiurui Pan , Ye Wang , Jie Zhang

Multimodal Large Language Models (MLLMs) have demonstrated remarkable performance across diverse applications. However, their computational overhead during deployment remains a critical bottleneck. While Key-Value (KV) caching effectively…

计算机视觉与模式识别 · 计算机科学 2025-03-26 Insu Han , Zeliang Zhang , Zhiyuan Wang , Yifan Zhu , Susan Liang , Jiani Liu , Haiting Lin , Mingjie Zhao , Chenliang Xu , Kun Wan , Wentian Zhao

Efficient inference for on-device Large Language Models (LLMs) remains challenging due to limited hardware resources and the high cost of the prefill stage, which processes the full input context to construct Key-Value (KV) caches. We…

网络与互联网体系结构 · 计算机科学 2026-05-06 Hongyao Liu , Liuqun Zhai , Junyi Wang , Zhengru Fang

We present PolyKV, a system in which multiple concurrent inference agents share a single, asymmetrically compressed KV cache pool. Rather than allocating a separate KV cache per agent -- the standard paradigm -- PolyKV writes a compressed…

机器学习 · 计算机科学 2026-04-29 Ishan Patel , Ishan Joshi

A critical approach for efficiently deploying computationally demanding large language models (LLMs) is Key-Value (KV) caching. The KV cache stores key-value states of previously generated tokens, significantly reducing the need for…

计算与语言 · 计算机科学 2024-09-10 Akide Liu , Jing Liu , Zizheng Pan , Yefei He , Gholamreza Haffari , Bohan Zhuang

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

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

Large Vision-Language Models (LVLMs) typically encode an image into a fixed number of visual tokens (e.g., 576) and process these tokens with a language model. Despite their strong performance, LVLMs face challenges in adapting to varying…

计算机视觉与模式识别 · 计算机科学 2024-06-10 Wenbo Hu , Zi-Yi Dou , Liunian Harold Li , Amita Kamath , Nanyun Peng , Kai-Wei Chang

Large Language Models (LLMs) have revolutionized a wide range of domains such as natural language processing, computer vision, and multi-modal tasks due to their ability to comprehend context and perform logical reasoning. However, the…

人工智能 · 计算机科学 2025-07-31 Haoyang Li , Yiming Li , Anxin Tian , Tianhao Tang , Zhanchao Xu , Xuejia Chen , Nicole Hu , Wei Dong , Qing Li , Lei Chen