中文
相关论文

相关论文: KeyDiff: Key Similarity-Based KV Cache Eviction fo…

200 篇论文

Recent reasoning models such as OpenAI-o1 and DeepSeek-R1 have shown strong performance on complex tasks including mathematical reasoning and code generation. However, this performance gain comes with substantially longer output sequences,…

机器学习 · 计算机科学 2026-04-28 Yi Su , Zhenxu Tian , Dan Qiao , Yuechi Zhou , Juntao Li , Min Zhang

Large Audio-Language Models (LALMs) have set new benchmarks in speech processing, yet their deployment is hindered by the memory footprint of the Key-Value (KV) cache during long-context inference. While general KV cache compression…

声音 · 计算机科学 2026-04-09 Yuxuan Wang , Peize He , Xiyan Gui , Xiaoqian Liu , Junhao He , Xuyang Liu , Zichen Wen , Xuming Hu , Linfeng Zhang

Large reasoning models (LRMs) often incur significant key-value (KV) cache overhead, due to their linear growth with the verbose chain-of-thought (CoT) reasoning. This incurs both memory overhead and throughput bottlenecks, limiting…

Given the quadratic complexity of attention, KV cache eviction is vital to accelerate model inference. Current KV cache eviction methods typically rely on instantaneous heuristic metrics, implicitly assuming that score magnitudes are…

机器学习 · 计算机科学 2026-02-10 Ziyao Tang , Pengkun Jiao , Xinhang Chen , Wei Liu , Shiyong Li , Jingjing Chen

Transformer-based large language models (LLMs) have achieved impressive performance in various natural language processing (NLP) applications. However, the high memory and computation cost induced by the KV cache limits the inference…

硬件体系结构 · 计算机科学 2025-04-11 Weikai Xu , Wenxuan Zeng , Qianqian Huang , Meng Li , Ru Huang

AlayaDB is a cutting-edge vector database system natively architected for efficient and effective long-context inference for Large Language Models (LLMs) at AlayaDB AI. Specifically, it decouples the KV cache and attention computation from…

As modern LLMs support thousands to millions of tokens, KV caches grow to hundreds of gigabytes, stressing memory capacity and bandwidth. Existing solutions, such as KV cache pruning and offloading, alleviate these but underutilize hardware…

性能 · 计算机科学 2026-04-21 Mao Lin , Xi Wang , Guilherme Cox , Dong Li , Hyeran Jeon

Recent advances in Large Language Models (LLMs) have highlighted the critical importance of extending context length, yet the quadratic complexity of attention mechanisms poses significant challenges for efficient long-context modeling. KV…

计算与语言 · 计算机科学 2025-11-07 Wanyun Cui , Mingwei Xu

As long-context language modeling becomes increasingly important, the cost of maintaining and attending to large Key/Value (KV) caches grows rapidly, becoming a major bottleneck in both training and inference. While prior works such as…

机器学习 · 计算机科学 2026-03-25 Dong Liu , Yanxuan Yu , Ben Lengerich , Ying Nian Wu

Autoregressive large language models (LLMs) have made remarkable progress in various natural language generation tasks. However, they incur high computation cost and latency resulting from the autoregressive token-by-token generation. To…

计算与语言 · 计算机科学 2023-07-07 Luciano Del Corro , Allie Del Giorno , Sahaj Agarwal , Bin Yu , Ahmed Awadallah , Subhabrata Mukherjee

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

The increasing memory demand of the Key-Value (KV) cache poses a significant bottleneck for Large Language Models (LLMs) in long-context applications. Existing low-rank KV compression methods reduce this footprint by modifying model…

计算与语言 · 计算机科学 2026-05-14 Shiyu Ji , Yixuan Wang , Yijun Liu , Qingfu Zhu , Wanxiang Che

Compressing the KV cache is a required step to deploy large language models on edge devices. Current quantization methods compress storage but fail to reduce bandwidth as attention calculation requires dequantizing keys from INT4/INT8 to…

机器学习 · 计算机科学 2026-01-16 Aryan Karmore

Key-value (KV) caching is critical for efficient inference in large language models (LLMs), yet its memory footprint scales linearly with context length, resulting in a severe scalability bottleneck. Existing approaches largely treat KV…

计算与语言 · 计算机科学 2026-04-23 Gradwell Dzikanyanga , Weihao Yang , Hao Huang , Donglei Wu , Shihao Wang , Wen Xia , Sanjeeb K C

Large Language Models (LLMs) have shown remarkable comprehension abilities but face challenges in GPU memory usage during inference, hindering their scalability for real-time applications like chatbots. To accelerate inference, we store…

计算与语言 · 计算机科学 2024-06-06 Dongjie Yang , XiaoDong Han , Yan Gao , Yao Hu , Shilin Zhang , Hai Zhao

Transformers, driven by attention mechanisms, form the foundation of large language models (LLMs). As these models scale up, efficient GPU attention kernels become essential for high-throughput and low-latency inference. Diverse LLM…

分布式、并行与集群计算 · 计算机科学 2025-04-23 Zihao Ye , Lequn Chen , Ruihang Lai , Wuwei Lin , Yineng Zhang , Stephanie Wang , Tianqi Chen , Baris Kasikci , Vinod Grover , Arvind Krishnamurthy , Luis Ceze

As context windows in LLMs scale to 100K+ tokens, the key-value (KV) cache becomes the dominant memory bottleneck, with recent methods claiming 80-90% savings and minimal benchmark degradation. We argue these evaluations miss a structural…

计算与语言 · 计算机科学 2026-03-03 Samhruth Ananthanarayanan , Ayan Sengupta , Tanmoy Chakraborty

Large Language Model or LLM inference has two phases, the prompt (or prefill) phase to output the first token and the extension (or decoding) phase to the generate subsequent tokens. In this work, we propose an efficient parallelization…

分布式、并行与集群计算 · 计算机科学 2024-05-15 Minsik Cho , Mohammad Rastegari , Devang Naik

Large language models (LLMs) with extended context windows have become increasingly prevalent for tackling complex tasks. However, the substantial Key-Value (KV) cache required for long-context LLMs poses significant deployment challenges.…

计算与语言 · 计算机科学 2025-06-16 Jie Hu , Shengnan Wang , Yutong He , Ping Gong , Jiawei Yi , Juncheng Zhang , Youhui Bai , Renhai Chen , Gong Zhang , Cheng Li , Kun Yuan

Data engineering workflows require reliable differencing across files, databases, and query outputs, yet existing tools falter under schema drift, heterogeneous types, and limited explainability. SmartDiff is a unified system that combines…

数据库 · 计算机科学 2025-09-03 Aryan Poduri , Yashwant Tailor