English
Related papers

Related papers: Not All Tokens Are Worth Caching: Learning Semanti…

200 papers

Large Language Models (LLMs) rely on optimizations like Automatic Prefix Caching (APC) to accelerate inference. APC works by reusing previously computed states for the beginning part of a request (prefix), when another request starts with…

Cryptography and Security · Computer Science 2026-05-21 Panagiotis Georgios Pennas , Konstantinos Papaioannou , Marco Guarnieri , Thaleia Dimitra Doudali

In the field of instruction-following large vision-language models (LVLMs), the efficient deployment of these models faces challenges, notably due to the high memory demands of their key-value (KV) caches. Conventional cache management…

Computer Vision and Pattern Recognition · Computer Science 2024-07-26 Zuyan Liu , Benlin Liu , Jiahui Wang , Yuhao Dong , Guangyi Chen , Yongming Rao , Ranjay Krishna , Jiwen Lu

Large Language Models (LLMs) rely heavily on Key-Value (KV) caching to minimize inference latency. However, standard KV caches are context-dependent: reusing a cached document in a new context requires recomputing KV states to account for…

Machine Learning · Computer Science 2026-04-20 Chuangtao Chen , Grace Li Zhang , Xunzhao Yin , Cheng Zhuo , Bing Li , Ulf Schlichtmann

Recently, large vision-language models (LVLMs) have rapidly gained popularity for their strong generation and reasoning capabilities given diverse multimodal inputs. However, these models incur significant computational and memory overhead…

Computer Vision and Pattern Recognition · Computer Science 2025-10-20 Ao Wang , Hui Chen , Jiaxin Li , Jianchao Tan , Kefeng Zhang , Xunliang Cai , Zijia Lin , Jungong Han , Guiguang Ding

KV cache eviction has emerged as an effective solution to alleviate resource constraints faced by LLMs in long-context scenarios. However, existing token-level eviction methods often overlook two critical aspects: (1) their irreversible…

Machine Learning · Computer Science 2026-01-21 Yi Zhao , Yajuan Peng , Cam-Tu Nguyen , Zuchao Li , Xiaoliang Wang , Hai Zhao , Xiaoming Fu

Transformer-based large language models (LLMs) use the key-value (KV) cache to significantly accelerate inference by storing the key and value embeddings of past tokens. However, this cache consumes significant GPU memory. In this work, we…

Large Language Model (LLM) agents tackle data-intensive tasks such as deep research and code generation. However, their effectiveness depends on frequent interactions with knowledge sources across remote clouds or regions. Such interactions…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-04 Chaoyi Ruan , Chao Bi , Kaiwen Zheng , Ziji Shi , Xinyi Wan , Jialin Li

As large language models (LLMs) become widely used, their environmental impact, especially carbon emission, has attracted more attention. Prior studies focus on compute-related carbon emissions. In this paper, we find that storage is…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-04-14 Yuyang Tian , Desen Sun , Yi Ding , Sihang Liu

Prefix caching is a key latency optimization for autoregressive LLM serving, yet existing systems assume dense per-token key/value reuse. State-space models change the structure of the problem: a recurrent layer can resume from a single…

Machine Learning · Computer Science 2026-05-08 Mikhail Shirokikh , Sergey Nikolenko

Recent advancements in Large Language Models (LLMs) have spurred interest in numerous applications requiring robust long-range capabilities, essential for processing extensive input contexts and continuously generating extended outputs. As…

Machine Learning · Computer Science 2025-07-22 Dachuan Shi , Yonggan Fu , Xiangchi Yuan , Zhongzhi Yu , Haoran You , Sixu Li , Xin Dong , Jan Kautz , Pavlo Molchanov , Yingyan , Lin

Large Language Models (LLMs) have achieved remarkable progress across reasoning, generation, and decision-making tasks, yet deploying them on mobile, embedded, and edge devices remains particularly challenging. On-device LLM inference is…

Computer Vision and Pattern Recognition · Computer Science 2026-04-07 Sayed Pedram Haeri Boroujeni , Niloufar Mehrabi , Patrick Woods , Gabriel Hillesheim , Abolfazl Razi

Whether attention key value (KV) states computed for one prompt for a small LLM can be reused to accelerate inference on a new similar prompt, giving an increase to the space to its context memory using an approach called token recycling.…

Machine Learning · Computer Science 2025-12-16 Prashant Pandey

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

As large language models (LLMs) continue to advance, the demand for higher quality and faster processing of long contexts across various applications is growing. KV cache is widely adopted as it stores previously generated key and value…

Computation and Language · Computer Science 2025-02-28 Yingxin Li , Ye Li , Yuan Meng , Xinzhu Ma , Zihan Geng , Shutao Xia , Zhi Wang

Large Language Models (LLMs) like ChatGPT and Llama have revolutionized natural language processing and search engine dynamics. However, these models incur exceptionally high computational costs. For instance, GPT-3 consists of 175 billion…

Machine Learning · Computer Science 2025-09-15 Waris Gill , Mohamed Elidrisi , Pallavi Kalapatapu , Ammar Ahmed , Ali Anwar , Muhammad Ali Gulzar

Large language models with long context windows can answer complex questions directly from full-length academic, technical, and policy documents, but passing entire documents is often costly, slow, and can degrade answer quality while…

As the demand for long-context large language models (LLMs) increases, models with context windows of up to 128K or 1M tokens are becoming increasingly prevalent. However, long-context LLM inference is challenging since the inference speed…

Computation and Language · Computer Science 2024-08-28 Jiaming Tang , Yilong Zhao , Kan Zhu , Guangxuan Xiao , Baris Kasikci , Song Han

The rapid adoption of large language models (LLMs) has created demand for faster responses and lower costs. Semantic caching, reusing semantically similar requests via their embeddings, addresses this need but breaks classic cache…

Computation and Language · Computer Science 2026-03-05 Dvir David Biton , Roy Friedman

Memory and computation remain core bottlenecks in long-horizon LLM inference due to the quadratic cost of self-attention and the ever-growing key-value (KV) cache. Existing strategies for memory-bounded inference, such as quantization,…

Machine Learning · Computer Science 2026-03-03 Ngoc Bui , Shubham Sharma , Simran Lamba , Saumitra Mishra , Rex Ying

When LLM conversations grow beyond the context window, old content must be evicted -- but how does the model recover it when needed? We propose cooperative paging: evicted segments are replaced with minimal keyword bookmarks ([pN:keywords],…

Computation and Language · Computer Science 2026-05-26 Ziyang Liu