中文
相关论文

相关论文: KV-CoRE: Benchmarking Data-Dependent Low-Rank Comp…

200 篇论文

Scaling language models to long contexts is often bottlenecked by the size of the key-value (KV) cache. In deployed settings, long contexts are typically managed through compaction in token space via summarization. However, summarization…

机器学习 · 计算机科学 2026-05-28 Adam Zweiger , Xinghong Fu , Han Guo , Yoon Kim

Multimodal Large Language Models (MLLMs) have demonstrated remarkable performance across diverse applications. However, their computational overhead during deployment remains a critical bottleneck. While Key-Value (KV) caching effectively…

计算机视觉与模式识别 · 计算机科学 2025-03-26 Insu Han , Zeliang Zhang , Zhiyuan Wang , Yifan Zhu , Susan Liang , Jiani Liu , Haiting Lin , Mingjie Zhao , Chenliang Xu , Kun Wan , Wentian Zhao

Large Language Models (LLMs) face limitations due to the high demand on GPU memory and computational resources when handling long contexts. While sparsify the Key-Value (KV) cache of transformer model is a typical strategy to alleviate…

机器学习 · 计算机科学 2024-02-06 Yumeng Wang , Zhenyang Xiao

With the development of large language models (LLMs), efficient inference through Key-Value (KV) cache compression has attracted considerable attention, especially for long-context generation. To compress the KV cache, recent methods…

计算与语言 · 计算机科学 2025-10-28 Qingyue Yang , Jie Wang , Xing Li , Zhihai Wang , Chen Chen , Lei Chen , Xianzhi Yu , Wulong Liu , Jianye Hao , Mingxuan Yuan , Bin Li

Large Language Models (LLMs) suffer inference-time memory bottlenecks dominated by the attention Key-Value (KV) cache, which scales with model size and context length. While KV-cache quantization alleviates this cost, bit allocation between…

机器学习 · 计算机科学 2026-05-12 Mohsen Hariri , Alan Luo , Weicong Chen , Shaochen Zhong , Tianyi Zhang , Qifan Wang , Xia Hu , Xiaotian Han , Vipin Chaudhary

Large language models (LLMs) rely on Key-Value (KV) cache to reduce time-to-first-token (TTFT) latency, but existing disk-based KV cache systems using file-per-object layouts suffer from severe scalability bottlenecks due to file system…

数据库 · 计算机科学 2025-11-26 Weiping Yu , Ye Jiarui , He Mengke , Junfeng Liu , Siqiang Luo

KV caching is a fundamental technique for accelerating Large Language Model (LLM) inference by reusing key-value (KV) pairs from previous queries, but its effectiveness under limited memory is highly sensitive to the eviction policy. The…

机器学习 · 计算机科学 2026-01-28 Fangzhou Wu , Sandeep Silwal , Qiuyi , Zhang

While Key-Value (KV) cache compression is essential for efficient LLM inference, current evaluations disproportionately focus on sparse retrieval tasks, potentially masking the degradation of High-Density Reasoning where Chain-of-Thought…

计算与语言 · 计算机科学 2026-05-13 Xiang Liu , Zhenheng Tang , Hong Chen , Peijie Dong , Zeyu Li , Xiuze Zhou , Bo Li , Xuming Hu , Xiaowen Chu

Rotary Position Embedding (RoPE) enables each attention head to capture multi-frequency information along the sequence dimension and is widely applied in foundation models. However, the nonlinearity introduced by RoPE complicates…

机器学习 · 计算机科学 2025-03-04 Yuhao Zhou , Sirui Song , Boyang Liu , Zhiheng Xi , Senjie Jin , Xiaoran Fan , Zhihao Zhang , Wei Li , Xuanjing Huang

Recent reasoning large language models (LLMs) excel in complex tasks but encounter significant computational and memory challenges due to long sequence lengths. KV cache compression has emerged as an effective approach to greatly enhance…

