滑动窗口注意力在长语境任务优于线性注意力

iScienceLuvr · x · 2026-08-31

研究发现,带 sinks 的滑动窗口注意力 (SWA) 在多个 LLM 和下游任务上的表现与线性注意力模型相当或更好。特别是在长语境推理任务 (Needle-in-a-Haystack, BABILong) 中,SWA 性能比线性注意力高出 2 到 10 倍。SWA 无需后训练,速度极快且内存占用低,是降低推理内存成本的廉价可靠方案。相比之下,线性注意力模型若要匹配 SWA 性能,可能需要从零开始训练或大量后训练。

所属事件:带sinks的滑动窗口注意力零训练胜过线性注意力(7 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →