English
Related papers

Related papers: On the Limits of Learned Importance Scoring for KV…

200 papers

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…

Computation and Language · Computer Science 2024-10-08 Isaac Rehg

Limited-information inference on New Keynesian Phillips Curves (NKPCs) and other single-equation macroeconomic relations is characterised by weak and high-dimensional instrumental variables (IVs). Beyond the efficiency concerns previously…

General Economics · Economics 2021-03-23 Max-Sebastian Dovì

The long-output context generation of large reasoning models enables extended chain of thought (CoT) but also drives rapid growth of the key-value (KV) cache, quickly overwhelming GPU memory. To address this challenge, we propose ThinKV, a…

Machine Learning · Computer Science 2026-05-11 Akshat Ramachandran , Marina Neseem , Charbel Sakr , Rangharajan Venkatesan , Brucek Khailany , Tushar Krishna

The comparison of benchmark error sets is an essential tool for the evaluation of theories in computational chemistry. The standard ranking of methods by their Mean Unsigned Error is unsatisfactory for several reasons linked to the…

Methodology · Statistics 2020-09-29 Pascal Pernot , Andreas Savin

Memory retention mechanisms play a central role in determining the efficiency of computational architectures designed for processing extended sequences. Conventional methods for token management often impose fixed retention thresholds or…

Computation and Language · Computer Science 2025-03-26 Jonathan Delena , Augustin Moreau , Dominic Ravensdale , Frederick Chatterton

Explicit high-order feature interactions efficiently capture essential structural knowledge about the data of interest and have been used for constructing generative models. We present a supervised discriminative High-Order Parametric…

Artificial Intelligence · Computer Science 2016-08-17 Martin Renqiang Min , Hongyu Guo , Dongjin Song

Sample-level rankings are increasingly used in data-centric NLP for analysis, filtering, debugging, and curation, yet existing pipelines typically score training examples pointwise and rank them as if they were independent. This assumption…

Information Retrieval · Computer Science 2026-05-05 Xu Zheng , Feiyu Wu , Linhong Wu , Zhuocheng Wang , Hui Li

The Support Vector Machine using Privileged Information (SVM+) has been proposed to train a classifier to utilize the additional privileged information that is only available in the training phase but not available in the test phase. In…

Machine Learning · Computer Science 2016-04-07 Xinxing Xu , Joey Tianyi Zhou , IvorW. Tsang , Zheng Qin , Rick Siow Mong Goh , Yong Liu

Steering large language models (LLMs) is usually done by either instruction prompting or activation steering. Prompting often gives strong control, but caches guidance tokens at every layer and can clutter long interactions; activation…

Machine Learning · Computer Science 2026-05-12 Andy Zeyi Liu , Michael Zhang , Ilana Greenberg , Adam Alnasser , Lucas Baker , John Sous

Sequence learning reduces to similarity-based retrieval over a temporally indexed representation space, a constraint on any sequence model, not a property of a specific architecture. We show that a spiking Sparse Distributed Memory sequence…

Neural and Evolutionary Computing · Computer Science 2026-05-04 Joy Bose

The size of the key-value (KV) cache plays a critical role in determining both the maximum context length and the number of concurrent requests supported during inference in modern language models. The KV cache size grows proportionally…

Computation and Language · Computer Science 2024-09-24 Shashank Rajput , Ying Sheng , Sean Owen , Vitaliy Chiley

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…

Machine Learning · Computer Science 2024-06-13 Harry Dong , Xinyu Yang , Zhenyu Zhang , Zhangyang Wang , Yuejie Chi , Beidi Chen

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…

Machine Learning · Computer Science 2024-07-24 Hanlin Tang , Yang Lin , Jing Lin , Qingsen Han , Shikuan Hong , Yiwu Yao , Gongyi Wang

Deep learning for Information Retrieval (IR) requires a large amount of high-quality query-document relevance labels, but such labels are inherently sparse. Label smoothing redistributes some observed probability mass over unobserved…

Information Retrieval · Computer Science 2022-05-10 Jihyuk Kim , Minsoo Kim , Seung-won Hwang

Scaling large language models to long contexts is challenging due to the quadratic computational cost of full attention. Mitigation approaches include KV-cache selection or compression techniques. We instead provide an effective and…

Machine Learning · Computer Science 2026-04-24 Yuzhen Mao , Michael Y. Li , Emily B. Fox

KV cache offloading enables long-context LLM inference by storing caches in CPU DRAM, but PCIe bandwidth limitations create severe bottlenecks. In this paper, we develops an analytical framework that derives $\kappa_{\text{crit}}$, the…

Hardware Architecture · Computer Science 2026-01-29 William Meng , Benjamin Lee , Hong Wang

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…

Machine Learning · Computer Science 2026-03-24 Yichun Xu , Navjot K. Khaira , Tejinder Singh

Key-value (KV) caching is essential for large language model inference, yet its memory overhead poses a critical bottleneck for long-context generation. Existing eviction policies predominantly rely on empirical heuristics, lacking a…

Machine Learning · Computer Science 2026-04-30 Jiaming Yang , Chenwei Tang , Liangli Zhen , Jiancheng Lv

Vision transformer has emerged as a new paradigm in computer vision, showing excellent performance while accompanied by expensive computational cost. Image token pruning is one of the main approaches for ViT compression, due to the facts…

Computer Vision and Pattern Recognition · Computer Science 2023-07-07 Xiangcheng Liu , Tianyi Wu , Guodong Guo

Information Pursuit (IP) is an explainable prediction algorithm that greedily selects a sequence of interpretable queries about the data in order of information gain, updating its posterior at each step based on observed query-answer pairs.…

Computer Vision and Pattern Recognition · Computer Science 2025-08-06 Stefan Kolek , Aditya Chattopadhyay , Kwan Ho Ryan Chan , Hector Andrade-Loarca , Gitta Kutyniok , Réne Vidal