English
Related papers

Related papers: DUAL-BLADE: Dual-Path NVMe-Direct KV-Cache Offload…

200 papers

Deploying large language models (LLMs) on edge platforms is challenged by their high computational and memory demands. Although recent low-bit quantization methods (e.g., BitNet, DeepSeek) compress weights to as little as 1.58 bits with…

Hardware Architecture · Computer Science 2025-04-28 Ye Qiao , Zhiheng Chen , Yifan Zhang , Yian Wang , Sitao Huang

Large Language Model or LLM inference has two phases, the prompt (or prefill) phase to output the first token and the extension (or decoding) phase to the generate subsequent tokens. In this work, we propose an efficient parallelization…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-05-15 Minsik Cho , Mohammad Rastegari , Devang Naik

With the skyrocketing costs of GPUs and their virtual instances in the cloud, there is a significant desire to use CPUs for large language model (LLM) inference. KV cache update, often implemented as allocation, copying, and in-place…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-11-18 Arun Ramachandran , Ramaswamy Govindarajan , Murali Annavaram , Prakash Raghavendra , Hossein Entezari Zarch , Lei Gao , Chaoyi Jiang

Although LLM inference has emerged as a critical workload for many downstream applications, efficiently inferring LLMs is challenging due to the substantial memory footprint and bandwidth requirements. In parallel, compute capabilities have…

Multi-agent LLM systems on edge devices need to hand off latent context efficiently, but the practical choices today are expensive re-prefill or full-precision KV transfer. We study QKVShare, a framework for quantized KV-cache handoff…

Artificial Intelligence · Computer Science 2026-05-06 Pratik Honavar , Tejpratap GVSL

In LLM serving, reusing the KV cache of prompts across requests is critical for reducing TTFT and serving costs. Cache-affinity scheduling, which co-locates requests with the same prompt prefix to maximize KV cache reuse, often conflicts…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-09 Ying Yuan , Pengfei Zuo , Bo Wang , Zhangyu Chen , Zhipeng Tan , Zhou Yu

The high memory demands of the Key-Value (KV) Cache during the inference of Large Language Models (LLMs) severely restrict their deployment in resource-constrained platforms. Quantization can effectively alleviate the memory pressure caused…

Machine Learning · Computer Science 2026-02-03 Fei Li , Song Liu , Weiguo Wu , Shiqiang Nie , Jinyu Wang

Large language model (LLM) applications often reuse previously processed context, such as chat history and documents, which introduces significant redundant computation. Existing LLM serving systems address such redundant computation by…

When key-value (KV) stores use SSDs for storing a large number of items, oftentimes they also require large in-memory data structures including indices and caches to be traversed to reduce IOs. This paper considers offloading most of such…

Diffusion language models (dLLMs) generate text by iteratively denoising multiple token positions in parallel, offering an attractive alternative to strictly autoregressive decoding. In practice, however, block-wise dLLM inference exposes a…

Machine Learning · Computer Science 2026-05-29 Xiaoyou Wu , Cheng-Jhih Shih , Binfei Ji , Yong Liu , Yingyan , Lin

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

Cost of serving large language models (LLM) is high, but the expensive and scarce GPUs are poorly efficient when generating tokens sequentially, unless the batch of sequences is enlarged. However, the batch size is limited by some…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-03-19 Jiaao He , Jidong Zhai

The rapid evolution of Large Language Models (LLMs) towards long-context reasoning and sparse architectures has pushed memory requirements far beyond the capacity of individual device HBM. While emerging supernode architectures offer…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-04 Fangxin Liu , Qinghua Zhang , Hanjing Shen , Zhibo Liang , Li Jiang , Haibing Guan , Chong Bao , Xuefeng Jin

Prefix KV caching has become a key mechanism in LLM serving: it reduces time to first token (TTFT) by avoiding redundant computation across requests that share a prefix (i.e., the system prompt). However, the accumulated KV cache is often…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-25 Yu Zhu , Aditya Dhakal , Yunming Xiao , Dejan Milojicic , Gustavo Alonso

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

LLM inference powers latency-critical production services nowadays. The bursty nature of inference traffic results in over-provisioning, which in turn leads to resource underutilization. While online-offline colocation promises to utilize…

Operating Systems · Computer Science 2026-04-10 Fangyue Liu , Hua Liu , Xinyuan Lyu , Shuo Ai , Hao Liang , Lingpeng Chen , Ziqian Hu , Chong Zha , Xin Jin , Hanmei Luo , Peng Chen

KV cache techniques in Transformer models aim to reduce redundant computations at the expense of substantially increased memory usage, making KV cache compression an important and popular research topic. Recently, state-of-the-art KV cache…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-05-20 Bingzhe Zhao , Ke Cheng , Aomufei Yuan , Yuxuan Tian , Ruiguang Zhong , Chengchen Hu , Tong Yang , Lian Yu

Large language models (LLMs) demonstrate remarkable capabilities but face substantial serving costs due to their high memory demands, with the key-value (KV) cache being a primary bottleneck. State-of-the-art KV cache compression…

Machine Learning · Computer Science 2025-09-03 Yanqi Zhang , Yuwei Hu , Runyuan Zhao , John C. S. Lui , Haibo Chen

Large Language Models (LLMs) have revolutionized the field of natural language processing, achieving unprecedented performance across a variety of applications. However, their increased computational and memory demands present significant…

Computation and Language · Computer Science 2025-02-28 Yuhui Xu , Zhanming Jie , Hanze Dong , Lei Wang , Xudong Lu , Aojun Zhou , Amrita Saha , Caiming Xiong , Doyen Sahoo

Disaggregated memory (DM) is a promising data center architecture that decouples CPU and memory into independent resource pools to improve resource utilization. Building on DM, memory-disaggregated key-value (KV) stores are adopted to…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-12-19 Zhisheng Hu , Jiacheng Shen , Ming-Chang Yang