English
Related papers

Related papers: ALISA: Accelerating Large Language Model Inference…

200 papers

Large language models (LLMs) now support context windows of hundreds of thousands to millions of tokens, enabling applications such as long-document summarization, large-scale code synthesis, multi-document question answering and persistent…

Computation and Language · Computer Science 2025-10-22 Siyuan Yan , Guo-Qing Jiang , Yuchen Zhang , Xiaoxing Ma , Ran Zhu , Chun Cao , Jingwei Xu

With the widespread deployment of long-context large language models (LLMs), there has been a growing demand for efficient support of high-throughput inference. However, as the key-value (KV) cache expands with the sequence length, the…

Machine Learning · Computer Science 2025-04-29 Hanshi Sun , Li-Wen Chang , Wenlei Bao , Size Zheng , Ningxin Zheng , Xin Liu , Harry Dong , Yuejie Chi , Beidi Chen

While Large Vision-Language Models (LVLMs) demonstrate exceptional multi-modal capabilities, the quadratic computational cost of processing high-resolution visual tokens remains a critical bottleneck. Though recent token reduction…

Computer Vision and Pattern Recognition · Computer Science 2026-03-19 Surendra Pathak , Bo Han

Vision-Language Large Models (VLLMs) face significant efficiency challenges when processing high-resolution inputs. The quadratic complexity in attention and autoregressive generation, as well as the constantly growing key value (KV) cache…

Multimedia · Computer Science 2025-10-31 Zhonghua Jiang , Kunxi Li , Yiyun Zhou , Sihao Liu , Zhaode Wang , Chengfei lv , Shengyu Zhang

High throughput serving of large language models (LLMs) requires batching sufficiently many requests at a time. However, existing systems struggle because the key-value cache (KV cache) memory for each request is huge and grows and shrinks…

Machine Learning · Computer Science 2023-09-13 Woosuk Kwon , Zhuohan Li , Siyuan Zhuang , Ying Sheng , Lianmin Zheng , Cody Hao Yu , Joseph E. Gonzalez , Hao Zhang , Ion Stoica

Large Language Models are increasingly being deployed in datacenters. Serving these models requires careful memory management, as their memory usage includes static weights, dynamic activations, and key-value caches. While static weights…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-05-08 Jiale Xu , Rui Zhang , Yi Xiong , Cong Guo , Zihan Liu , Yangjie Zhou , Weiming Hu , Hao Wu , Changxu Shao , Ziqing Wang , Yongjie Yuan , Junping Zhao , Minyi Guo , Jingwen Leng

The Transformer architecture has significantly advanced deep learning, particularly in natural language processing, by effectively managing long-range dependencies. However, as the demand for understanding complex relationships grows,…

Computation and Language · Computer Science 2024-06-18 Qian Chen , Wen Wang , Qinglin Zhang , Siqi Zheng , Shiliang Zhang , Chong Deng , Hai Yu , Jiaqing Liu , Yukun Ma , Chong Zhang

Multi-head Latent Attention (MLA) significantly reduces KVCache memory usage in Large Language Models while introducing substantial computational overhead and intermediate variable expansion. This poses challenges for efficient hardware…

Machine Learning · Computer Science 2025-10-23 Qichen Liao , Chengqiu Hu , Fangzheng Miao , Bao Li , Yiyang Liu , Junlong Lyu , Lirui Jiang , Jun Wang , Lingchao Zheng , Jun Li , Yuwei Fan

Large language models (LLMs) have demonstrated remarkable performance across a wide range of language processing tasks. However, this success comes at the cost of substantial computation and memory requirements, which significantly impedes…

Machine Learning · Computer Science 2026-01-21 Fen-Yu Hsieh , Yun-Chang Teng , Ding-Yong Hong , Jan-Jan Wu

Transformer-based large language models (LLMs) demonstrate impressive performance in long context generation. Extending the context length has disproportionately shifted the memory footprint of LLMs during inference to the key-value cache…

