中文
相关论文

相关论文: NestedKV: Nested Memory Routing for Long-Context K…

200 篇论文

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…

计算与语言 · 计算机科学 2025-04-01 Jeffrey Willette , Heejun Lee , Youngwan Lee , Myeongjae Jeon , Sung Ju Hwang

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

The key-value (KV) cache is a foundational optimization in Transformer-based large language models (LLMs), eliminating redundant recomputation of past token representations during autoregressive generation. However, its memory footprint…

机器学习 · 计算机科学 2026-03-24 Yichun Xu , Navjot K. Khaira , Tejinder Singh

While Key-Value (KV) cache succeeds in reducing redundant computations in auto-regressive models, it introduces significant memory overhead, limiting its practical deployment in long-sequence scenarios. Existing KV retrieval methods…

机器学习 · 计算机科学 2025-10-14 Wenbo Wu , Qingyi Si , Xiurui Pan , Ye Wang , Jie Zhang

Long-context language modeling is increasingly constrained by the Key-Value (KV) cache, whose memory and decode-time access costs scale linearly with the prefix length. This bottleneck has motivated a range of context-compression methods,…

机器学习 · 计算机科学 2026-05-13 Yoav Gelberg , Yam Eitan , Michael Bronstein , Yarin Gal , Haggai Maron

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

Large Language Models (LLMs) excel across a variety of language tasks yet are constrained by limited input lengths and high computational costs. Existing approaches\textemdash such as relative positional encodings (e.g., RoPE, ALiBi) and…

计算与语言 · 计算机科学 2025-02-18 Kun-Hui Lee , Eunhwan Park , Donghoon Han , Seung-Hoon Na

Agentic code tasks such as fault localization and patch generation require processing long codebases under tight memory constraints, where the Key-Value (KV) cache becomes the primary inference bottleneck. Existing compression methods rely…

计算与语言 · 计算机科学 2026-04-14 Qiujiang Chen , Jing Xiong , Chenyang Zhao , Sidi Yang , Ngai Wong

Large Language Models capable of handling extended contexts are in high demand, yet their inference remains challenging due to substantial Key-Value cache size and high memory bandwidth requirements. Previous research has demonstrated that…

机器学习 · 计算机科学 2025-10-29 Junlin Mu , Hantao Huang , Jihang Zhang , Minghui Yu , Tao Wang , Yidong Li

Many computational factors limit broader deployment of large language models. In this paper, we focus on a memory bottleneck imposed by the key-value (KV) cache, a computational shortcut that requires storing previous KV pairs during…

机器学习 · 计算机科学 2024-06-13 Harry Dong , Xinyu Yang , Zhenyu Zhang , Zhangyang Wang , Yuejie Chi , Beidi Chen

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…

计算与语言 · 计算机科学 2025-06-05 Yifeng Gu , Zicong Jiang , Jianxiu Jin , Kailing Guo , Ziyang Zhang , Xiangmin Xu

Long-horizon LLM inference turns the key--value (KV) cache into the dominant GPU memory consumer and makes per-token attention increasingly expensive. Many common eviction policies use static recency windows or historical attention, leaving…

机器学习 · 计算机科学 2026-05-26 Yubo Li , Yidi Miao

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

Serving large language models (LLMs) at scale necessitates efficient key-value (KV) cache management. KV caches can be reused across conversation turns via shared-prefix prompts that are common in iterative code editing and chat. However,…

计算与语言 · 计算机科学 2026-03-12 Konrad Staniszewski , Adrian Łańcucki

KV cache compression methods have mainly relied on scalar quantization techniques to reduce the memory requirements during decoding. In this work, we apply residual vector quantization, which has been widely used for high fidelity audio…

机器学习 · 计算机科学 2024-10-22 Ankur Kumar

Large Language Models (LLMs) have revolutionized a wide range of domains such as natural language processing, computer vision, and multi-modal tasks due to their ability to comprehend context and perform logical reasoning. However, the…

人工智能 · 计算机科学 2025-07-31 Haoyang Li , Yiming Li , Anxin Tian , Tianhao Tang , Zhanchao Xu , Xuejia Chen , Nicole Hu , Wei Dong , Qing Li , Lei Chen

Large language models (LLMs) are widely deployed with rapidly expanding context windows to support increasingly demanding applications. However, long contexts pose significant deployment challenges, primarily due to the KV cache whose size…

机器学习 · 计算机科学 2026-03-10 Guangda Liu , Chengwei Li , Zhenyu Ning , Jing Lin , Yiwu Yao , Danning Ke , Minyi Guo , Jieru Zhao

With reasoning becoming the generative paradigm for large language models (LLMs), the memory bottleneck caused by KV cache during the decoding phase has become a critical factor limiting high-concurrency service. Although existing KV cache…

分布式、并行与集群计算 · 计算机科学 2026-03-11 Mengqi Liao , Lu Wang , Chaoyun Zhang , Bo Qiao , Si Qin , Qingwei Lin , Saravan Rajmohan , Dongmei Zhang , Huaiyu Wan

The key-value (KV) cache in transformer models is a critical component for efficient decoding or inference, yet its memory demands scale poorly with sequence length, posing a major challenge for scalable deployment of large language models.…

计算与语言 · 计算机科学 2025-06-06 Vinay Joshi , Pratik Prabhanjan Brahma , Zicheng Liu , Emad Barsoum

Reasoning large language models exhibit complex reasoning behaviors via extended chain-of-thought generation that are highly fragile to information loss during decoding, creating critical challenges for KV cache compression. Existing…

计算与语言 · 计算机科学 2026-05-28 Wenjie Du , Li Jiang , Keda Tao , Xue Liu , Huan Wang