跳到正文
arXiv · AI· Minsoo Cheong, Sungjoo Yoo·· 8 小时前AI 评分24

LoopVQ:循环 Transformer 中 KV 缓存低于 1 bit 的量化方法

LoopVQ: Sub 1-bit KV Cache Quantization for Looped Transformers

AI 导读

针对循环 Transformer 中 KV 缓存随循环步长线性增长导致内存瓶颈的问题,LoopVQ 提出一种低于 1 bit 的 KV 缓存向量量化方法,并在 Ouro 和 Huginn 模型上实现 0.5 bit 每元素的精度保持,甚至在 0.25 bit 时结合 TaSQ 仍能保留大部分 BF16 准确率。

来源:arXiv · AI · arxiv.org