中文
相关论文

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

200 篇论文

We introduce Lexico, a novel KV cache compression method that leverages sparse coding with a universal dictionary. Our key finding is that key-value cache in modern LLMs can be accurately approximated using sparse linear combination from a…

机器学习 · 计算机科学 2024-12-13 Junhyuck Kim , Jongho Park , Jaewoong Cho , Dimitris Papailiopoulos

Large Language Models (LLMs) have achieved impressive accomplishments in recent years. However, the increasing memory consumption of KV cache has possessed a significant challenge to the inference system. Eviction methods have revealed the…

计算与语言 · 计算机科学 2025-07-10 Zicong Tang , Shi Luohe , Zuchao Li , Baoyuan Qi , Guoming Liu , Lefei Zhang , Ping Wang

The memory and computational demands of Key-Value (KV) cache present significant challenges for deploying long-context language models. Previous approaches attempt to mitigate this issue by selectively dropping tokens, which irreversibly…

机器学习 · 计算机科学 2024-07-24 Hanlin Tang , Yang Lin , Jing Lin , Qingsen Han , Shikuan Hong , Yiwu Yao , Gongyi Wang

The Key-Value (KV) cache is crucial for efficient Large Language Models (LLMs) inference, but excessively long contexts drastically increase KV cache memory footprint. Existing KV cache compression methods typically rely on input-side…

计算与语言 · 计算机科学 2026-03-13 Zhenxu Tian , Yi Su , Juntao Li , Min Zhang

Transformer-based large language models (LLMs) have already achieved remarkable results on long-text tasks, but the limited GPU memory (VRAM) resources struggle to accommodate the linearly growing demand for key-value (KV) cache as the…

计算与语言 · 计算机科学 2025-03-21 Shibo Jie , Yehui Tang , Kai Han , Zhi-Hong Deng , Jing Han

Efficient KV cache management in LLMs is crucial for long-context tasks like RAG and summarization. Existing KV cache compression methods enforce a fixed pattern, neglecting task-specific characteristics and reducing the retention of…

计算与语言 · 计算机科学 2025-05-28 Xiabin Zhou , Wenbin Wang , Minyan Zeng , Jiaxian Guo , Xuebo Liu , Li Shen , Min Zhang , Liang Ding

Visual autoregressive modeling (VAR) via next-scale prediction has emerged as a scalable image generation paradigm. While Key and Value (KV) caching in large language models (LLMs) has been extensively studied, next-scale prediction…

计算机视觉与模式识别 · 计算机科学 2025-11-21 Boxun Xu , Yu Wang , Zihu Wang , Peng Li

Vector Quantization (VQ) is an appealing model compression method to obtain a tiny model with less accuracy loss. While methods to obtain better codebooks and codes under fixed clustering dimensionality have been extensively studied,…

计算机视觉与模式识别 · 计算机科学 2022-11-22 Zezhou Zhu , Yucong Zhou , Zhao Zhong

Large Language Models (LLMs) are increasingly deployed in large-scale online services, enabling sophisticated applications. However, the computational overhead of generating key-value (KV) caches in the prefill stage presents a major…

机器学习 · 计算机科学 2025-02-24 Shuowei Jin , Xueshen Liu , Qingzhao Zhang , Z. Morley Mao

Large Language Models (LLMs) are increasingly being deployed on edge devices for long-context settings, creating a growing need for fast and efficient long-context inference. In these scenarios, the Key-Value (KV) cache is the primary…

In large language models performing long-form reasoning, the KV cache grows rapidly with decode length, creating bottlenecks in memory and inference stability. Existing reasoning-oriented KV compression has mostly followed an…

人工智能 · 计算机科学 2026-04-14 Buseong Kim , Heejun Gwon

For the efficient inference of Large Language Models (LLMs), the effective compression of key-value (KV) cache is essential. Three main types of KV cache compression techniques, namely sparsity, channel compression, and quantization, have…

机器学习 · 计算机科学 2025-02-24 Hong Yankun , Li Xing , Zhen Hui-Ling , Yu Xianzhi , Liu Wulong , Yuan Mingxuan

Large language models face significant computational and memory challenges when processing long contexts. During inference, efficient management of the key-value (KV) cache, which stores intermediate activations for autoregressive…

计算与语言 · 计算机科学 2025-09-30 Yuxuan Zhu , Ali Falahati , David H. Yang , Mohammad Mohammadi Amiri

Despite the remarkable progress of Large Language Models (LLMs), the escalating memory footprint of the Key-Value (KV) cache remains a critical bottleneck for efficient inference. While dimensionality reduction offers a promising…

计算与语言 · 计算机科学 2026-03-06 Liming Lu , Kaixi Qiu , Jiayu Zhou , Jushi Kai , Haoyan Zhang , Huanyu Wang , Jingwen Leng , Ziwei He , Zhouhan Lin

Key-Value (KV) cache has become a de facto component of modern Large Vision-Language Models (LVLMs) for inference. While it enhances decoding efficiency in Large Language Models (LLMs), its direct adoption in LVLMs introduces substantial…

计算机视觉与模式识别 · 计算机科学 2026-05-04 Xihao Chen , Yangyang Guo , Roger Zimmermann

While LLMs are powerful embedding backbones, their application in training-free settings faces two structural challenges: causal attention restricts early tokens from accessing subsequent context, and the next-token prediction objective…

计算与语言 · 计算机科学 2026-01-06 Yixuan Tang , Yi Yang

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…

In this study, we investigate whether attention-based information flow inside large language models (LLMs) is aggregated through noticeable patterns for long context processing. Our observations reveal that LLMs aggregate information…

计算与语言 · 计算机科学 2025-05-16 Zefan Cai , Yichi Zhang , Bofei Gao , Yuliang Liu , Yucheng Li , Tianyu Liu , Keming Lu , Wayne Xiong , Yue Dong , Junjie Hu , Wen Xiao

Long-running agentic tasks, such as deep research, require multi-hop reasoning over information distributed across multiple webpages and documents. In such tasks, the LLM context is dominated by tokens from external retrieval, causing…

人工智能 · 计算机科学 2026-03-03 Sanjay Kariyappa , G. Edward Suh

KV cache stores key and value states from previous tokens to avoid re-computation, yet it demands substantial storage space, especially for long sequences. Adaptive KV cache compression seeks to discern the saliency of tokens, preserving…

机器学习 · 计算机科学 2024-05-24 Yefei He , Luoming Zhang , Weijia Wu , Jing Liu , Hong Zhou , Bohan Zhuang