研究人员推出了一种新方法 Kronecker Product Attention (KPA),用于压缩自回归模型的键值 (KV) 缓存。KPA 使用紧凑矩阵因子的 Kronecker 积之和来表示 KV 激活,这严格地推广了现有的技术,如 Tensor Product Attention。该方法旨在减少限制高效生成的内存占用,同时不牺牲模型的表达能力或需要复杂的 positional encoding 机制。经验表明,KPA 在各种模型规模和基准测试中均表现出具有竞争力或更优的性能,同时实现了高达 10 倍的 KV 缓存压缩。 AI
影响 该方法可以显著降低运行大型语言模型的内存需求,从而可能实现更高效的部署和可访问性。
排序理由 该条目是一篇研究论文,详细介绍了一种用于 LLM KV 缓存压缩的新方法。[lever_c_demoted from research: ic=1 ai=1.0]
- Grouped-Query Attention
- Hugging Face
- Kronecker Product Attention
- KV cache
- Multi-Head Latent Attention
- Multi-Query Attention
- RoPE
- Tensor Product Attention
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →