中文
相关论文

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

200 篇论文

The key-value (KV) cache is a major bottleneck in long-context inference, where memory and computation grow with sequence length. Existing KV eviction methods reduce this cost but typically degrade performance relative to full-cache…

机器学习 · 计算机科学 2026-05-12 Ngoc Bui , Hieu Trung Nguyen , Arman Cohan , Rex Ying

Large language models (LLMs) are widely deployed with rapidly expanding context windows to support increasingly demanding applications. However, long contexts pose significant deployment challenges, primarily due to the KV cache whose size…

机器学习 · 计算机科学 2026-03-10 Guangda Liu , Chengwei Li , Zhenyu Ning , Jing Lin , Yiwu Yao , Danning Ke , Minyi Guo , Jieru Zhao

Generative large language models (LLMs) have revolutionized multiple domains. Modern LLMs predominantly rely on an autoregressive decoding strategy, which generates output tokens sequentially and employs a key-value cache (KV cache) to…

密码学与安全 · 计算机科学 2026-02-13 Ye Yu , Yifan Zhou , Yi Chen , Pedro Soto , Wenjie Xiong , Meng Li

The KV cache used in large language models has linearly growing time complexity, so LLMs face memory blow-up and reduced decoding efficiency when they process long contexts. Current KV Cache eviction has become an important research…

人工智能 · 计算机科学 2026-05-26 Wei Luo , Yi Huang , Songchen Ma , Huanyu Qu , Jiang Cai , Mingkun Xu

The efficiency of large language models (LLMs) remains a critical challenge, particularly in contexts where computational resources are limited. Traditional attention mechanisms in these models, while powerful, require significant…

计算与语言 · 计算机科学 2024-07-19 Bingli Liao , Danilo Vasconcellos Vargas

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…

计算机视觉与模式识别 · 计算机科学 2025-07-24 Kai Huang , Hao Zou , Bochen Wang , Ye Xi , Zhen Xie , Hao Wang

Recently, significant progress has been made in developing reasoning-capable Large Language Models (LLMs) through long Chain-of-Thought (CoT) techniques. However, this long-CoT reasoning process imposes substantial memory overhead due to…

计算与语言 · 计算机科学 2025-05-27 Tengxuan Liu , Shiyao Li , Jiayi Yang , Tianchen Zhao , Feng Zhou , Xiaohui Song , Guohao Dai , Shengen Yan , Huazhong Yang , Yu Wang

Generating long sequences of tokens given a long-context input is a very compute-intensive inference scenario for large language models (LLMs). One prominent inference speed-up approach is to construct a smaller key-value (KV) cache,…

计算与语言 · 计算机科学 2025-03-04 Fangyuan Xu , Tanya Goyal , Eunsol Choi

The emergence of LLMs has ignited a fresh surge of breakthroughs in NLP applications, particularly in domains such as question-answering systems and text generation. As the need for longer context grows, a significant bottleneck in model…

计算与语言 · 计算机科学 2024-04-15 Shichen Dong , Wen Cheng , Jiayu Qin , Wei Wang

Multi-agent Large Language Model (LLM) systems face a critical bottleneck: redundant transmission of contextual information between agents consumes excessive bandwidth and computational resources. Traditional approaches discard internal…

计算与语言 · 计算机科学 2025-12-23 Boris Kriuk , Logic Ng

The key-value (KV) cache accelerates LLMs decoding by storing KV tensors from previously generated tokens. It reduces redundant computation at the cost of increased memory usage. To mitigate this overhead, existing approaches compress KV…

机器学习 · 计算机科学 2025-07-30 Hao Wang , Ligong Han , Kai Xu , Akash Srivastava

Large language models (LLMs) utilize key-value (KV) cache to store historical information during sequence processing. The size of KV cache grows linearly as the length of the sequence extends, which seriously affects memory usage and…

计算与语言 · 计算机科学 2026-01-16 Yijun Liu , Yixuan Wang , Yuzhuang Xu , Shiyu Ji , Yang Xu , Qingfu Zhu , Wanxiang Che

How to efficiently serve LLMs in practice has become exceptionally challenging due to their prohibitive memory and computation requirements. In this study, we investigate optimizing the KV cache, whose memory footprint poses a critical…

计算与语言 · 计算机科学 2025-06-10 Akshat Sharma , Hangliang Ding , Jianping Li , Neel Dani , Minjia Zhang

Transformer-based large language models (LLMs) use the key-value (KV) cache to significantly accelerate inference by storing the key and value embeddings of past tokens. However, this cache consumes significant GPU memory. In this work, we…

Recently the generative Large Language Model (LLM) has achieved remarkable success in numerous applications. Notably its inference generates output tokens one-by-one, leading to many redundant computations. The widely-used KV-Cache…

机器学习 · 计算机科学 2024-12-10 Weizhuo Li , Zhigang Wang , Yu Gu , Ge Yu

Visual autoregressive modeling (VAR) via next-scale prediction has emerged as a scalable image generation paradigm. While Key and Value (KV) caching in large language models (LLMs) has been extensively studied, next-scale prediction…

计算机视觉与模式识别 · 计算机科学 2025-11-21 Boxun Xu , Yu Wang , Zihu Wang , Peng Li

Large language models (LLMs) support long-context inference but suffer from substantial memory and runtime overhead due to Key-Value (KV) Cache growth. Existing KV Cache eviction methods primarily rely on local attention weights, neglecting…

计算与语言 · 计算机科学 2026-05-11 Tho Mai , Joo-Young Kim

Large Language Models (LLMs) are increasingly expected to operate over long contexts, yet standard softmax attention incurs a KV cache that grows linearly with sequence length, quickly becoming the bottleneck for long context inference. A…

计算与语言 · 计算机科学 2026-05-26 Xintong Yang , Hao Gu , Binxing Xu , Lujun Li , Bei Liu , Jiacheng Liu , Qiyuan Zhu , Sirui Han , Yike Guo

While LLMs are powerful embedding backbones, their application in training-free settings faces two structural challenges: causal attention restricts early tokens from accessing subsequent context, and the next-token prediction objective…

计算与语言 · 计算机科学 2026-01-06 Yixuan Tang , Yi Yang

Transformer-based Large Language Models rely critically on the KV cache to efficiently handle extended contexts during the decode phase. Yet, the size of the KV cache grows proportionally with the input length, burdening both memory…

计算与语言 · 计算机科学 2025-08-14 Payman Behnam , Yaosheng Fu , Ritchie Zhao , Po-An Tsai , Zhiding Yu , Alexey Tumanov