跳到正文
arXiv · Machine Learning· Tianqi Chen, Xunchao Li, Linchuan Xie, Guanghua Yu, Jianchen Zhu, Peisong Wang, Jian Cheng·· 13 小时前AI 评分49

RAQ-MLA:面向多头潜在注意力缓存的角色感知量化

RAQ-MLA: Role-Aware Quantization for Multi-Head Latent Attention Caches

AI 导读

RAQ-MLA 针对 Multi-Head Latent Attention(MLA)缓存中 latent 与 positional key 承担不同作用的问题,提出无需训练的角色感知量化方法。该方法将完整 MLA 缓存量化至约 2.5 bits per element,使理论 KV-cache 容量提升 6.5x,并在四个 MLA 模型的推理和编码任务上保持近乎无损。

来源:arXiv · Machine Learning · arxiv.org