中文
相关论文

相关论文: UniGist: Towards General and Hardware-aligned Sequ…

200 篇论文

Scaling the input context length of a large language model (LLM) incurs a significant increase in computation cost and memory footprint to maintain the attention key-value (KV) cache. Existing KV cache compression methods suffer from…

计算与语言 · 计算机科学 2025-01-31 Yuxiang Huang , Binhang Yuan , Xu Han , Chaojun Xiao , Zhiyuan Liu

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

Context lengths of Large Language Models (LLMs) have exploded in recent years, with 128k-token context becoming a standard and million-token context becoming a reality. Efficiently supporting long-context inference remains challenging as…

计算与语言 · 计算机科学 2024-10-08 Isaac Rehg

The key-value (KV) cache is a major bottleneck in long-context inference, where memory and computation grow with sequence length. Existing KV eviction methods reduce this cost but typically degrade performance relative to full-cache…

机器学习 · 计算机科学 2026-05-12 Ngoc Bui , Hieu Trung Nguyen , Arman Cohan , Rex Ying

As the context length of current large language models (LLMs) rapidly increases, the memory demand for the Key-Value (KV) cache is becoming a bottleneck for LLM deployment and batch processing. Traditional KV cache compression methods…

计算与语言 · 计算机科学 2025-12-23 Aomufei Yuan , Zhiming Wang , Ruijie Miao , Dayu Wang , Yuxuan Tian , Zihan Wang , Yebo Peng , Yuhan Wu , Bairen Yi , Xin Liu , Tong Yang

Incorporating external knowledge in large language models (LLMs) enhances their utility across diverse applications, but existing methods have trade-offs. Retrieval-Augmented Generation (RAG) fetches evidence via similarity search, but key…

计算与语言 · 计算机科学 2025-03-10 Giulio Corallo , Orion Weller , Fabio Petroni , Paolo Papotti

The Key-Value (KV) cache in generative large language models (LLMs) introduces substantial memory overhead. Existing works mitigate this burden by offloading or compressing the KV cache. However, loading the entire cache incurs significant…

计算与语言 · 计算机科学 2025-05-28 Dingyu Yao , Bowen Shen , Zheng Lin , Wei Liu , Jian Luan , Bin Wang , Weiping Wang

Large Language Model (LLM) inference is increasingly constrained by GPU memory capacity rather than compute throughput, driven by growing model sizes and the linear growth of the key-value (KV) cache during autoregressive decoding. Existing…

机器学习 · 计算机科学 2026-02-03 Nikhil Gopal , Kostis Kaffes

Key-Value (KV) cache remains a major bottleneck for deploying Large Language Models (LLMs) in long-generation tasks. Prior work often applies uniform compression across both prefill and decoding caches, but compressing the prefill cache…

人工智能 · 计算机科学 2026-05-29 Soumyadeep Jana , Sagar Nishad , Sanasam Ranbir Singh

Scaling language models to longer contexts is essential for capturing rich dependencies across extended discourse. However, na\"ive context extension imposes significant computational and memory burdens, often resulting in inefficiencies…

计算与语言 · 计算机科学 2026-02-03 Wenhao Li , Bangcheng Sun , Weihao Ye , Tianyi Zhang , Daohai Yu , Fei Chao , Rongrong Ji

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

The KV cache used in large language models has linearly growing time complexity, so LLMs face memory blow-up and reduced decoding efficiency when they process long contexts. Current KV Cache eviction has become an important research…

人工智能 · 计算机科学 2026-05-26 Wei Luo , Yi Huang , Songchen Ma , Huanyu Qu , Jiang Cai , Mingkun Xu

Large Language Models (LLMs) face computational inefficiencies and redundant processing when handling long context inputs, prompting a focus on compression techniques. While existing semantic vector-based compression methods achieve…

计算与语言 · 计算机科学 2025-02-18 Shaoshen Chen , Yangning Li , Zishan Xu , Yinghui Li , Xin Su , Zifei Shan , Hai-tao Zheng

Large language models (LLMs) increasingly rely on long-context modeling for tasks such as document understanding, code analysis, and multi-step reasoning. However, scaling context windows to the million-token level brings prohibitive…

计算机视觉与模式识别 · 计算机科学 2025-10-22 Jiale Cheng , Yusen Liu , Xinyu Zhang , Yulin Fei , Wenyi Hong , Ruiliang Lyu , Weihan Wang , Zhe Su , Xiaotao Gu , Xiao Liu , Yushi Bai , Jie Tang , Hongning Wang , Minlie Huang

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…

分布式、并行与集群计算 · 计算机科学 2025-08-27 Zhiqiang Xie , Ziyi Xu , Mark Zhao , Yuwei An , Vikram Sharma Mailthody , Scott Mahlke , Michael Garland , Christos Kozyrakis

Large Language models (LLMs) have become a research hotspot. To accelerate the inference of LLMs, storing computed caches in memory has become the standard technique. However, as the inference length increases, growing KV caches might lead…

计算与语言 · 计算机科学 2024-12-13 Meizhi Zhong , Xikai Liu , Chen Zhang , Yikun Lei , Yan Gao , Yao Hu , Kehai Chen , Min Zhang

Rapid advances in Large Language Models (LLMs) have spurred demand for processing extended context sequences in contemporary applications. However, this progress faces two challenges: performance degradation due to sequence lengths…

计算与语言 · 计算机科学 2025-10-10 Wei Wu , Zhuoshi Pan , Chao Wang , Liyi Chen , Yunchu Bai , Tianfu Wang , Kun Fu , Zheng Wang , Hui Xiong

Long-context processing is a critical ability that constrains the applicability of large language models (LLMs). Although there exist various methods devoted to enhancing the long-context processing ability of LLMs, they are developed in an…

Large Language Models (LLMs) have made significant strides in natural language processing and generation, yet their ability to handle long-context input remains constrained by the quadratic complexity of attention computation and…

计算与语言 · 计算机科学 2025-06-16 Manlai Liang , Wanyi Huang , Mandi Liu , Huaijun Li , Jinlong Li

Long-range tasks demand reasoning over long inputs. However, existing solutions are limited, e.g., long-context models require large compute budgets, parameter-efficient fine-tuning (PEFT) needs training data, and retrieval-augmented…

人工智能 · 计算机科学 2025-08-26 Dulhan Jayalath , James Bradley Wendt , Nicholas Monath , Sandeep Tata , Beliz Gunel