长上下文混合模型的机制 Part 1.1:从混合注意力到混合位置编码
摘要
本文分析了结合全注意力与滑动窗口注意力(SWA)或门控线性注意力(GLA/GDN)的长上下文混合大语言模型的机制,发现上下文扩展中存在跷跷板效应,并提出了滑动窗口线性注意力,实现了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.
相似文章
重新思考高效注意力在混合架构中的作用
本文系统分析了高效注意力模块在混合语言模型架构中的作用,发现不同设计在充分训练下长上下文性能趋于一致,且长距离检索主要由全注意力承担,而高效注意力塑造了优化轨迹,揭示了一个称为“大窗口懒惰”的现象。
@seclink: https://x.com/seclink/status/2072187033263784397
Hybrid Sliding Window Attention (Hybrid SWA) 是一种在长文本语言模型中平衡计算效率与全局长距离依赖的混合注意力机制,通过交替使用局部SWA层和全局注意力层,显著压缩KV Cache同时保持推理能力。文章详细介绍了其设计原理、在Gemma、Qwen等模型中的应用,以及vLLM和HuggingFace等开源项目中的最佳实践。
无需训练的滑动窗口适配中NLL引导的全注意力层选择
提出了一种无需训练的NLL引导方法,用于在混合注意力模型中选择保留全注意力的层,在长上下文任务中,使用1/4全注意力层即可达到与1/2周期性基线相当的准确率。
Dynamic Linear Attention
DLA引入了自适应状态合并和容量受限的内存建模,用于多状态线性注意力,提升了长上下文LLM的性能。
滑动窗口优于线性注意力
带有sink的滑动窗口注意力在长上下文任务上优于经过后训练的线性注意力,无需重新训练,为LLMs提供更便宜、更可靠的推理解决方案。