English
Related papers

Related papers: LRAgent: Efficient KV Cache Sharing for Multi-LoRA…

200 papers

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…

Machine Learning · Computer Science 2025-09-15 Yiqun Shen , Song Yuan , Zhengze Zhang , Xiaoliang Wang , Daxin Jiang , Nguyen Cam-Tu

We introduce LogQuant, a groundbreaking 2-bit quantization technique for KV Cache in large language model (LLM) inference, delivering substantial memory savings while preserving superior performance. Previous methods either assume that…

Machine Learning · Computer Science 2026-05-19 Han Chen , Zicong Jiang , Zining Zhang , Bingsheng He , Pingyi Luo , Mian Lu , Yuqiang Chen

Multi-head latent attention (MLA) is designed to optimize KV cache memory through low-rank key-value joint compression. Rather than caching keys and values separately, MLA stores their compressed latent representations, reducing memory…

Computation and Language · Computer Science 2025-09-09 Guihong Li , Mehdi Rezagholizadeh , Mingyu Yang , Vikram Appia , Emad Barsoum

Large Language Models (LLMs) based on autoregressive, decoder-only Transformers generate text one token at a time, where a token represents a discrete unit of text. As each newly produced token is appended to the partial output sequence,…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-05-06 Dimitrios Kafetzis , Ramin Khalili , Iordanis Koutsopoulos

Low-Rank Adaptation (LoRA) has become the de facto method for parameter-efficient fine-tuning of large language models (LLMs), enabling rapid adaptation to diverse domains. In production, LoRA-based models are served at scale, creating…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-12-01 Shashwat Jaiswal , Shrikara Arun , Anjaly Parayil , Ankur Mallick , Spyros Mastorakis , Alind Khare , Chloi Alverti , Renee St Amant , Chetan Bansal , Victor Rühle , Josep Torrellas

Recent advances in long-text understanding have pushed the context length of large language models (LLMs) up to one million tokens. It boosts LLMs's accuracy and reasoning capacity but causes exorbitant computational costs and…

Computation and Language · Computer Science 2025-05-19 Huan Yang , Renji Zhang , Mingzhe Huang , Weijun Wang , Yin Tang , Yuanchun Li , Yunxin Liu , Deyu Zhang

Large language models (LLMs) rely on Key-Value (KV) cache to reduce time-to-first-token (TTFT) latency, but existing disk-based KV cache systems using file-per-object layouts suffer from severe scalability bottlenecks due to file system…

Databases · Computer Science 2025-11-26 Weiping Yu , Ye Jiarui , He Mengke , Junfeng Liu , Siqiang Luo

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

Multi-head Latent Attention (MLA) is an innovative architecture proposed by DeepSeek, designed to ensure efficient and economical inference by significantly compressing the Key-Value (KV) cache into a latent vector. Compared to MLA,…

Computation and Language · Computer Science 2025-10-06 Tao Ji , Bin Guo , Yuanbin Wu , Qipeng Guo , Lixing Shen , Zhan Chen , Xipeng Qiu , Qi Zhang , Tao Gui

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…

Machine Learning · Computer Science 2024-10-11 Zihao Wang , Bin Cui , Shaoduo Gan

Multi-agent large language model (LLM) systems are increasingly adopted for complex language processing tasks that require communication and coordination among agents. However, these systems often suffer substantial overhead from repeated…

Multiagent Systems · Computer Science 2025-11-04 Hancheng Ye , Zhengqi Gao , Mingyuan Ma , Qinsi Wang , Yuzhe Fu , Ming-Yu Chung , Yueqian Lin , Zhijian Liu , Jianyi Zhang , Danyang Zhuo , Yiran Chen

As Large Language Models (LLMs) scale in size and context length, the memory requirements of the key value (KV) cache have emerged as a major bottleneck during autoregressive decoding. The KV cache grows with sequence length and embedding…

Machine Learning · Computer Science 2025-12-09 Sourjya Roy , Shrihari Sridharan , Surya Selvam , Anand Raghunathan

Retrieval-Augmented Generation (RAG) systems enhance the performance of large language models (LLMs) by incorporating supplementary retrieved documents, enabling more accurate and context-aware responses. However, integrating these external…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-03-25 Wenfeng Wang , Xiaofeng Hou , Peng Tang , Hengyi Zhou , Jing Wang , Xinkai Wang , Chao Li , Minyi Guo

Large Language Models (LLMs) are increasingly deployed in large-scale online services, enabling sophisticated applications. However, the computational overhead of generating key-value (KV) caches in the prefill stage presents a major…

Machine Learning · Computer Science 2025-02-24 Shuowei Jin , Xueshen Liu , Qingzhao Zhang , Z. Morley Mao

Large language models (LLMs) with chain-of-thought reasoning achieve state-of-the-art performance across complex problem-solving tasks, but their verbose reasoning traces and large context requirements make them impractical for edge…

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

Large Language Models struggle with memory demands from the growing Key-Value (KV) cache as context lengths increase. Existing compression methods homogenize head dimensions or rely on attention-guided token pruning, often sacrificing…

Computation and Language · Computer Science 2025-06-16 Xiaoran Liu , Siyang He , Qiqi Wang , Ruixiao Li , Yuerong Song , Zhigeng Liu , Linlin Li , Qun Liu , Zengfeng Huang , Qipeng Guo , Ziwei He , Xipeng Qiu

LLM agents today communicate via text, which incurs considerable latency and information loss due to the need to autoregressively decode the sharer model's state and encode at the receiver model. Recent work such as Cache-to-Cache (C2C; Fu…

Machine Learning · Computer Science 2026-05-25 Maximillian Rossi , Prajwal Raghunath , Eugene Wu

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…

Computer Vision and Pattern Recognition · Computer Science 2024-11-01 Dezhan Tu , Danylo Vashchilenko , Yuzhe Lu , Panpan Xu

PagedAttention is a popular approach for dynamic memory allocation in LLM serving systems. It enables on-demand allocation of GPU memory to mitigate KV cache fragmentation -- a phenomenon that crippled the batch size (and consequently…

Machine Learning · Computer Science 2025-01-30 Ramya Prabhu , Ajay Nayak , Jayashree Mohan , Ramachandran Ramjee , Ashish Panwar