arXiv:2605.22337cs.AI2026-05

动态生成软令牌压缩键值缓存,缓解长文本推理的内存与效率瓶颈。

Meta-Soft: Leveraging Composable Meta-Tokens for Context-Preserving KV Cache Compression

论文配图:Meta-Soft: Leveraging Composable Meta-Tokens for Context-Preserving KV Cache Compression
图 1 · 摘自论文原文
  • 通过可学习基矩阵和门控采样,动态合成针对输入的软令牌
  • 在多个数据集上优于现有方法,保留95%以上上下文信息
  • 适合长文本生成、高并发推理等场景

大型语言模型中的键值缓存(KV cache)随上下文长度线性增长,导致内存爆炸和解码效率下降。现有基于固定软令牌(如Judge Q)的淘汰方法依赖静态参数评估重要性,无法适应不同输入,且淘汰后信息永久丢失,造成上下文断裂。为此,我们提出Meta-Soft,一种基于探针驱动的上下文融合动态压缩框架。构建可学习正交基矩阵$\\)。实验表明,该方法在多个数据集上超越现有最优淘汰方法,为KV缓存压缩提供新方案。

原文摘要 · Abstract (English)

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 direction; however, existing methods based on fixed Soft Tokens (e.g., Judge Q) rely on a static parameter set as the query to evaluate the importance of KV pairs, so they cannot adapt dynamically to different input prompts, and they cannot precisely capture complex and changing task relevance. Also, evicted KV pairs are discarded permanently, so this causes irreversible information loss and context breaks. To address this problem, we propose Meta-Soft, a dynamic compression framework based on probe-driven context integration. Specifically, we build a meta-library with a learnable orthogonal basis matrix $\mathcal{L}$, and we use a selector network with Gumbel-Softmax to produce differentiable sparse combination weights, so we dynamically synthesize the most targeted $k$ Soft Tokens from the input prompt features. We append these Soft Tokens to the end of the input sequence to probe key information. We also introduce an attention-flow based integration mechanism, which redistributes the semantic information of removed tokens into retained tokens, and this keeps the dropped context information effectively. Experiments on multiple datasets show that our method outperforms existing state-of-the-art eviction methods and provides a new solution for KV Cache compression.

KV缓存压缩大模型推理动态生成软令牌

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。