中文
相关论文

相关论文: OCTOPUS: Optimized KV Cache for Transformers via O…

200 篇论文

Accurate decoding of quantum error-correcting codes is a crucial ingredient in protecting quantum information from decoherence. It requires characterizing the error channels corrupting the logical quantum state and providing this…

量子物理 · 物理学 2025-04-28 Volodymyr Sivak , Michael Newman , Paul Klimov

We give query complexity lower bounds for convex optimization and the related feasibility problem. We show that quadratic memory is necessary to achieve the optimal oracle complexity for first-order convex optimization. In particular, this…

机器学习 · 计算机科学 2023-05-22 Moïse Blanchard , Junhui Zhang , Patrick Jaillet

We compare two strategies for compressing the KV cache in transformer inference: rank reduction (discard dimensions) and quantization (keep all dimensions, reduce precision). At matched storage budgets across five models (124M-14B, MHA and…

机器学习 · 计算机科学 2026-04-14 Samuel Salfati

Post-training compression of Transformer models commonly relies on truncated singular value decomposition (SVD). However, enforcing a single shared subspace can degrade accuracy even at moderate compression. Sparse dictionary learning…

机器学习 · 计算机科学 2026-02-18 Denis Makhov , Dmitriy Shopkhoev , Magauiya Zhussip , Ammar Ali , Baher Mohammad , Stamatios Lefkimmiatis

The quantum capacity of a memoryless channel is often used as a single figure of merit to characterize its ability to transmit quantum information coherently. The capacity determines the maximal rate at which we can code reliably over…

量子物理 · 物理学 2016-05-10 Marco Tomamichel , Mario Berta , Joseph M. Renes

The linear growth of Key-Value (KV) cache remains a bottleneck for multi-turn LLM deployment. Existing KV cache compression methods often fail to account for the structural properties of multi-turn dialogues, relying on heuristic eviction…

计算与语言 · 计算机科学 2026-01-30 Hong Chen , Xiang Liu , Bo Wang , Yuxuan Fan , Yuanlin Chu , Zongluo Li , Xiaowen Chu , Xuming Hu

Key-value (KV) cache compression has emerged as a critical technique for reducing the memory and latency overhead of autoregressive language models during inference. Prior approaches predominantly rely on query-key attention scores to rank…

计算与语言 · 计算机科学 2025-09-19 Ayan Sengupta , Siddhant Chaudhary , Tanmoy Chakraborty

In this paper, we study a variant of the quadratic penalty method for linearly constrained convex problems, which has already been widely used but actually lacks theoretical justification. Namely, the penalty parameter steadily increases…

数值分析 · 数学 2017-11-30 Huan Li , Cong Fang , Zhouchen Lin

The key-value (KV) cache is a primary memory bottleneck in Transformers. We propose Low-Rank Key-Value (LRKV) attention, which reduces KV cache memory by exploiting redundancy across attention heads, while being compute efficient. Each…

机器学习 · 计算机科学 2026-04-09 James O'Neill , Robert Clancy , Mariia Matskevichus , Fergal Reid

Transformers have become central to natural language processing and large language models, but their deployment at scale faces three major challenges. First, the attention mechanism requires massive matrix multiplications and frequent…

硬件体系结构 · 计算机科学 2026-01-22 Xiaoxuan Yang , Peilin Chen , Tergel Molom-Ochir , Yiran Chen

Long-context inference is increasingly constrained by the KV cache: resident memory grows with context length, and decoding becomes limited by repeated High Bandwidth Memory (HBM) streaming rather than arithmetic. Existing methods such as…

We present an exact $n$-qubit computational-basis amplitude encoder of real- or complex-valued data vectors of $d=\binom{n}{k}$ components into a subspace of fixed Hamming weight $k$. This represents a polynomial space compression of degree…

Video large language models (VideoLLMs) have demonstrated the capability to process longer video inputs and enable complex reasoning and analysis. However, due to the thousands of visual tokens from the video frames, the key-value (KV)…

计算机视觉与模式识别 · 计算机科学 2025-09-30 Keda Tao , Haoxuan You , Yang Sui , Can Qin , Huan Wang

The memory-for-computation paradigm of KV caching is essential for accelerating large language model (LLM) inference service, but limited GPU high-bandwidth memory (HBM) capacity motivates offloading the KV cache to cheaper external storage…

In this paper, we propose a novel optimizer state compression algorithm, namely $\pi$-Quant, which leverages the properties of irrational numbers (e.g., $\pi$) for memory-efficient training. The core idea is based on our mathematical…

机器学习 · 计算机科学 2025-01-23 Zhen Tian , Wayne Xin Zhao , Ji-Rong Wen

Large Language Models (LLMs) have been widely adopted to process long-context tasks. However, the large memory overhead of the key-value (KV) cache poses significant challenges in long-context scenarios. Existing training-free KV cache…

机器学习 · 计算机科学 2024-10-22 Luning Wang , Shiyao Li , Xuefei Ning , Zhihang Yuan , Shengen Yan , Guohao Dai , Yu Wang

Fracton topological phases have a large number of materialized symmetries that enforce a rigid structure on their excitations. Remarkably, we find that the symmetries of a quantum error-correcting code based on a fracton phase enable us to…

量子物理 · 物理学 2020-04-02 Benjamin J. Brown , Dominic J. Williamson

This paper presents a novel network compression framework Kernel Quantization (KQ), targeting to efficiently convert any pre-trained full-precision convolutional neural network (CNN) model into a low-precision version without significant…

机器学习 · 计算机科学 2020-03-12 Zhongzhi Yu , Yemin Shi , Tiejun Huang , Yizhou Yu

Autoregressive (AR) visual generation has achieved remarkable performance but suffers from high memory usage and low throughput, as it requires caching previously generated visual tokens. Recent research has shown that retaining only a few…

计算机视觉与模式识别 · 计算机科学 2026-05-21 Guotao Liang , Baoquan Zhang , Zhiyuan Wen , Yunming Ye

Extended reasoning in large language models (LLMs) creates severe KV cache memory bottlenecks. Leading KV cache compression methods estimate KV importance using attention scores from recent post-RoPE queries. However, queries rotate with…

计算与语言 · 计算机科学 2026-04-07 Weian Mao , Xi Lin , Wei Huang , Yuxin Xie , Tianfu Fu , Bohan Zhuang , Song Han , Yukang Chen