中文
相关论文

相关论文: BUZZ: Beehive-structured Sparse KV Cache with Segm…

200 篇论文

Large language models (LLMs) rely on key-value cache (KV cache) to accelerate decoding by reducing redundant computations. However, the KV cache memory usage grows substantially with longer text sequences, posing challenges for efficient…

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

Key-value (KV) caching has emerged as a crucial optimization technique for accelerating inference in large language models (LLMs). By allowing the attention operation to scale linearly rather than quadratically with the total sequence…

计算与语言 · 计算机科学 2026-01-06 Gopi Krishna Jha , Sameh Gobriel , Liubov Talamanova , Nilesh Jain

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

KV cache quantization can improve Large Language Models (LLMs) inference throughput and latency in long contexts and large batch-size scenarios while preserving LLMs effectiveness. However, current methods have three unsolved issues:…

机器学习 · 计算机科学 2025-11-21 Xing Li , Zeyu Xing , Yiming Li , Linping Qu , Hui-Ling Zhen , Wulong Liu , Yiwu Yao , Sinno Jialin Pan , Mingxuan Yuan

Efficient inference with Large Language Models (LLMs) increasingly relies on Key-Value (KV) caches to store previously computed key and value vectors at each layer. These caches are essential to minimize redundant computation during…

硬件体系结构 · 计算机科学 2026-04-08 Oteo Mamo , Olga Kogiou , Hyunjin Yi , Weikuan Yu

Transformer-based large language models (LLMs) demonstrate impressive performance across various natural language processing tasks. Serving LLM inference for generating long contents, however, poses a challenge due to the enormous memory…

机器学习 · 计算机科学 2024-07-01 Wonbeom Lee , Jungi Lee , Junghwan Seo , Jaewoong Sim

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

Large Language Models (LLMs) have demonstrated remarkable proficiency across a wide range of tasks. However, LLMs often require larger batch sizes to enhance throughput or longer context lengths to meet task demands, which significantly…

机器学习 · 计算机科学 2025-05-23 Zhihang Cai , Xingjun Zhang , Zhendong Tan , Zheng Wei

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

Large language models (LLMs) based on Transformer Decoders have become the preferred choice for conversational generative AI. Despite the overall superiority of the Decoder architecture, the gradually increasing Key-Value (KV) cache during…

计算与语言 · 计算机科学 2025-07-16 Luohe Shi , Zuchao Li , Lefei Zhang , Guoming Liu , Baoyuan Qi , Hai Zhao

The efficiency of large vision-language models (LVLMs) is constrained by the computational bottleneck of the attention mechanism during the prefill phase and the memory bottleneck of fetching the key-value (KV) cache in the decoding phase,…

计算机视觉与模式识别 · 计算机科学 2024-12-19 Yefei He , Feng Chen , Jing Liu , Wenqi Shao , Hong Zhou , Kaipeng Zhang , Bohan Zhuang

Large language models (LLMs) often incorporate multiple text chunks in their inputs to provide the necessary contexts. To speed up the prefill of the long LLM inputs, one can pre-compute the KV cache of a text and re-use the KV cache when…

机器学习 · 计算机科学 2025-04-07 Jiayi Yao , Hanchen Li , Yuhan Liu , Siddhant Ray , Yihua Cheng , Qizheng Zhang , Kuntai Du , Shan Lu , Junchen Jiang

Optimizing the Key-Value (KV) cache of the Large Language Model (LLM) has been considered critical to saving the cost of inference. Most of the existing KV-cache compression algorithms attempted to sparsify the sequence of tokens by taking…

机器学习 · 计算机科学 2024-10-11 Zihao Wang , Bin Cui , Shaoduo Gan

Large Language Model (LLM) inference, where a trained model generates text one word at a time in response to user prompts, is a computationally intensive process requiring efficient scheduling to optimize latency and resource utilization. A…

机器学习 · 计算机科学 2026-01-16 Patrick Jaillet , Jiashuo Jiang , Konstantina Mellou , Marco Molinaro , Chara Podimata , Zijie Zhou

Growing context lengths in transformer-based language models have made the key-value (KV) cache a critical inference bottleneck. While many KV cache pruning methods have been proposed, they have not yet been adopted in major inference…

机器学习 · 计算机科学 2026-02-04 Simon Jegou , Maximilian Jeblick

With the widespread deployment of long-context large language models (LLMs), there has been a growing demand for efficient support of high-throughput inference. However, as the key-value (KV) cache expands with the sequence length, the…

机器学习 · 计算机科学 2025-04-29 Hanshi Sun , Li-Wen Chang , Wenlei Bao , Size Zheng , Ningxin Zheng , Xin Liu , Harry Dong , Yuejie Chi , Beidi Chen

The key-value (KV) cache is a foundational optimization in Transformer-based large language models (LLMs), eliminating redundant recomputation of past token representations during autoregressive generation. However, its memory footprint…

机器学习 · 计算机科学 2026-03-24 Yichun Xu , Navjot K. Khaira , Tejinder Singh

With the rapid development of large language models (LLMs), handling long context has become one of the vital abilities in LLMs. Such long-context ability is accompanied by difficulties in deployment, especially due to the increased…

计算与语言 · 计算机科学 2025-08-19 Zhuorui Liu , Chen Zhang , Dawei Song

Large Language Models (LLMs), despite their remarkable performance across a wide range of tasks, necessitate substantial GPU memory and consume significant computational resources. Beyond the memory taken up by model weights, the memory…

计算与语言 · 计算机科学 2024-06-24 Jincheng Dai , Zhuowei Huang , Haiyun Jiang , Chen Chen , Deng Cai , Wei Bi , Shuming Shi

How to efficiently serve Large Language Models (LLMs) has become a pressing issue because of their huge computational cost in their autoregressive generation process. To mitigate computational costs, LLMs often employ the KV Cache technique…

计算与语言 · 计算机科学 2024-07-23 Zheng Wang , Boxiao Jin , Zhongzhi Yu , Minjia Zhang