English
Related papers

Related papers: GPU-Accelerated INT8 Quantization for KV Cache Com…

200 papers

Large Language Models (LLMs) are increasingly deployed in scenarios demanding ultra-long context reasoning, such as agentic workflows and deep research understanding. However, long-context inference is constrained by the KV cache, a…

Hardware Architecture · Computer Science 2026-03-11 Jianlong Lei , Shashikant Ilager

Improving the deployment efficiency of transformer-based language models has been challenging given their high computation and memory cost. While INT8 quantization has recently been shown to be effective in reducing both the memory cost and…

Computation and Language · Computer Science 2023-06-01 Xiaoxia Wu , Cheng Li , Reza Yazdani Aminabadi , Zhewei Yao , Yuxiong He

This paper presents an optimized methodology to design and deploy Speech Enhancement (SE) algorithms based on Recurrent Neural Networks (RNNs) on a state-of-the-art MicroController Unit (MCU), with 1+8 general-purpose RISC-V cores. To…

Sound · Computer Science 2022-10-17 Manuele Rusci , Marco Fariselli , Martin Croome , Francesco Paci , Eric Flamand

Quantization is a key method for reducing the GPU memory requirement of training large language models (LLMs). Yet, current approaches are ineffective for 4-bit activations and 8-bit gradients, which would easily cause slow convergence or…

Computation and Language · Computer Science 2026-05-12 Wenxiang Lin , Juntao Huang , Luhan Zhang , Laili Li , Xiang Bao , Mengyang Zhang , Bing Wang , Shaohuai Shi

Key-value~(KV) caching is an important technique to accelerate the inference of large language models~(LLMs), but incurs significant memory overhead. To compress the size of KV cache, existing methods often compromise precision or require…

Computation and Language · Computer Science 2024-05-22 Peiyu Liu , Ze-Feng Gao , Wayne Xin Zhao , Yipeng Ma , Tao Wang , Ji-Rong Wen

Large Language Models (LLMs) have been emerging as prominent AI models for solving many natural language tasks due to their high performance (e.g., accuracy) and capabilities in generating high-quality responses to the given inputs.…

Neural and Evolutionary Computing · Computer Science 2026-04-22 Rachmad Vidya Wicaksana Putra , Pasindu Wickramasinghe , Muhammad Shafique

As the length of input text increases, the key-value (KV) cache in LLMs imposes prohibitive GPU memory costs and limits long-context inference on resource constrained devices. Existing approaches, such as KV quantization and pruning, reduce…

Machine Learning · Computer Science 2025-12-24 Tenghui Li , Guoxu Zhou , Xuyang Zhao , Yuning Qiu , Qibin Zhao

The increasing adoption of large language models (LLMs) on heterogeneous computing platforms poses significant challenges to achieving high inference efficiency. To address these efficiency bottlenecks across diverse platforms, this paper…

Distributed, Parallel, and Cluster Computing · Computer Science 2026-02-06 Yaozheng Zhang , Wei Wang , Jie Kong , Jiehan Zhou , Xianwei Zhang , Huanqing Cui , Han Bao , Yuhai Liu

Optimizing the Key-Value (KV) cache of the Large Language Model (LLM) has been considered critical to saving the cost of inference. Most of the existing KV-cache compression algorithms attempted to sparsify the sequence of tokens by taking…

Machine Learning · Computer Science 2024-10-11 Zihao Wang , Bin Cui , Shaoduo Gan

Cost of serving large language models (LLM) is high, but the expensive and scarce GPUs are poorly efficient when generating tokens sequentially, unless the batch of sequences is enlarged. However, the batch size is limited by some…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-03-19 Jiaao He , Jidong Zhai

QLoRA reduces the memory-cost of fine-tuning a large language model (LLM) with LoRA by quantizing the base LLM. However, quantization introduces quantization errors that negatively impact model performance after fine-tuning. In this paper…

Machine Learning · Computer Science 2024-10-22 Neal Lawton , Aishwarya Padmakumar , Judith Gaspers , Jack FitzGerald , Anoop Kumar , Greg Ver Steeg , Aram Galstyan

Several works have developed eviction policies to remove key-value (KV) pairs from the KV cache for more efficient inference. The focus has been on compressing the KV cache after the input prompt has been processed for faster token…

Computation and Language · Computer Science 2025-07-04 Michael R. Metel , Boxing Chen , Mehdi Rezagholizadeh

The increasing size of the Key-Value (KV) cache during the Large Language Models long-context inference is the main obstacle for its balance between the deployment cost and task accuracy. To reduce the KV cache size in such scenarios, most…

Machine Learning · Computer Science 2025-07-25 Manlai Liang , JiaMing Zhang , Xiong Li , Jinlong Li

Large Language Models (LLMs) have made remarkable progress in processing extensive contexts, with the Key-Value (KV) cache playing a vital role in enhancing their performance. However, the growth of the KV cache in response to increasing…

Computation and Language · Computer Science 2024-06-18 Yuhong Li , Yingbing Huang , Bowen Yang , Bharat Venkitesh , Acyr Locatelli , Hanchen Ye , Tianle Cai , Patrick Lewis , Deming Chen

Recent large language models (LLMs) are rapidly extending their context windows, yet inference throughput lags due to increasing GPU memory and bandwidth demands. This is because the key-value (KV) cache, an intermediate structure storing…

Efficient key-value (KV) cache management is crucial for the practical deployment of large language models (LLMs), yet existing compression techniques often incur a trade-off between performance degradation and computational overhead. We…

Machine Learning · Computer Science 2026-02-10 Jang-Hyun Kim , Dongyoon Han , Sangdoo Yun

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

Efficient inference of large language models (LLMs) is hindered by an ever-growing key-value (KV) cache, making KV cache compression a critical research direction. Traditional methods selectively evict less important KV cache entries, which…

Machine Learning · Computer Science 2025-12-01 Yuxuan Tian , Zihan Wang , Yebo Peng , Aomufei Yuan , Zhiming Wang , Bairen Yi , Xin Liu , Yong Cui , Tong Yang

Reinforcement learning (RL) for large language models (LLMs) is increasingly bottlenecked by rollout (generation), where long output sequence lengths make attention and KV-cache memory dominate end-to-end step time. FP8 offers an attractive…

Machine Learning · Computer Science 2026-04-13 Zhaopeng Qiu , Shuang Yu , Jingqi Zhang , Shuai Zhang , Xue Huang , Jingyi Yang , Junjie Lai

Large language models (LLMs) have demonstrated impressive abilities in various domains while the inference cost is expensive. Many previous studies exploit quantization methods to reduce LLM inference cost by reducing latency and memory…

Machine Learning · Computer Science 2024-11-12 Jinhao Li , Jiaming Xu , Shiyao Li , Shan Huang , Jun Liu , Yaoxiu Lian , Guohao Dai