Machine Learning · Computer Science 2025-02-19 Cheng Luo , Zefan Cai , Hanshi Sun , Jinqi Xiao , Bo Yuan , Wen Xiao , Junjie Hu , Jiawei Zhao , Beidi Chen , Anima Anandkumar

Transformer-based large language models (LLMs) excel in natural language processing tasks by capturing long-range dependencies through self-attention mechanisms. However, long-context modeling faces significant computational inefficiencies…

Computation and Language · Computer Science 2025-08-15 Shuhai Zhang , Zeng You , Yaofo Chen , Zhiquan Wen , Qianyue Wang , Zhijie Qiu , Yuanqing Li , Mingkui Tan

Modern multimodal large language models (MLLMs) adopt a unified self-attention design that processes visual and textual tokens at every Transformer layer, incurring substantial computational overhead. In this work, we revisit the necessity…

Computer Vision and Pattern Recognition · Computer Science 2026-05-28 Wenjie Liu , Hao Wu , Xin Qiu , Xudong Wang , Yingqi Fan , Yihan Zhang , Anhao Zhao , Yunpu Ma , Xiaoyu Shen

Inference for Large Language Models (LLMs) is computationally demanding. To reduce the cost of auto-regressive decoding, Key-Value (KV) cache is used to store intermediate activations, which significantly lowers the computational overhead…

Machine Learning · Computer Science 2025-06-05 Chaoyi Jiang , Lei Gao , Hossein Entezari Zarch , Murali Annavaram

Large language model (LLM) inference demands significant amount of computation and memory, especially in the key attention mechanism. While techniques, such as quantization and acceleration algorithms, like FlashAttention, have improved…

Machine Learning · Computer Science 2024-12-18 Hao Kang , Srikant Bharadwaj , James Hensman , Tushar Krishna , Victor Ruhle , Saravan Rajmohan

Large language models (LLMs) rely on key-value (KV) caches for efficient autoregressive decoding; however, cache size grows linearly with context length and model depth, becoming a major bottleneck in long-context inference. Prior KV cache…

Machine Learning · Computer Science 2025-09-22 Dmitry Akulov , Mohamed Sana , Antonio De Domenico , Tareq Si Salem , Nicola Piovesan , Fadhel Ayed

Large Language Models (LLMs) have been widely deployed in a variety of applications, and the context length is rapidly increasing to handle tasks such as long-document QA and complex logical reasoning. However, long context poses…

Machine Learning · Computer Science 2025-06-17 Guangda Liu , Chengwei Li , Jieru Zhao , Chenqi Zhang , Minyi Guo

Scaling Transformers to ultra-long contexts is bottlenecked by the $O(n^2 d)$ cost of self-attention. Existing methods reduce this cost along the sequence axis through local windows, kernel approximations, or token-level sparsity, but these…

Machine Learning · Computer Science 2026-03-31 Yan Xie , Tiansheng Wen , Tangda Huang , Bo Chen , Chenyu You , Stefanie Jegelka , Yifei Wang

The transformer's context window is vital for tasks such as few-shot learning and conditional generation as it preserves previous tokens for active memory. However, as the context lengths increase, the computational costs grow…

Computation and Language · Computer Science 2025-04-01 Jeffrey Willette , Heejun Lee , Youngwan Lee , Myeongjae Jeon , Sung Ju Hwang

Large Language Models (LLMs) excel at multi-step reasoning problems with explicit chain-of-thought (CoT), but verbose traces incur significant computational costs and memory overhead, and often carry redundant, stylistic artifacts. Latent…

Machine Learning · Computer Science 2026-05-08 Anna Kuzina , Maciej Pioro , Paul N. Whatmough , Babak Ehteshami Bejnordi

Large Language Models (LLMs) have significantly advanced the field of Artificial Intelligence. However, their deployment is resource-intensive, not only due to the large number of model parameters but also because the (Key-Value) KV cache…

Computation and Language · Computer Science 2025-06-05 Yifeng Gu , Zicong Jiang , Jianxiu Jin , Kailing Guo , Ziyang Zhang , Xiangmin Xu