核心信息
一篇新论文表明,在后训练阶段,将标准注意力替换为带注意力汇聚的滑动窗口注意力掩码,可在不增加额外成本的情况下超越线性注意力。
要点
- 方法简单:改用带注意力汇聚的滑动窗口注意力掩码。
- 在后训练场景中,该方法优于线性注意力,尽管概念上非常直接。
- 该研究由多位合作者共同完成,完整论文已在线公开。
一篇新论文表明,在后训练阶段,将标准注意力替换为带注意力汇聚的滑动窗口注意力掩码,可在不增加额外成本的情况下超越线性注意力。
Simple beats complicated: We show that switching to a sliding-window attention mask with attention sinks (at no cost) beats linear attention post-training. Huge thanks to my collaborators @RheaSukthanker, @CameronPashmina, and @Emy_Aze. Paper: https://t.co/h8DIc223Su