中文
相关论文

相关论文: Make Each Token Count: Towards Improving Long-Cont…

200 篇论文

Vision-Language Models (VLMs) have demonstrated impressive performance across a versatile set of tasks. A key challenge in accelerating VLMs is storing and accessing the large Key-Value (KV) cache that encodes long visual contexts, such as…

计算机视觉与模式识别 · 计算机科学 2024-11-01 Dezhan Tu , Danylo Vashchilenko , Yuzhe Lu , Panpan Xu

Retrieval-augmented generation (RAG) for long-context question answering is bottlenecked by inference-time prefilling over large retrieved contexts. A common strategy is to precompute key-value (KV) caches for individual documents and…

机器学习 · 计算机科学 2026-03-06 Xin Teng , Canyu Zhang , Shaoyi Zheng , Danyang Zhuo , Tianyi Zhou , Shengjie Wang

KV Cache is commonly used to accelerate LLM inference with long contexts, yet its high memory demand drives the need for cache compression. Existing compression methods, however, are largely heuristic and lack dynamic budget allocation. To…

机器学习 · 计算机科学 2025-09-15 Yiqun Shen , Song Yuan , Zhengze Zhang , Xiaoliang Wang , Daxin Jiang , Nguyen Cam-Tu

Generative inference in Large Language Models (LLMs) is impeded by the growing memory demands of Key-Value (KV) cache, especially for longer sequences. Traditional KV cache eviction strategies, which discard less critical KV pairs based on…

计算与语言 · 计算机科学 2025-03-14 Zhongwei Wan , Xinjian Wu , Yu Zhang , Yi Xin , Chaofan Tao , Zhihong Zhu , Xin Wang , Siqi Luo , Jing Xiong , Longyue Wang , Mi Zhang

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

Long-context LLM inference is bottlenecked by the memory and bandwidth cost of reading large KV caches during decoding. KV compression reduces this cost by keeping only part of the cache, but task accuracy alone does not identify why a…

机器学习 · 计算机科学 2026-05-12 Ruijie Zhang , Haozhe Liang , Da Chang , Li Hu , Fanqi Kong , Huaxiao Yin , Yu Li

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

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

The Key-Value (KV) cache is integral to efficient autoregressive inference in large language models (LLMs), yet its unbounded growth in stateful multi-turn scenarios presents major challenges. This paper examines the interplay between KV…

机器学习 · 计算机科学 2025-11-10 Pratik Poudel

Long-context language modeling is increasingly constrained by the Key-Value (KV) cache, whose memory and decode-time access costs scale linearly with the prefix length. This bottleneck has motivated a range of context-compression methods,…

机器学习 · 计算机科学 2026-05-13 Yoav Gelberg , Yam Eitan , Michael Bronstein , Yarin Gal , Haggai Maron

In large language models performing long-form reasoning, the KV cache grows rapidly with decode length, creating bottlenecks in memory and inference stability. Existing reasoning-oriented KV compression has mostly followed an…

人工智能 · 计算机科学 2026-04-14 Buseong Kim , Heejun Gwon

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

Modern Large Language Models (LLMs) are increasingly trained to support very large context windows. We present Compactor, a training-free, query-agnostic KV compression strategy that uses approximate leverage scores to determine token…

计算与语言 · 计算机科学 2025-12-10 Vivek Chari , Benjamin Van Durme

Autoregressive video diffusion models have enabled real-time, action-conditioned world generation. However, sustaining a persistent world, where revisiting a previously seen viewpoint yields consistent content, remains an open problem. Full…

计算机视觉与模式识别 · 计算机科学 2026-05-22 Jung Yi , Minjae Kim , Paul Hyunbin Cho , Wooseok Jang , Sangdoo Yun , Seungryong Kim

Large language models have revolutionized natural language processing, yet their deployment remains hampered by the substantial memory and runtime overhead of the transformer's Key-Value cache. To mitigate this, recent methods employ a…

计算与语言 · 计算机科学 2025-10-16 Yuan Feng , Haoyu Guo , JunLin Lv , S. Kevin Zhou , Xike Xie

Transformer-based large language models (LLMs) cache context as key-value (KV) pairs during inference. As context length grows, KV cache sizes expand, leading to substantial memory overhead and increased attention latency. This paper…

数据库 · 计算机科学 2025-10-01 Jang-Hyun Kim , Jinuk Kim , Sangwoo Kwon , Jae W. Lee , Sangdoo Yun , Hyun Oh Song

With the development of large language models (LLMs), efficient inference through Key-Value (KV) cache compression has attracted considerable attention, especially for long-context generation. To compress the KV cache, recent methods…

计算与语言 · 计算机科学 2025-10-28 Qingyue Yang , Jie Wang , Xing Li , Zhihai Wang , Chen Chen , Lei Chen , Xianzhi Yu , Wulong Liu , Jianye Hao , Mingxuan Yuan , Bin Li

Multimodal Large Language Models (MLLMs) have advanced unified reasoning over text, images, and videos, but their inference is hindered by the rapid growth of key-value (KV) caches. Each visual input expands into thousands of tokens,…

人工智能 · 计算机科学 2026-04-08 Bowen Zeng , Feiyang Ren , Jun Zhang , Xiaoling Gu , Ke Chen , Lidan Shou , Huan Li

Autoregressive decoding in large language models (LLMs) requires caching a growing list of past key-value (KV) pairs, making long-context inference a memory-bound problem. While recent methods have explored quantizing the cache, evicting…

计算与语言 · 计算机科学 2025-10-08 Harshil Vejendla

Supporting long-context LLMs is challenging due to the substantial memory demands of the key-value (KV) cache. Existing offloading systems store the full cache in host memory and selectively fetch critical entries during decoding, but this…

计算与语言 · 计算机科学 2026-05-19 Jian Lin , Jiazhi Mi , Zicong Hong , Haodong Wang , Qianli Liu , Haodyue Zhang , Peng Li , Song Guo