pause-tokens

标签

Cards List
#pause-tokens

AbridgeKV:将长推理压缩至学习得到的暂停词元的 KV 缓存中

arXiv cs.CL ↗ · 4天前 缓存

AbridgeKV 训练模型,将已生成完毕的思维链文本块压缩到少量学习得到的暂停 token 的 KV 条目中,在仅使用完整 KV 预算四分之一的情况下,实现缓存缩小至原来的 1/3.5,并在满负载 H100 上将解码速度提升至 3.2 倍,且准确率无损。

0 人收藏 0 人点赞
← 返回首页

提交意见反馈