English
Related papers

Related papers: TriForce: Lossless Acceleration of Long Sequence G…

200 papers

Speculative decoding is a widely used technique for accelerating inference in large language models (LLMs), but its performance degrades as input length grows, with significant drops even at moderate lengths. Yet, this early degradation has…

Computation and Language · Computer Science 2026-01-21 Jungyoub Cha , Hyunjong Kim , Sungzoon Cho

Vision-Language Models (VLMs) have emerged as a critical and fast-growing extension of Large Language Models (LLMs) that enable multimodal reasoning through both text and image inputs. Although VLMs enrich the capabilities of language…

Computer Vision and Pattern Recognition · Computer Science 2026-05-19 Yingbing Huang , Tharun Adithya Srikrishnan , Steven K. Reinhardt , Deming Chen

Long-context inference in LLMs faces the dual challenges of quadratic attention complexity and prohibitive KV cache memory. While token-level sparse attention offers superior accuracy, its indexing overhead is costly; block-level methods…

Computation and Language · Computer Science 2026-04-10 Yuxuan Hu , Jianchao Tan , Jiaqi Zhang , Wen Zan , Pingwei Sun , Yifan Lu , Yerui Sun , Yuchen Xie , Xunliang Cai , Jing Zhang

Recently, sharing key-value (KV) cache across layers has been found effective in efficient inference of large language models (LLMs). To systematically investigate different techniques of cross-layer KV sharing, we propose a unified…

Computation and Language · Computer Science 2025-02-06 You Wu , Haoyi Wu , Kewei Tu

Large language models (LLMs) have demonstrated exceptional capabilities in generating text, images, and video content. However, as context length grows, the computational cost of attention increases quadratically with the number of tokens,…

Computation and Language · Computer Science 2025-04-23 Neusha Javidnia , Bita Darvish Rouhani , Farinaz Koushanfar

Large language models (LLMs) utilize key-value (KV) cache to store historical information during sequence processing. The size of KV cache grows linearly as the length of the sequence extends, which seriously affects memory usage and…

Computation and Language · Computer Science 2026-01-16 Yijun Liu , Yixuan Wang , Yuzhuang Xu , Shiyu Ji , Yang Xu , Qingfu Zhu , Wanxiang Che

Recently, significant progress has been made in developing reasoning-capable Large Language Models (LLMs) through long Chain-of-Thought (CoT) techniques. However, this long-CoT reasoning process imposes substantial memory overhead due to…

Computation and Language · Computer Science 2025-05-27 Tengxuan Liu , Shiyao Li , Jiayi Yang , Tianchen Zhao , Feng Zhou , Xiaohui Song , Guohao Dai , Shengen Yan , Huazhong Yang , Yu Wang

Large language models (LLMs) suffer from low efficiency as the mismatch between the requirement of auto-regressive decoding and the design of most contemporary GPUs. Specifically, billions to trillions of parameters must be loaded to the…

Computation and Language · Computer Science 2024-05-02 Bin Xiao , Chunan Shi , Xiaonan Nie , Fan Yang , Xiangwei Deng , Lei Su , Weipeng Chen , Bin Cui

Large Language Models (LLMs) with expanding context windows face significant performance hurdles. While caching key-value (KV) states is critical for avoiding redundant computation, the storage footprint of long-context caches quickly…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-08-27 Zhiqiang Xie , Ziyi Xu , Mark Zhao , Yuwei An , Vikram Sharma Mailthody , Scott Mahlke , Michael Garland , Christos Kozyrakis

Key-Value cache (\texttt{KV} \texttt{cache}) compression has emerged as a promising technique to optimize Large Language Model (LLM) serving. It primarily decreases the memory consumption of \texttt{KV} \texttt{cache} to reduce the…

Machine Learning · Computer Science 2025-04-01 Wei Gao , Xinyu Zhou , Peng Sun , Tianwei Zhang , Yonggang Wen

Deploying Large Language Models (LLMs) on memory-constrained AI Personal Computers (AIPCs) enables low-latency, privacy-preserving inference, but long-context generation is fundamentally bottlenecked by the linearly growing Key-Value (KV)…

Hardware Architecture · Computer Science 2026-03-26 He Sun , Shinan Liu , Li Li , Mingjun Xiao

LLMs are seeing growing use for applications which require large context windows, and with these large context windows KV cache activations surface as the dominant contributor to memory consumption during inference. Quantization is a…

Machine Learning · Computer Science 2025-05-30 Coleman Hooper , Sehoon Kim , Hiva Mohammadzadeh , Michael W. Mahoney , Yakun Sophia Shao , Kurt Keutzer , Amir Gholami

Long-context Multimodal Large Language Models (MLLMs) demand substantial computational resources for inference as the growth of their multimodal Key-Value (KV) cache, in response to increasing input lengths, challenges memory and time…

Computation and Language · Computer Science 2024-06-27 Zhongwei Wan , Ziang Wu , Che Liu , Jinfa Huang , Zhihong Zhu , Peng Jin , Longyue Wang , Li Yuan

The use of Large Language Models (LLMs) for querying relational data has given rise to relQuery, a workload pattern that applies templated LLM calls to structured tables. As relQuery services become more widely adopted in applications such…

Databases · Computer Science 2026-01-21 Xin Zhang , Shihong Gao , Yanyan Shen , Haoyang Li , Lei Chen

Auto-regressive decoding in Large Language Models (LLMs) is inherently memory-bound: every generation step requires loading the model weights and intermediate results from memory (e.g., High-Bandwidth Memory (HBM) for GPU servers), making…

Machine Learning · Computer Science 2026-05-13 Yuning Han , Yangchenchen Jin , Dylan Zhao , Jingwei Sun

Large language models (LLMs) are increasingly used for long-content generation (e.g., long Chain-of-Thought reasoning) where decoding efficiency becomes a critical bottleneck: Autoregressive decoding is inherently limited by its sequential…

Computation and Language · Computer Science 2025-06-05 Zhepei Wei , Wei-Lin Chen , Xinyu Zhu , Yu Meng

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

The growing complexity of LLM usage today, e.g., multi-round conversation and retrieval-augmented generation (RAG), makes contextual states (i.e., KV cache) reusable across user requests. Given the capacity constraints of GPU memory, only a…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-10-08 Shiwei Gao , Youmin Chen , Jiwu Shu

Large language models (LLMs) have demonstrated remarkable performance, but their long-context reasoning remains constrained by the excessive memory required for the Key-Value (KV) cache. This makes KV cache compression a critical step…

Machine Learning · Computer Science 2025-09-30 Xianglong Yan , Zhiteng Li , Tianao Zhang , Haotong Qin , Linghe Kong , Yulun Zhang , Xiaokang Yang

The increasing complexity of AI tasks has shifted the paradigm from monolithic models toward multi-agent large language model (LLM) systems. However, these collaborative architectures introduce a critical bottleneck: redundant prefill…

Machine Learning · Computer Science 2026-03-17 Yingsheng Geng , Yuchong Gao , Weihong Wu , Guyue Liu , Jiang Liu