跳到正文
热点事件持续更新

注意力偏置折叠进键,压缩KV缓存兼容标准内核

1 篇报道1 个报道来源12 小时前更新

先了解这件事

AI 综述

研究者 Utkarsh Kanwat 提出一种把 Attention Matching 的逐 key 偏置折叠进 key 的 KV 缓存压缩方法:沿每个 head 的一条闭式方向、用上下文查询拟合,把偏置并入 key,使压缩后的缓存变成普通缓存,尺寸和 head 维度与 Attention Matching 一致,并沿用模型自身的 query 路径。据该研究称,这样得到的缓存可直接被 FlashAttention 和 vLLM 的 paged cache 使用,无需专用内核。目前尚无独立验证或实际部署效果的数据。

AI 根据报道生成 · 3 小时前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月28日
  1. arXiv cs.CL
    将注意力偏置折叠进 Key:压缩 KV 缓存可直接跑在标准内核上

    研究者提出把 Attention Matching 的逐 key 偏置沿每个 head 的一条闭式方向折叠进 key,使压缩后的 KV 缓存变成普通缓存,从而能被 FlashAttention 和 vLLM 的 paged cache 直接使用。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。