计算与语言 · 计算机科学 2025-12-02 Mengqi Liao , Lu Wang , Chaoyun Zhang , Zekai Shen , Xiaowei Mao , Si Qin , Qingwei Lin , Saravan Rajmohan , Dongmei Zhang , Huaiyu Wan

Recently, large language models (LLMs) have been able to handle longer and longer contexts. However, a context that is too long may cause intolerant inference latency and GPU memory usage. Existing methods propose mixed-precision…

计算与语言 · 计算机科学 2025-04-01 Wei Tao , Bin Zhang , Xiaoyang Qu , Jiguang Wan , Jianzong Wang

KV cache in autoregressive LLMs eliminates redundant recomputation but has emerged as the dominant memory and bandwidth bottleneck during inference, notably with long contexts and test-time scaling. KV quantization is a key lever for…

机器学习 · 计算机科学 2026-02-03 Ji Zhang , Yiwei Li , Shaoxiong Feng , Peiwen Yuan , Xinglin Wang , Jiayi Shi , Yueqi Zhang , Chuyi Tan , Boyuan Pan , Yao Hu , Kan Li

Diffusion-based large language models (dLLMs) rely on bidirectional attention, which prevents lossless KV caching and requires a full forward pass at every denoising step. Existing approximate KV caching methods reduce this cost by…

计算与语言 · 计算机科学 2026-03-20 Minsoo Cheong , Donghyun Son , Woosang Lim , Sungjoo Yoo

LLMs are widely adopted in production, pushing inference systems to their limits. Disaggregated LLM serving (e.g., PD separation and KV state disaggregation) improves scalability and cost efficiency, but it also turns KV into an explicit…

分布式、并行与集群计算 · 计算机科学 2026-05-14 Zedong Liu , Xinyang Ma , Dejun Luo , Hairui Zhao , Bing Lu , Wenjing Huang , Yida Gu , Xingchen Liu , Zheng Wei , Jinyang Liu , Dingwen Tao , Guangming Tan

Key-Value (KV) cache has become a bottleneck of LLMs for long-context generation. Despite the numerous efforts in this area, the optimization for the decoding phase is generally ignored. However, we believe such optimization is crucial,…

计算与语言 · 计算机科学 2025-06-04 Jialong Wu , Zhenglin Wang , Linhai Zhang , Yilong Lai , Yulan He , Deyu Zhou

Large Language Models (LLMs) require substantial computational resources during generation. While the Key-Value (KV) cache significantly accelerates this process by storing attention intermediates, its memory footprint grows linearly with…

计算与语言 · 计算机科学 2025-08-05 Yi Su , Quantong Qiu , Yuechi Zhou , Juntao Li , Qingrong Xia , Ping Li , Xinyu Duan , Zhefeng Wang , Min Zhang

The linear memory growth of the KV cache poses a significant bottleneck for LLM inference in long-context tasks. Existing static compression methods often fail to preserve globally important information. Although recent dynamic retrieval…

计算与语言 · 计算机科学 2026-04-21 Zhiyuan Shi , Qibo Qiu , Feng Xue , Zhonglin Jiang , Li Yu , Jian Jiang , Xiaofei He , Wenxiao Wang

Key-value (KV) cache compression has emerged as a critical technique for reducing the memory and latency overhead of autoregressive language models during inference. Prior approaches predominantly rely on query-key attention scores to rank…

计算与语言 · 计算机科学 2025-09-19 Ayan Sengupta , Siddhant Chaudhary , Tanmoy Chakraborty

Efficient deployment of Large Language Models (LLMs) requires batching multiple requests together to improve throughput. As the batch size, context length, or model size increases, the size of the key and value (KV) cache can quickly become…

机器学习 · 计算机科学 2024-05-08 Tianyi Zhang , Jonah Yi , Zhaozhuo Xu , Anshumali Shrivastava

Growing context lengths in transformer-based language models have made the key-value (KV) cache a critical inference bottleneck. While many KV cache pruning methods have been proposed, they have not yet been adopted in major inference…

机器学习 · 计算机科学 2026-02-04 Simon Jegou , Maximilian Jeblick