arXiv:2511.06029cs.LG2025-11AAAI被引 7

让推理型大模型生成更高效,动态修剪缓存提升速度。

Lethe: Layer- and Time-Adaptive KV Cache Pruning for Reasoning-Intensive LLM Serving

  • 按层分配删减预算,根据注意力冗余程度智能决策
  • 多轮生成中结合新旧信息重要性,减少缓存占用
  • 适合长文本推理场景,显著提升吞吐量

生成式推理任务常涉及长序列解码,导致键值(KV)缓存积累带来巨大内存与延迟开销。现有压缩方法主要针对长输入序列的预填充阶段,难以应对推理过程中动态变化且分层敏感的缓存需求。本文提出Lethe框架,实现解码过程在空间与时间维度上的自适应管理。空间上,基于注意力冗余估计,对各层变压器层进行稀疏感知的剪枝预算分配;时间上,通过一种考虑演化注意力模式的“近期感知选择性保留”机制,在生成过程中执行多轮令牌剪枝。实验证明,Lethe在多种模型和任务上实现了效率与生成质量的良好平衡,吞吐量最高提升2.56倍。

原文摘要 · Abstract (English)

Generative reasoning with large language models (LLMs) often involves long decoding sequences, leading to substantial memory and latency overheads from accumulating key-value (KV) caches. While existing KV compression methods primarily focus on reducing prefill memory from long input sequences, they fall short in addressing the dynamic and layer-sensitive nature of long-form generation, which is central to reasoning tasks. We propose Lethe, a dynamic KV cache management framework that introduces adaptivity along both the spatial and temporal dimensions of decoding. Along the spatial dimension, Lethe performs layerwise sparsity-aware allocation, assigning token pruning budgets to each transformer layer based on estimated attention redundancy. Along the temporal dimension, Lethe conducts multi-round token pruning during generation, driven by a Recency-Aware Selective Retention} (RASR) mechanism. RASR extends traditional recency-based heuristics by also considering token relevance derived from evolving attention patterns, enabling informed decisions about which tokens to retain or evict. Empirical results demonstrate that Lethe achieves a favorable balance between efficiency and generation quality across diverse models and tasks, increases throughput by up to 2.56x.

大模型推理缓存优化生成效率

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