相对位置注意力提速:新算法将成 FlashAttention 式即插即用方案

研究者提出一种精确注意力算法,把相对位置项直接融入 FlashAttention 式分块计算,并给出统一的 Triton 实现。若论文结论成立,长期困扰相对位置注意力的显存与带宽开销有望明显下降。

一句话看懂:研究者提出一种精确注意力算法,把相对位置项直接融入 FlashAttention 式分块计算,并给出统一的 Triton 实现。若论文结论成立,长期困扰相对位置注意力的显存与带宽开销有望明显下降。

事件核心:发生了什么

根据 Hugging Face Papers 收录的论文摘要,相对位置注意力在音频等领域表现优于 RoPE,但它的位置项依赖 token 对之间的偏移量,需要成对计算或查表,难以塞进融合注意力内核,序列变长后内存、带宽和算力开销迅速放大。作者提出的方法采用偏移感知分块、跨 token 对复用查询与键的关系乘积,并在反向传播中用基于偏移的梯度分片与聚合来避免原子竞争。摘要给出的测试条件是 8K token 序列:前向加速 1.8–6.4 倍,反向加速 2.0–21 倍,峰值内存最多降低 2.9–128 倍(前向)和 1.4–54 倍(反向);长上下文端到端转写快 1.8 倍、省 2.4 倍内存。需注意,摘要未说明同行评审状态,上述数据也仅对应其设定的基线与任务。

为什么重要

FlashAttention 已成为大模型训练和推理的基础设施,但相对位置注意力长期被挡在高效内核之外。这项工作试图证明“精确算法 + 结构感知内核”可以兼得,而不必退回近似或额外查表。对于需要在长音频、长文档上使用位置偏置模型的研究与工程团队,这意味着训练成本和推理显存可能同时下降,也降低了相对位置方案与 RoPE 生态竞争的门槛。

对用户/开发者/创作者的影响

开发者层面,统一的 Triton 实现若开源可得,有望作为 drop-in 模块替换现有注意力内核,而不改动模型结构;长上下文转写、音频理解等应用的部署成本可能受益。创作者端短期内难以直接感知,更多体现在长音频转写、长文档处理类工具的速度与可用长度上。企业采购时,可将其视为推理降本的技术储备,而非成熟商用产品。

值得关注的后续

一是代码与内核是否公开发布,以及能否在主流训练框架中稳定接入;二是独立复现能否在更大序列和不同相对位置变体上保持加速;三是音频、长文本等场景的实际产品是否跟进采用。

来源:Hugging Face Papers

celebrityanime
celebrityanime
文章: 33662

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注