English
Related papers

Related papers: StepCache: Step-Level Reuse with Lightweight Verif…

200 papers

Retrieval-Augmented Generation (RAG) systems suffer from severe time-to-first-token (TTFT) bottlenecks due to long input sequences. Existing KV cache reuse methods face a fundamental trade-off: prefix caching requires identical prefixes…

Machine Learning · Computer Science 2026-05-22 Bin Yang , Qiuyu Leng , Jun Zeng , Zhenhua Wu

KV cache restoration has emerged as a dominant bottleneck in serving long-context LLM workloads, including multi-turn conversations, retrieval-augmented generation, and agentic pipelines. Existing approaches treat restoration as a…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-04-29 Sean Nian , Jiahao Fang , Qilong Feng , Zhiyu Wu , Fan Lai

Serving large language models (LLMs) is important for cloud providers, and caching intermediate results (KV\$) after processing each request substantially improves serving throughput and latency. However, there is limited understanding of…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-17 Jiahao Wang , Jinbo Han , Xingda Wei , Sijie Shen , Dingyan Zhang , Chenguang Fang , Rong Chen , Wenyuan Yu , Haibo Chen

Last level caches (LLCs) occupy a large chip-area and there size is expected to grow further to offset the limitations of memory bandwidth and speed. Due to high leakage consumption of SRAM device, caches designed with SRAM consume large…

Hardware Architecture · Computer Science 2014-08-12 Sparsh Mittal

Test-time compute scaling has emerged as a powerful paradigm for enhancing mathematical reasoning in large language models (LLMs) by allocating additional computational resources during inference. However, current methods employ uniform…

Computation and Language · Computer Science 2025-12-02 Yang Xiao , Chunpu Xu , Ruifeng Yuan , Jiashuo Wang , Wenjie Li , Pengfei Liu

We describe KVLink, an approach for efficient key-value (KV) cache reuse in large language models (LLMs). In many LLM applications, different inputs can share overlapping context, such as the same retrieved document appearing in multiple…

Computation and Language · Computer Science 2025-11-11 Jingbo Yang , Bairu Hou , Wei Wei , Yujia Bao , Shiyu Chang

During LLM inference, KVCache memory usage grows linearly with sequence length and batch size and often exceeds GPU capacity. Recent proposals offload KV states to host memory and reduce transfers using top-k attention. But their…

Machine Learning · Computer Science 2026-03-30 Jiawei Yi , Ping Gong , Youhui Bai , Zewen Jin , Shengnan Wang , Jiaqi Ruan , Jia He , Jiaan Zhu , Pengcheng Wang , Haibo Wang , Weiguang Wang , Xia Zhu , Cheng Li

Finetuning on domain-specific data is a well-established method for enhancing LLM performance on downstream tasks. Training on each dataset produces a new set of model weights, resulting in a multitude of checkpoints saved in-house or on…

Machine Learning · Computer Science 2026-03-12 Sofia Maria Lo Cicero Vaina , Artem Chumachenko , Max Ryabinin

Diffusion and rectified flow (RF) models generate high-fidelity images and videos, but their iterative velocity-field evaluations are computationally expensive. Existing caching methods accelerate sampling by skipping timesteps, yet their…

Computer Vision and Pattern Recognition · Computer Science 2026-05-19 Xiao Liu , Kai Liu , Naiyang Guan , Hongliang Lu , Zhixin Wang , Zhikai Chen , Renjing Pei , Yulun Zhang

Large Language Models (LLMs) rely on Key-Value (KV) caching to accelerate inference, and many serving systems further share the KV cache across users' requests to reduce redundant computation. While widely adopted, unrestricted cross-user…

Cryptography and Security · Computer Science 2026-05-25 Guanlong Wu , Zhaohan li , Yao Zhang , Zheng Zhang , Jianyu Niu , Ye Wu , Yinqian Zhang

Side-channel attacks on shared hardware resources increasingly threaten confidentiality, especially with the rise of Large Language Models (LLMs). In this work, we introduce Spill The Beans, a novel application of cache side-channels to…

Cryptography and Security · Computer Science 2025-05-05 Andrew Adiletta , Berk Sunar

High-throughput inference serving is essential for applications built on large language models (LLMs). Existing serving frameworks reduce request-level and batch-level bubbles through batching and scheduling, but often overlook bubbles…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-25 Fengyao Bai , Hongbin Zhang , Zhitao Chen , Jiangsu Du , Zhiguang Chen , Yutong Lu

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

Embodied AI agents increasingly rely on large language models (LLMs) for planning, yet per-step LLM calls impose severe latency and cost. In this paper, we show that embodied tasks exhibit strong plan locality, where the next plan is…

Machine Learning · Computer Science 2026-04-28 Hojoon Kim , Yuheng Wu , Thierry Tambe

In cost-sensitive deployments, RAID arrays may combine SSDs with different performance levels. Such heterogeneity arises when aging SSDs degrade yet remain usable, or when failed drives are replaced with new devices of explicitly better…

Operating Systems · Computer Science 2026-04-03 Jialin Liu , Liang Shi , Dingcui Yu

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

Prefix caching is crucial to accelerate multi-turn interactions and requests with shared prefixes. At the cluster level, existing prefix caching systems are tightly coupled with request scheduling to optimize cache efficiency and…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-26 Bingyang Wu , Zili Zhang , Yinmin Zhong , Guanzhe Huang , Yibo Zhu , Xuanzhe Liu , Xin Jin

The quadratic complexity of the attention mechanism and the substantial memory footprint of the Key-Value (KV) cache present severe computational and memory challenges for Large Language Models (LLMs) processing long contexts. Existing…

Machine Learning · Computer Science 2026-03-10 Dongfang Li , Zixuan Liu , Gang Lin , Baotian Hu , Min Zhang

Instance-level image retrieval aims to find images containing the same object as a given query, despite variations in size, position, or appearance. To address this challenging task, we propose Patchify, a simple yet effective patch-wise…

Computer Vision and Pattern Recognition · Computer Science 2025-12-16 Wonseok Choi , Sohwi Lim , Nam Hyeon-Woo , Moon Ye-Bin , Dong-Ju Jeong , Jinyoung Hwang , Tae-Hyun Oh

Long-context agentic workflows have emerged as a defining use case for large language models, making attention efficiency critical for both inference speed and serving cost. Sparse attention addresses this challenge effectively, and…

Computation and Language · Computer Science 2026-03-13 Yushi Bai , Qian Dong , Ting Jiang , Xin Lv , Zhengxiao Du , Aohan Zeng , Jie Tang , Juanzi Li