English
Related papers

Related papers: KVSwap: Disk-aware KV Cache Offloading for Long-Co…

200 papers

The large size of the KV cache has become a major bottleneck for serving LLMs with increasing context lengths. In response, many KV cache compression methods, such as token dropping and quantization, have been proposed. However, almost all…

Hardware Architecture · Computer Science 2026-05-19 Jiayi Yao , Samuel Shen , Kuntai Du , Shaoting Feng , Dongjoo Seo , Rui Zhang , Yuyang Huang , Yuhan Liu , Shan Lu , Junchen Jiang

Large Language Models (LLMs) have been widely deployed in a variety of applications, and the context length is rapidly increasing to handle tasks such as long-document QA and complex logical reasoning. However, long context poses…

Machine Learning · Computer Science 2025-06-17 Guangda Liu , Chengwei Li , Jieru Zhao , Chenqi Zhang , Minyi Guo

Large language models (LLMs) have demonstrated strong capabilities in processing long contexts, enabling them to tackle tasks involving long textual inputs such as multi-turn conversations, legal documents, or retrieved documents in…

Computation and Language · Computer Science 2025-11-25 Yuechi Zhou , Yi Su , Jianxin Zhang , Juntao Li , Qingrong Xia , Zhefeng Wang , Xinyu Duan , Baoxing Huai

As large language models (LLMs) continue to scale, the memory footprint of key-value (KV) caches during inference has become a significant bottleneck. Existing approaches primarily focus on compressing KV caches within a single prompt or…

Computation and Language · Computer Science 2025-12-18 Xinye Zhao , Spyridon Mastorakis

In Text-to-SQL tasks, existing LLM-based methods often include extensive database schemas in prompts, leading to long context lengths and increased prefilling latency. While user queries typically focus on recurrent table sets-offering an…

Computation and Language · Computer Science 2026-01-14 Jinbo Su , Yuxuan Hu , Cuiping Li , Hong Chen , Jia Li , Lintao Ma , Jing Zhang

The context caching technique is employed to accelerate the Multimodal Large Language Model (MLLM) inference by prevailing serving platforms currently. However, this approach merely reuses the Key-Value (KV) cache of the initial sequence of…

Machine Learning · Computer Science 2025-09-23 Shiju Zhao , Junhao Hu , Rongxiao Huang , Jiaqi Zheng , Guihai Chen

LLM inference is essential for applications like text summarization, translation, and data analysis, but the high cost of GPU instances from Cloud Service Providers (CSPs) like AWS is a major burden. This paper proposes InferSave, a…

Machine Learning · Computer Science 2025-04-17 Kihyun Kim , Jinwoo Kim , Hyunsun Chung , Myung-Hoon Cha , Hong-Yeon Kim , Youngjae Kim

Large language models (LLMs) excel at processing long sequences, boosting demand for key-value (KV) caching. While recent efforts to evict KV cache have alleviated the inference burden, they often fail to allocate resources rationally…

Computation and Language · Computer Science 2025-12-25 Ziran Qin , Yuchen Cao , Mingbao Lin , Wen Hu , Shixuan Fan , Ke Cheng , Weiyao Lin , Jianguo Li

In this study, we introduce adaptive KV cache compression, a plug-and-play method that reduces the memory footprint of generative inference for Large Language Models (LLMs). Different from the conventional KV cache that retains key and…

Computation and Language · Computer Science 2024-10-31 Suyu Ge , Yunan Zhang , Liyuan Liu , Minjia Zhang , Jiawei Han , Jianfeng Gao

Deploying Large Language Models (LLMs) on memory-constrained AI Personal Computers (AIPCs) enables low-latency, privacy-preserving inference, but long-context generation is fundamentally bottlenecked by the linearly growing Key-Value (KV)…

Hardware Architecture · Computer Science 2026-03-26 He Sun , Shinan Liu , Li Li , Mingjun Xiao

Private large language model (LLM) inference based on secure multi-party computation (MPC) achieves formal data privacy protection but suffers from significant latency overhead, especially for long input sequences. While key-value (KV)…

Cryptography and Security · Computer Science 2025-10-21 Wenxuan Zeng , Ye Dong , Jinjin Zhou , Jin Tan , Lei Wang , Tao Wei , Runsheng Wang , Meng Li

Large language models (LLMs) are essential in natural language processing but often struggle with inference speed and computational efficiency, limiting real-time deployment. The key-value (KV) cache mechanism reduces computational overhead…

Computation and Language · Computer Science 2024-10-31 Junqi Zhao , Zhijin Fang , Shu Li , Shaohui Yang , Shichao He

Serving transformer language models with high throughput requires caching Key-Values (KVs) to avoid redundant computation during autoregressive generation. The memory footprint of KV caching is significant and heavily impacts serving costs.…

Machine Learning · Computer Science 2026-04-28 Anastasiia Filippova , David Grangier , Marco Cuturi , João Monteiro

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…

Computer Vision and Pattern Recognition · Computer Science 2026-05-04 Xihao Chen , Yangyang Guo , Roger Zimmermann

The efficiency of Large Language Model~(LLM) inference is often constrained by substantial memory bandwidth and capacity demands. Existing techniques, such as pruning, quantization, and mixture of experts/depth, reduce memory capacity…

Hardware Architecture · Computer Science 2025-04-23 Rui Xie , Asad Ul Haq , Linsen Ma , Yunhua Fang , Zirak Burzin Engineer , Liu Liu , Tong Zhang

KV cache accelerates LLM inference by avoiding redundant computation, at the expense of memory. To support larger KV caches, prior work extends GPU memory with CPU memory via CPU-offloading. This involves swapping KV cache between GPU and…

Operating Systems · Computer Science 2025-10-31 Ruihao Li , Shagnik Pal , Vineeth Narayan Pullu , Prasoon Sinha , Jeeho Ryoo , Lizy K. John , Neeraja J. Yadwadkar

Recent advances in Post-Training Quantization (PTQ) techniques have significantly increased demand for serving quantized large language models (LLMs), enabling higher throughput and substantially reduced memory usage with minimal accuracy…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-16 Kyungmin Bin , Seungbeom Choi , Jimyoung Son , Jieun Choi , Daseul Bae , Daehyeon Baek , Kihyo Moon , Minsung Jang , Hyojung Lee

Large language models encounter critical GPU memory capacity constraints during long-context inference, where KV cache memory consumption severely limits decode batch sizes. While existing research has explored offloading KV cache to DRAM,…

Machine Learning · Computer Science 2026-03-31 Qiuyang Zhang , Kai Zhou , Ding Tang , Kai Lu , Cheng Li , Zhenyu Yang , Peng Xu , Jiguang Wan

As the length of input text increases, the key-value (KV) cache in LLMs imposes prohibitive GPU memory costs and limits long-context inference on resource constrained devices. Existing approaches, such as KV quantization and pruning, reduce…

Machine Learning · Computer Science 2025-12-24 Tenghui Li , Guoxu Zhou , Xuyang Zhao , Yuning Qiu , Qibin Zhao

Large Reasoning Models (LRMs) are becoming integral to many AI inference systems, enhancing their capabilities with advanced reasoning. However, deploying these models in production environments presents a significant QoS challenge: the…

Machine Learning · Computer Science 2026-05-15 Kaiwen Chen , Xin Tan , Minchen Yu , Jingzong Li , Hong Xu