跳到正文
原文
arXiv · Machine Learning Theory· Bahareh Nikpour, Hossein Rajabzadeh, Maryam Dialameh, Vahid Partovi Nia, Parsa Omidi·· 2 小时前AI 评分37

Kronecker Product Attention:用结构化分解实现内存高效的 KV 压缩

Kronecker Product Attention: Memory-Efficient KV Compression via Structured Factorization

AI 导读

研究者提出 Kronecker Product Attention(KPA),将 key 和 value 激活表示为紧凑的、依赖 token 的矩阵因子的 Kronecker 乘积之和,严格泛化了 Tensor Product Attention(TPA)。

来源:arXiv · Machine Learning Theory · arxiv.org