arXiv cs.LG·21 天前★ 3

Kronecker 乘积注意力:结构化分解压缩 KV 缓存

原文标题:Kronecker Product Attention: Memory-Efficient KV Compression via Structured Factorization

重要度 3/5 · AI 综合评估报道影响力与新鲜度,4–5 分为高价值资讯。

AI 智能摘要

论文提出 Kronecker Product (KPA),将键值激活表示为紧凑、依赖 的矩阵因子之 Kronecker 乘积和。KPA 严格泛化 TPA,并配合结构化旋转嵌入保持 相对位置性质。在四个模型规模、九个下游基准上匹配或超越 MHA、MQA、GQA、MLA,实现最高 10 倍 KV 缓存压缩。

深读 · 结论先行AI 生成展开看要点与影响 ↓

KPA用Kronecker乘积分解KV激活,在四个规模九个基准上匹配或超越MHA/GQA/MLA,实现最高10倍KV缓存压缩。

  • •将键值激活表示为紧凑、依赖token的矩阵因子之Kronecker乘积和
  • •严格泛化TPA:秩一向量外积分分解是其退化情形,非退化矩阵因子可表达更丰富的块结构模式
  • •配套结构化旋转嵌入,只旋转紧凑因子即可保持RoPE相对位置性质,无需MLA式的解耦RoPE分支
  • •在四个模型规模、九个下游基准、相同训练条件下匹配或超越TPA、MHA、MQA、GQA、MLA
  • •最高实现10倍KV缓存压缩,Kronecker头切分是可调超参,用于权衡压缩与精度

为什么重要:KV缓存是的主要显存瓶颈,10倍压缩且不牺牲精度、还省掉解耦分支,对做与长部署的团队有直接价值。

以上深读由 AI 依据标题/摘要/正文节选生成,仅供参考;事实以原文为准。

阅读原文 +0
AI 处理过程展开/收起
原始标题:Kronecker Product Attention: Memory-Efficient KV Compression via Structured Factorization
AI 中文标题:Kronecker 乘积注意力:结构化分解压缩 KV 缓存
来源:arXiv cs.LG · 发布于 21 天前
分类:大模型 · AI 重要度:★ 3(3)
标签:注意力机制 / KV 缓存 / 模型压缩 / Kronecker 乘积 / 推理效率
关键词:KV 缓存压缩 / 注意力机制 / Kronecker 乘积注意力 / 大模型推理优化 / RoPE
AI 处理:2026/10/12 12:30:14 · 模型 deepseek-chat · 规则版本 2026-10-10.1
AI 生成以上内容由 AI 自动生成:中文标题、摘要、分类与重要度;同一事件的多家报道会自动归并展示。 内容仅供参考,可能存在偏差,请以原文为准, 不构成任何投资或专业建议。
版权归原作者与原发布平台所有;若涉及侵权或信息错误,请在举报与侵权投诉页反馈,我们会及时处理。
每天 09:30,3 条 AI 圈真正重要的变化,免费订阅 →
Kronecker 乘积注意力:结构化分解压缩 KV 缓存 - 智讯 AI News