English
Related papers

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

200 papers

Large Language Models (LLMs) are revolutionizing how users interact with information systems, yet their high inference cost poses serious scalability and sustainability challenges. Caching inference responses, allowing them to be retrieved…

Machine Learning · Computer Science 2026-02-16 Xutong Liu , Baran Atalar , Xiangxiang Dai , Jinhang Zuo , Siwei Wang , John C. S. Lui , Wei Chen , Carlee Joe-Wong

Efficient long-context inference is critical as large language models (LLMs) adopt context windows of ranging from 128K to 1M tokens. However, the growing key-value (KV) cache and the high computational complexity of attention create…

Computation and Language · Computer Science 2025-03-13 Guangtao Wang , Shubhangi Upasani , Chen Wu , Darshan Gandhi , Jonathan Li , Changran Hu , Bo Li , Urmish Thakker

Large Language Models (LLMs) have ignited an innovative surge of AI applications, marking a new era of exciting possibilities equipped with extended context windows. However, hosting these models is cost-prohibitive mainly due to the…

Computation and Language · Computer Science 2024-08-09 Yilong Chen , Guoxia Wang , Junyuan Shang , Shiyao Cui , Zhenyu Zhang , Tingwen Liu , Shuohuan Wang , Yu Sun , Dianhai Yu , Hua Wu

The growing memory footprint of the Key-Value (KV) cache poses a severe scalability bottleneck for long-context Large Language Model (LLM) inference. While KV cache eviction has emerged as an effective solution by discarding less critical…

Computation and Language · Computer Science 2025-10-28 Jinhan Chen , Jianchun Liu , Hongli Xu , Xianjun Gao , Shilong Wang

Large Language Models (LLMs) are increasingly expected to operate over long contexts, yet standard softmax attention incurs a KV cache that grows linearly with sequence length, quickly becoming the bottleneck for long context inference. A…

Computation and Language · Computer Science 2026-05-26 Xintong Yang , Hao Gu , Binxing Xu , Lujun Li , Bei Liu , Jiacheng Liu , Qiyuan Zhu , Sirui Han , Yike Guo

Large Language Models (LLMs) exhibit enhanced capabilities by Chain-of-Thought reasoning. However, the extended reasoning sequences introduce significant GPU memory overhead due to increased key-value (KV) cache. Existing KV cache…

Machine Learning · Computer Science 2025-10-16 Haoyue Zhang , Hualei Zhang , Xiaosong Ma , Jie Zhang , Song Guo

Large language models (LLMs) rely on key-value cache (KV cache) to accelerate decoding by reducing redundant computations. However, the KV cache memory usage grows substantially with longer text sequences, posing challenges for efficient…

Computation and Language · Computer Science 2025-11-18 Yixuan Wang , Shiyu Ji , Yijun Liu , Yuzhuang Xu , Yang Xu , Qingfu Zhu , Wanxiang Che

In long-context Large Language Model (LLM) inference, the Time-To-First-Token (TTFT) latency incurred by the prefill stage has become the foremost bottleneck limiting interactive performance and deployment cost. KV Cache reuse offers a…

Hardware Architecture · Computer Science 2026-05-26 Fei li , Song Liu , Yan Liu , Jinhua Cui , Shiqiang Nie , Jinyu Wang , Weiguo Wu

Large language models (LLMs) increasingly play an important role in a wide range of information processing and management tasks in industry. Many of these tasks are performed in large batches or even offline, and the performance indicator…

Computation and Language · Computer Science 2026-04-23 Zhen Zheng , Xin Ji , Taosong Fang , Fanghao Zhou , Chuanjie Liu , Gang Peng

The key-value (KV) cache is a major bottleneck in long-context inference, where memory and computation grow with sequence length. Existing KV eviction methods reduce this cost but typically degrade performance relative to full-cache…

Machine Learning · Computer Science 2026-05-12 Ngoc Bui , Hieu Trung Nguyen , Arman Cohan , Rex Ying

Transformer-based large language models (LLMs) have already achieved remarkable results on long-text tasks, but the limited GPU memory (VRAM) resources struggle to accommodate the linearly growing demand for key-value (KV) cache as the…

Computation and Language · Computer Science 2025-03-21 Shibo Jie , Yehui Tang , Kai Han , Zhi-Hong Deng , Jing Han

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

Self-attention is an essential component of large language models (LLM) but a significant source of inference latency for long sequences. In multi-tenant LLM serving scenarios, the compute and memory operation cost of self-attention can be…

Machine Learning · Computer Science 2024-08-02 Lu Ye , Ze Tao , Yong Huang , Yang Li

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

During inference for transformer-based large language models (LLM), prefilling is the computation of the key-value (KV) cache for input tokens in the prompt prior to autoregressive generation. For longer input prompt lengths, prefilling…

Machine Learning · Computer Science 2024-04-16 Siyan Zhao , Daniel Israel , Guy Van den Broeck , Aditya Grover

Large language models (LLMs) are often used for infilling tasks, which involve predicting or generating missing information in a given text. These tasks typically require multiple interactions with similar context. To reduce the computation…

Computation and Language · Computer Science 2025-05-30 Tianyu Guo , Hande Dong , Yichong Leng , Feng Liu , Cheater Lin , Nong Xiao , Xianwei Zhang

Large Language Models (LLMs) have become increasingly popular, transforming a wide range of applications across various domains. However, the real-world effectiveness of their query cache systems has not been thoroughly investigated. In…

Computation and Language · Computer Science 2024-06-04 Jiaxing Li , Chi Xu , Feng Wang , Isaac M von Riedemann , Cong Zhang , Jiangchuan Liu

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

Large Language Models (LLMs) are transforming recommendation from ranking into a generative task, but industrial deployment remains limited by the high latency of processing long, personalized prompts. Standard prefix caching provides…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-11 Zhan Zhao , Yuxin Wang , Amelie Chi Zhou

Large Language Models (LLMs) show great capabilities in a wide range of applications, but serving them efficiently becomes increasingly challenging as requests (prompts) become more complex. Context caching improves serving performance by…

Machine Learning · Computer Science 2025-05-28 Junhao Hu , Wenrui Huang , Weidong Wang , Haoyi Wang , Tiancheng Hu , Qin Zhang , Hao Feng , Xusheng Chen , Yizhou Shan , Tao Xie
‹ Prev 1 2 3 10 Next ›