中文
相关论文

相关论文: Homogeneous Keys, Heterogeneous Values: Exploiting…

200 篇论文

Large Language Models (LLMs) have demonstrated remarkable capabilities across diverse natural language processing tasks. However, their extensive memory requirements, particularly due to KV cache growth during long-text understanding and…

计算与语言 · 计算机科学 2025-10-14 Haoqi Yang , Yao Yao , Zuchao Li , Baoyuan Qi , Guoming Liu , Hai Zhao

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…

Large language models (LLMs) represent a groundbreaking advancement in the domain of natural language processing due to their impressive reasoning abilities. Recently, there has been considerable interest in increasing the context lengths…

机器学习 · 计算机科学 2024-11-11 Utkarsh Saxena , Gobinda Saha , Sakshi Choudhary , Kaushik Roy

The impressive capabilities of Large Language Models (LLMs) come at the cost of substantial computational resources during deployment. While KV Cache can significantly reduce recomputation during inference, it also introduces additional…

计算与语言 · 计算机科学 2025-05-19 Yi Su , Yuechi Zhou , Quantong Qiu , Juntao Li , Qingrong Xia , Ping Li , Xinyu Duan , Zhefeng Wang , Min Zhang

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

The deployment of efficient long-context LLMs in applications like autonomous agents, long-chain reasoning, and creative writing is fundamentally bottlenecked by the linear growth of KV cache memory. Existing compression and eviction…

计算与语言 · 计算机科学 2026-02-10 Jitai Hao , Qiang Huang , Yaowei Wang , Min Zhang , Jun Yu

Despite the significant success of large language models (LLMs), their extensive memory requirements pose challenges for deploying them in long-context token generation. The substantial memory footprint of LLM decoders arises from the…

机器学习 · 计算机科学 2024-02-12 Amir Zandieh , Insu Han , Vahab Mirrokni , Amin Karbasi

Agentic code tasks such as fault localization and patch generation require processing long codebases under tight memory constraints, where the Key-Value (KV) cache becomes the primary inference bottleneck. Existing compression methods rely…

计算与语言 · 计算机科学 2026-04-14 Qiujiang Chen , Jing Xiong , Chenyang Zhao , Sidi Yang , Ngai Wong

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

Vision-language models (VLMs) show remarkable performance in multimodal tasks. However, excessively long multimodal inputs lead to oversized Key-Value (KV) caches, resulting in significant memory consumption and I/O bottlenecks. Previous KV…

计算与语言 · 计算机科学 2025-01-28 Zunhai Su , Wang Shen , Linge Li , Zhe Chen , Hanyu Wei , Huangqi Yu , Kehong Yuan

Key-Value (KV) cache has become a de facto component of modern Large Vision-Language Models (LVLMs) for inference. While it enhances decoding efficiency in Large Language Models (LLMs), its direct adoption in LVLMs introduces substantial…

计算机视觉与模式识别 · 计算机科学 2026-05-04 Xihao Chen , Yangyang Guo , Roger Zimmermann

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

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

Autoregressive language models rely on a Key-Value (KV) Cache, which avoids re-computing past hidden states during generation, making it faster. As model sizes and context lengths grow, the KV Cache becomes a significant memory bottleneck,…

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…

机器学习 · 计算机科学 2025-05-19 Bokai Lin , Zihao Zeng , Zipeng Xiao , Siqi Kou , Tianqi Hou , Xiaofeng Gao , Hao Zhang , Zhijie Deng

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

Key-value (KV) caching has become the de-facto to accelerate generation speed for large language models (LLMs) inference. However, the growing cache demand with increasing sequence length has transformed LLM inference to be a memory bound…

机器学习 · 计算机科学 2024-10-02 Hao Kang , Qingru Zhang , Souvik Kundu , Geonhwa Jeong , Zaoxing Liu , Tushar Krishna , Tuo Zhao

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…

计算与语言 · 计算机科学 2025-10-15 Xiang Liu , Zhenheng Tang , Peijie Dong , Zeyu Li , Yue Liu , Bo Li , Xuming Hu , Xiaowen Chu

KV cache compression methods have mainly relied on scalar quantization techniques to reduce the memory requirements during decoding. In this work, we apply residual vector quantization, which has been widely used for high fidelity audio…

机器学习 · 计算机科学 2024-10-22 Ankur Kumar

With the growing demand for long-context LLMs across a wide range of applications, the key-value (KV) cache has become a critical bottleneck for both latency and memory usage. Recently, KV-cache offloading has emerged as a promising…

机器学习 · 计算机科学 2026-05-18 Andrey Bocharnikov , Ivan Ermakov , Denis Kuznedelev , Vyacheslav Zhdanovskiy , Yegor Yershov