长上下文混合模型的机制 Part 1.1:从混合注意力到混合位置编码

Hugging Face Daily Papers 论文

摘要

本文分析了结合全注意力与滑动窗口注意力(SWA)或门控线性注意力(GLA/GDN)的长上下文混合大语言模型的机制,发现上下文扩展中存在跷跷板效应,并提出了滑动窗口线性注意力,实现了16倍免训练长度外推,在64k上下文下NIAH-SK1准确率达到100%。

大语言模型(LLM)的架构设计正在从传统的纯全注意力模型转向混合模型,通过组合不同的注意力模块来提升长上下文效率以及长度外推和上下文扩展的性能。为了解释混合模型为何有效以及如何更好地设计它们,我们提出了长上下文混合模型的机制这一系列研究。作为该系列的 Part 1.1,我们从全注意力与滑动窗口注意力(SWA)或线性注意力(LA)的门控变体(以 GLA 和 GDN 为代表)的混合模型开始。我们首先观察到上下文扩展中的跷跷板效应:LA 混合模型从长上下文持续预训练中获益更多,而 SWA 混合模型在长度外推方面表现更好。我们将这一现象归因于这些注意力机制所引入的位置归纳偏置的差异。我们发现 SWA 混合模型存在短上下文学习陷阱、短窗口倦怠和长窗口懒惰等问题,需要扩展窗口才能在持续长上下文预训练中提升性能。对于 LA 混合模型,我们总结了混合位置外推中的马太效应,并提出了滑动窗口线性注意力,实现了16倍免训练长度外推,同时在64k上下文长度下 NIAH-SK1 准确率保持在100%。
查看原文
查看缓存全文

缓存时间: 2026/10/08 12:30

Paper page - 长上下文混合模型机制 Part 1.1:从混合注意力到混合位置

Source: https://huggingface.co/papers/2610.10114

Abstract

大型语言模型(LLM)的架构设计正在从传统的纯全注意力模型转向混合模型,这类模型通过组合不同的注意力模块来提升长上下文效率以及长度外推和上下文扩展方面的性能。为了解释混合模型为何有效以及如何更好地设计它们,我们提出了「长上下文混合模型机制」(Mechanics of Long-Context Hybrid Models)系列。作为本系列的 Part 1.1,我们从全注意力与滑动窗口注意力(SWA)或门控线性注意力(LA)变体的混合开始研究,其中 LA 的代表包括 GLA 和 GDN。我们首先观察到上下文扩展中的「跷跷板效应」(Seesaw Effect):LA 混合模型从长上下文持续预训练中获益更多,而 SWA 混合模型在长度外推场景下表现更好。我们将这一行为归因于这些注意力机制所引入的位置归纳偏置(positional inductive biases)的差异。我们发现 SWA 混合模型会陷入「短上下文学习陷阱」(Short-Context Learning Trap),表现出「短窗口疲劳」(Short-Window Weariness)与「长窗口懒惰」(Long-Window Laziness),并且需要更长的窗口才能在持续长上下文预训练中提升性能。对于 LA 混合模型,我们总结了「混合位置外推的马太效应」(Matthew Effect of Hybrid Position Extrapolation),并提出了 Sliding-Window Linear Attention,实现了 16 倍的免训练长度外推,同时在 64k 上下文长度下于 NIAH-SK1 上保持 100% 准确率。

View arXiv page (https://arxiv.org/abs/2610.10114)View PDF (https://arxiv.org/pdf/2610.10114)GitHub2 (https://github.com/OpenMOSS/Hybrid-Mechanics)Add to collection (https://huggingface.co/login?next=%2Fpapers%2F2610.10114)

Get this paper in your agent:

hf papers read 2610.10114

Don’t have the latest CLI?curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper0

No model linking this paper

Cite arxiv.org/abs/2610.10114 in a model README.md to link it from this page.

Datasets citing this paper0

No dataset linking this paper

Cite arxiv.org/abs/2610.10114 in a dataset README.md to link it from this page.

Spaces citing this paper1

Collections including this paper0

No Collection including this paper

Add this paper to acollection (https://huggingface.co/new-collection)to link it from this page.

相似文章

重新思考高效注意力在混合架构中的作用

arXiv cs.CL

本文系统分析了高效注意力模块在混合语言模型架构中的作用,发现不同设计在充分训练下长上下文性能趋于一致,且长距离检索主要由全注意力承担,而高效注意力塑造了优化轨迹,揭示了一个称为“大窗口懒惰”的现象。

@seclink: https://x.com/seclink/status/2072187033263784397

X AI KOLs Timeline

Hybrid Sliding Window Attention (Hybrid SWA) 是一种在长文本语言模型中平衡计算效率与全局长距离依赖的混合注意力机制,通过交替使用局部SWA层和全局注意力层,显著压缩KV Cache同时保持推理能力。文章详细介绍了其设计原理、在Gemma、Qwen等模型中的应用,以及vLLM和HuggingFace等开源项目中的最佳实践。

Dynamic Linear Attention

Hugging Face Daily Papers

DLA引入了自适应状态合并和容量受限的内存建模,用于多状态线性注意力,提升了长上下文LLM的性能。

滑动窗口优于线性注意力

Hugging Face Daily Papers

带有sink的滑动窗口注意力在长上下文任务上优于经过后训练的线性注意力,无需重新训练,为LLMs提供更便宜、更可靠的推理解决方案。