English
Related papers

Related papers: A Systematic Study of Cross-Layer KV Sharing for E…

200 papers

Large Language Models (LLMs) are increasingly deployed in multi-agent systems, where effective inter-model communication is crucial. Existing communication protocols either rely on natural language, incurring high inference costs and…

Machine Learning · Computer Science 2026-02-24 Xiangyu Shi , Marco Chiesa , Gerald Q. Maguire , Dejan Kostic

KV cache has traditionally been stored in GPU memory to accelerate the decoding phase of large language model (LLM) inference. However, it is increasingly necessary to move KV caches outside GPU devices, to enable cache reuse across…

Machine Learning · Computer Science 2025-12-08 Yuhan Liu , Yihua Cheng , Jiayi Yao , Yuwei An , Xiaokun Chen , Shaoting Feng , Yuyang Huang , Samuel Shen , Rui Zhang , Kuntai Du , Junchen Jiang

Long-context reasoning is a critical capability of large language models (LLMs), enabling applications such as long-document understanding, summarization, and code generation. However, efficient autoregressive inference relies on the…

Computation and Language · Computer Science 2026-04-28 Zahra Dehghanighobadi , Asja Fischer

The Key-Value (KV) cache is a crucial component in serving transformer-based autoregressive large language models (LLMs), enabling faster inference by storing previously computed KV vectors. However, its memory consumption scales linearly…

Machine Learning · Computer Science 2024-10-07 Rongzhi Zhang , Kuang Wang , Liyuan Liu , Shuohang Wang , Hao Cheng , Chao Zhang , Yelong Shen

Cross-layer key-value (KV) compression has been found to be effective in efficient inference of large language models (LLMs). Although they reduce the memory consumption of the KV cache, such methods usually introduce non-negligible…

Computation and Language · Computer Science 2026-04-16 You Wu , Ziheng Chen , Yizhen Zhang , Haoyi Wu , Chengting Yu , Yuchi Xu , Wenbo Su , Bo Zheng , Kewei Tu

Efficient deployment of Large Language Models (LLMs) requires batching multiple requests together to improve throughput. As the batch size, context length, or model size increases, the size of the key and value (KV) cache can quickly become…

Machine Learning · Computer Science 2024-05-08 Tianyi Zhang , Jonah Yi , Zhaozhuo Xu , Anshumali Shrivastava

Long-context Large Language Models (LLMs) face significant memory bottlenecks during inference due to the linear growth of key-value (KV) cache with sequence length. While individual optimization techniques like KV cache quantization,…

Machine Learning · Computer Science 2025-12-02 Sai Gokhale , Devleena Das , Rajeev Patwari , Ashish Sirasao , Elliott Delaye

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…

Machine Learning · Computer Science 2024-07-01 Wonbeom Lee , Jungi Lee , Junghwan Seo , Jaewoong Sim

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…

Computation and Language · Computer Science 2025-05-19 Yi Su , Yuechi Zhou , Quantong Qiu , Juntao Li , Qingrong Xia , Ping Li , Xinyu Duan , Zhefeng Wang , Min Zhang

Many computational factors limit broader deployment of large language models. In this paper, we focus on a memory bottleneck imposed by the key-value (KV) cache, a computational shortcut that requires storing previous KV pairs during…

Machine Learning · Computer Science 2024-06-13 Harry Dong , Xinyu Yang , Zhenyu Zhang , Zhangyang Wang , Yuejie Chi , Beidi Chen

Transformer-based large language models (LLMs) have demonstrated remarkable potential across a wide range of practical applications. However, long-context inference remains a significant challenge due to the substantial memory requirements…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-01-09 Bo Jiang , Taolue Yang , Youyuan Liu , Xubin He , Sheng Di , Sian Jin

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…

Computation and Language · Computer Science 2024-09-10 Akide Liu , Jing Liu , Zizheng Pan , Yefei He , Gholamreza Haffari , Bohan Zhuang

Large Language Model (LLM) inference is increasingly constrained by memory bandwidth, with frequent access to the key-value (KV) cache dominating data movement. While attention sparsity reduces some memory traffic, the relevance of past…

Hardware Architecture · Computer Science 2025-09-16 Yunhua Fang , Rui Xie , Asad Ul Haq , Linsen Ma , Kaoutar El Maghraoui , Naigang Wang , Meng Wang , Liu Liu , Tong Zhang

Key-value (KV) caching is critical for efficient inference in large language models (LLMs), yet its memory footprint scales linearly with context length, resulting in a severe scalability bottleneck. Existing approaches largely treat KV…

Computation and Language · Computer Science 2026-04-23 Gradwell Dzikanyanga , Weihao Yang , Hao Huang , Donglei Wu , Shihao Wang , Wen Xia , Sanjeeb K C

Recent large language models (LLMs) face increasing inference latency as input context length and model size continue to grow. In particular, the retrieval-augmented generation (RAG) technique, which enhances LLM responses by incorporating…

Artificial Intelligence · Computer Science 2025-04-17 Hyungwoo Lee , Kihyun Kim , Jinwoo Kim , Jungmin So , Myung-Hoon Cha , Hong-Yeon Kim , James J. Kim , Youngjae Kim

Large Language Models (LLMs) have achieved impressive accomplishments in recent years. However, the increasing memory consumption of KV cache has possessed a significant challenge to the inference system. Eviction methods have revealed the…

Computation and Language · Computer Science 2025-07-10 Zicong Tang , Shi Luohe , Zuchao Li , Baoyuan Qi , Guoming Liu , Lefei Zhang , Ping Wang

Large language models (LLMs) are typically served from clusters of GPUs/NPUs that consist of large number of devices. Unfortunately, communication between these devices incurs significant overhead, increasing the inference latency and cost…

Artificial Intelligence · Computer Science 2025-05-27 Ahmet Caner Yüzügüler , Jiawei Zhuang , Lukas Cavigelli

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…

Computation and Language · Computer Science 2024-07-23 Zheng Wang , Boxiao Jin , Zhongzhi Yu , Minjia Zhang

Large Language Models (LLMs), epitomized by ChatGPT's release in late 2022, have revolutionized various industries with their advanced language comprehension. However, their efficiency is challenged by the Transformer architecture's…

Computation and Language · Computer Science 2024-11-21 Luohe Shi , Hongyi Zhang , Yao Yao , Zuchao Li , Hai Zhao

Large language models (LLMs) with extended context windows have become increasingly prevalent for tackling complex tasks. However, the substantial Key-Value (KV) cache required for long-context LLMs poses significant deployment challenges.…

Computation and Language · Computer Science 2025-06-16 Jie Hu , Shengnan Wang , Yutong He , Ping Gong , Jiawei Yi , Juncheng Zhang , Youhui Bai , Renhai Chen , Gong Zhang , Cheng Li , Kun Yuan