元软:利用可组合的元标记实现上下文保持的KV缓存压缩 / Meta-Soft: Leveraging Composable Meta-Tokens for Context-Preserving KV Cache Compression
1️⃣ 一句话总结
这篇论文提出了一种名为Meta-Soft的新方法,通过动态生成可组合的软标记来压缩大语言模型中的KV缓存,在高效减少内存占用的同时,避免了传统方法因丢弃缓存导致的信息丢失和上下文断裂问题。
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 this http URL 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 this http URL, 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 this http URL 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 this http URL 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缓存压缩 / Meta-Soft: Leveraging Composable Meta-Tokens for Context-Preserving KV Cache Compression
这篇论文提出了一种名为Meta-Soft的新方法,通过动态生成可组合的软标记来压缩大语言模型中的KV缓存,在高效减少内存占用的同时,避免了传统方法因丢弃缓存导致的信息丢失和上下文断裂问题。
源自 arXiv: 2605.22337