混合线性注意力大语言模型中的巨激活:全注意力前尖峰与尖峰间平台 Massive Activations in Hybrid Linear Attention Large Language Models: Pre-Attention Spikes and Inter-Spike Plateaus
首次揭示混合线性注意力LLM中巨激活的前尖峰与平台形态,并用取消时机统一解释
前置知识
巨激活(Massive Activations, MAs)
指LLM隐藏状态中极少数数值异常巨大的激活分量,其绝对值可比典型激活高出几个数量级,且集中出现在特定token位置(如序列首token、换行/句号等标点、the/is/of 等虚词)。后续研究表明它们并非数值噪声,而是与注意力行为深度耦合的系统性现象,是Transformer内部动力学的标志性特征。
本文的研究对象就是巨激活:整篇论文的核心是在混合线性注意力架构中追踪这些超大幅度激活的位置、跨层演化和形成机制,并给出新的形态学分类。
注意力沉降(Attention Sink)
Softmax注意力中某些token(尤其是序列第一个token)会吸引不成比例的注意力质量,即使它们几乎不携带语义信息。主流解释是softmax每行归一化到1的约束使多余注意力'预算'需要泄压口,而携带巨激活的token恰好承担这一角色。
作者发现在混合模型中'按幅度排序'已不能可靠锁定巨激活token,因此改用跨全注意力层平均的'共识沉降'作为跨层追踪锚点,sink识别是全文方法论的地基。
线性注意力与循环状态
线性注意力用固定大小的循环状态替代token两两交互:$S_t = F_t S_{t-1} + k_t^\top v_t$,$y_t = q_t S_t W_O$,计算与显存随序列长度线性增长。变体区别在状态结构与更新规则:RetNet用固定指数衰减,HGRN/GLA用输入依赖门控,DeltaNet/GDN用delta规则修正历史键值关联。
本文覆盖上述全部五种线性骨干,考察它们与全注意力交错堆叠时巨激活如何被重塑;理解循环状态的'无逐token交互'特性是理解PAS/ISP成因的前提。
混合线性注意力(HLA)与混合化比率
HLA在深度方向以层为单位交替堆叠线性注意力层与少量全注意力层,兼顾线性效率与召回能力。混合化比率 $\rho = L/L_{FA}$ 表示每个全注意力层对应多少个序列混合层:$\rho=1$ 退化为全注意力,$\rho=\infty$ 为纯线性注意力,$\rho$ 越大全注意力越稀疏。
论文的核心经验规律是巨激活形态随 $ ho$ 系统变化:PAS在所有配置下存在,ISP则随全注意力变密而不断增强,$ ho$ 是贯穿全文的关键自变量。
系统性离群值生命周期
把巨激活视为残差流中的'有符号离群事件':某一层的更新在特定token-特征坐标 $(t, j)$ 写入极大值,随后另一层以反号更新将其抵消。写入、沉降耦合与抵消的相对时序决定了离群值在深度上的形态,这一框架源自近期对Transformer系统离群值的研究。
论文第4节用该框架统一解释所有现象:PAS对应局部的'写-沉-抵消',ISP对应'延迟抵消',全注意力的稳定巨激活对应进一步推迟的抵消,构成连续谱。
输出门控(Output Gating)
在序列混合层输出上施加逐元素门控,如 $o_t = g_t \odot W_O y_t$,其中 $g_t$ 由输入经线性投影加sigmoid产生。GDN等线性层原生带输出门;近期工作表明给全注意力加输出门控可以消除巨激活和attention sink。
论文用两组对照干预检验门控的作用:给全注意力加门(GDN-GatedFA)强烈衰减巨激活幅度,去掉GDN原生门(GDN-NoOutGate)只带来温和放大,揭示全注意力在组织巨激活中的中心地位。
研究动机
Softmax全注意力的计算与显存开销随上下文长度二次增长,线性注意力虽可通过固定大小循环状态 $S_t = F_t S_{t-1} + k_t^\top v_t$ 实现线性复杂度,但状态容量受限、召回密集任务表现差,因此Qwen3-Next、Qwen3.5、Kimi Linear、Kimi K3、Zamba、Nemotron-H等新一代模型普遍采用层交错混合架构。然而,混合化如何重塑模型内部激活动力学几乎无人研究。巨激活(比典型激活大几个数量级、集中于特定token的稀疏分量)在此前的文献中被认为依赖softmax注意力的结构性质并与attention sink强耦合,已有的基于幅度排序的追踪方法和KV cache量化等下游应用都建立在全注意力模型的经验之上。这些结论能否迁移到HLA模型、巨激活在混合架构中是否仍存在、以何种形态存在,在本文之前完全是空白。
本文的目标是本文的中心问题是:混合化如何塑造HLA LLM中的巨激活动力学,这些动力学又揭示了其内部计算的什么规律?具体目标分解为四步:(1)建立可靠的巨激活跨层追踪方法——作者发现现有'按幅度排序'的做法在HLA模型中与注意力行为脱钩,必须修正;(2)在受控条件下验证形态普遍性:覆盖RetNet/HGRN/GLA/DeltaNet/GDN五种线性架构、$\rho \in \{1,3,6,12,24,\infty\}$ 六种混合配置、五个数据域,以及Qwen3.5、Kimi Linear、Nemotron-H、Zamba2等1.2B–397B参数的开源模型;(3)通过340M/1.3B规模的从零受控预训练,追溯PAS与ISP在训练过程中的涌现时序,并分离全注意力门控与GDN门控的不对称作用;(4)给出机制层面的统一解释,把混合模型的巨激活形态与全注意力LLM的已知稳定形态连接成连续谱。
与已有工作不同的是,本文的独特切入有三点。其一,研究对象上,这是对层交错HLA LLM巨激活的首次系统研究,发现了两种此前未知的架构对齐形态——全注意力前尖峰(PAS)与尖峰间平台(ISP),并把全注意力LLM的稳定巨激活形态解释为二者在 $\rho \to 1$ 极限下的合并,建立了从纯线性到全注意力的形态学连续谱。其二,方法论上,作者观察到HLA模型中sink token的幅度排名跨层不稳定、最大激活token在相邻层间频繁切换(Figure 2),据此放弃幅度排序、改用跨全注意力层聚合注意力质量得到的'共识沉降'作锚点,这是对既有巨激活追踪协议的重要修正。其三,机制解释上不止于现象描述,而是把有符号离群值分析延伸到固定坐标 $(t^\star, j^\star)$,用'抵消时机'这一单一变量统一PAS、ISP与全注意力形态,把描述性发现提升为生命周期账户。
核心方法
直觉上,每个全注意力层都需要通过attention sink为softmax归一化约束'泄压',而巨激活正是泄压的载体:因此在每个全注意力层之前,残差流会被写入一个极大激活以备随后的全注意力消费,这就是PAS;当相邻两个全注意力层之间的线性注意力层既不削弱也不抵消该离群值时,激活保持高位、把前后尖峰连成平台,即ISP;全注意力越密,平台越连续,最终在 $\rho=1$ 时合并成全注意力LLM特有的全深度稳定形态。技术上,论文先以共识沉降(式5)确定追踪锚点 $t^\star_x$,再逐层计算该token的最大绝对激活 $m^{(\ell)}_{x,t} = \|X^{(\ell)}_{x,t,:}\|_\infty$;用'沉降-尖峰对齐率'(式6)量化PAS定位精度,用'尖峰间保留分数' $\mathrm{ISR}$(式7)量化ISP持续性;最后结合跨层有符号更新分解与固定坐标追踪,验证'写-沉-抵消'生命周期。
核心创新是'架构对齐的巨激活形态学 + 取消时机统一解释'。与已有工作把巨激活视为softmax注意力的专属副产品不同,本文证明线性注意力层并不消除巨激活,反而让全注意力层成为组织巨激活的'节拍器':无论线性骨干是固定衰减(RetNet)、输入依赖门控(HGRN/GLA)还是delta规则(DeltaNet/GDN),尖峰都精确出现在全注意力层的前一层,位置随全注意力层的插入而 predictable 地移动。机制上的本质区别在于用'离群值被反号更新抵消的时机'这一单一变量统一三种形态:快速局部抵消产生局域化的PAS,抵消被推迟则离群值持续存在形成ISP,抵消进一步推迟直至几乎不发生即恢复全注意力LLM的全深度稳定巨激活。PAS、ISP与全注意力形态由此被纳入同一个生命周期连续谱,而非三个孤立现象。
方法步骤详情
第一步(锚点识别):对输入 $x$,把所有全注意力层、所有头接收到的注意力概率按有效查询数 $|Q_t|$ 归一化后取平均,选总和最大的token为共识沉降 $t^\star_x$(式5),跨层跨头平均以抑制局部波动。第二步(跨层追踪):固定该token位置,逐层记录 $m^{(\ell)}_{x,t^\star} = \|X^{(\ell)}_{x,t^\star,:}\|_\infty$,允许最大激活的特征下标随深度演化。第三步(PAS量化):对每个全注意力层 $f$ 及其前方线性块 $B_f$,检验共识沉降是否在 $f-1$ 层取得块内最大,统计500条输入(5域×100条)的对齐率;另定义连续的峰超额 $E_{x,t,f} = \log_2\big(m^{(f-1)}_{x,t} / \max_{\ell \in B_f \setminus \{f-1\}} m^{(\ell)}_{x,t}\big)$ 度量尖峰显著度,与非sink token、首token及期望为 $1/11 \approx 9.1\%$ 的随机层基线对照。第四步(ISP量化):对相邻PAS对 $(p_i, p_{i+1})$ 之间的层集 $J_i$,计算 $\mathrm{ISR} = \min\big(1, \text{平均层间激活} / \min(m_{p_i}, m_{p_{i+1}})\big)$ 并宏平均,比值截断为1防止超参考水平的激活虚增分数。第五步(大规模验证):在Kimi Linear 48B-A3B(Base/Instruct)、Qwen3.5 35B-A3B/122B-A10B、Nemotron-H 56B、Zamba2等12个开源检查点上重复上述分析。第六步(受控预训练):用Flash Linear Attention框架从零训练24层GDN模型(340M训10B tokens、每1B记录;1.3B训50B tokens、每5B记录),把单个全注意力层插入第4/12/20层研究PAS定位与幅度,用3:1配置研究ISP;再做两组门控干预——给全注意力加输出门(GDN-GatedFA)与去掉GDN原生门(GDN-NoOutGate),其余设置全部冻结。第七步(机制验证):在1.3B GDN 12:1模型的 $f=12$ 全注意力前,固定坐标 $(t^\star, j^\star)$($j^\star = \arg\max_j |X^{(f-1)}_{t^\star,j}|$),分三阶段追踪有符号激活与逐模块更新:层11写入极值(写)、层12全注意力中 $t^\star$ 吸引不成比例注意力(沉)、后续反号更新大幅抵消(消)。
技术新颖性
技术新颖性体现在四个层面。问题层面:此前所有巨激活/系统性离群值文献(Sun et al. 2024、An et al. 2025等)均限于softmax Transformer,本文首次系统研究层交错HLA,并证明其形态定性不同——不是稳定平台而是尖峰+平台的混合组织。方法层面:发现幅度排序与注意力行为在HLA中脱钩(sink token的幅度排名跨层漂移、最大激活token相邻层切换频繁),提出以全注意力层间共识沉降为锚的追踪协议,并配套对齐率、峰超额、ISR、绝对层间激活 $A_{ISP}$(式9)一套量化指标,均可被后续研究复用。发现层面:PAS/ISP在五种线性机制、六个混合比、五个数据域、base与instruct变体、周期与非均匀注意力调度、1.2B–397B参数规模上一致复现,且受控预训练证明二者在训练早期(1B tokens内)即出现并随优化固化,排除了'精选检查点的伪影'这一替代解释。理论层面:'取消时机'账户把全注意力上的有符号离群值框架推广到混合架构,给出了纯线性(无形态)→稀疏全注意力(PAS主导)→密集全注意力(全深度平台)的统一连续谱。
实验结果
发现一:幅度排序在HLA中失效。1.3B 12:1混合模型中,sink token的幅度排名跨层漂移、最大激活token相邻层切换率远高于全注意力Transformer(Figure 2),故改用共识沉降追踪。发现二:PAS普遍存在。Table 1显示,12:1配置下10个架构-规模对的沉降-尖峰对齐率达99.4%–100.0%(如GDN 1.3B/340M均为100.0%,RetNet为99.9%/100.0%),五个数据域(WikiText、Scientific Papers、GSM8K、CodeSearchNet、FLORES-200)结果几乎一致;Table 6的对照实验显示sink比非sink的对齐率高6.5–58.6个百分点、峰超额高0.97–4.42个 $\log_2$ 单位(所有配对bootstrap 95%置信区间不含零),随机层基线仅约9.1%。发现三:ISP随全注意力密度增强。Table 2中1.3B GDN的ISR从12:1的18.4%升至6:1的26.6%再到3:1的77.8%;全部20个架构-规模相邻比比较均单调上升(12:1→6:1增5.5–44.5pp,6:1→3:1增17.0–52.3pp,95% CI均不含零),绝对层间激活在19/20个比较中同步上升;ISP出现时机因架构而异(DeltaNet在6:1即出现,GDN主要在3:1)。发现四:大规模开源模型复现。Kimi Linear 48B-A3B Base与Instruct的PAS位置和ISP边界几乎重合,Qwen3.5(原生全注意力输出门控)仍保留两种形态,Nemotron-H与Zamba2证明状态空间混合器同样成立,形态比幅度更稳定。发现五:受控预训练。PAS在1B tokens内即出现并随训练增强,位置随全注意力层插入点(4/12/20层)可预测移动,但层4插入只产生弱尖峰;ISP在340M上最终ISR为90.56%,1.3B(50B tokens)升至94.47%。发现六:门控不对称。给少量全注意力层加输出门强烈衰减PAS与ISP的绝对幅度(但不能消除其层间组织),而去掉所有GDN层的门只带来温和放大——门控效果取决于位置而非数量,说明全注意力在组织巨激活中居中心地位。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 巨激活定位(沉降-尖峰对齐,12:1混合,5域宏平均) | Sink–spike 对齐率 (%) | 10个架构-规模对全部达99.4–100.0%(Table 1) | 随机层基线(在11层前置线性块中均匀选取,期望对齐率9.1%) | 高出随机基线约90个百分点,且所有配对bootstrap 95% CI不含零 |
| PAS显著性对照(sink vs 非sink token) | 对齐率差距 (pp) / 峰超额差距 (log2) | 对齐率高6.5–58.6 pp;峰超额高0.97–4.42 log2单位(Table 6) | 非sink token(前三个共识沉降之外token的平均) | 10个架构-规模对的配对bootstrap 95%置信区间全部排除零 |
| 尖峰间巨激活保留(1.3B GDN,随混合密度变化) | ISR (%) | 12:1为18.4% → 6:1为26.6% → 3:1为77.8%(Table 2) | 最稀疏的12:1配置 | 全注意力密度从12:1提到3:1,ISR提升59.4个百分点;20个相邻比比较全部单调上升 |
| 受控预训练中ISP的跨规模复现(GDN 3:1) | 最终ISR (%) | 340M/10B tokens为90.56%,1.3B/50B tokens为94.47%(Table 10) | 同一配置的340M小规模运行 | 更大规模与更长训练使平台保留率提高3.91个百分点(作者强调不暗示可控的scaling效应) |
| 输出门控干预与召回任务(340M GDN,FA@12,10B tokens) | NIAH-2 准确率 (%) | GDN-GatedFA为16.80,GDN-NoOutGate为94.40 | 标准GDN(FA门关、GDN门开)为97.00(Table 9) | 全注意力门控使MA大幅衰减的同时NIAH-2暴跌80.2pp——提示MA与召回能力可能相关,但属相关性证据 |
局限与改进
作者承认的局限:(1)'取消时机'账户仍是相关性描述,什么因素调控抵消时机尚不清楚;(2)瞬态PAS与持续ISP是否承担不同计算角色未有答案;(3)输出门控只能强烈衰减而不能消除巨激活的层间组织;(4)受控预训练仅340M/1.3B两个规模,作者明确声明ISR从90.56%到94.47%的提升'不暗示可控的scaling效应'。我自己的观察:(5)因果链未闭合——Table 9/10中门控干预使NIAH-2从97.00跌至16.80、NIAH-3从63.60跌至5.00,但NoOutGate在PAS配置改善NIAH-1(61.80→80.40)却在ISP配置恶化它(78.60→57.20),下游指标波动大,巨激活形态与召回能力的因果关系需要专门干预实验;(6)只研究层交错(layer-interleaved)混合,未覆盖Jamba式块级混合等其他模式;(7)主分析依赖短输入(运行示例仅一句话)加500条样本,长上下文、多轮对话下的sink分布与PAS/ISP形态未系统检验;(8)共识沉降方法依赖全注意力层的存在,纯线性极限($\rho=\infty$)下注意力不可得,方法在那一端失效,纯线性模型的巨激活刻画较弱;(9)ISR是相对指标,可与绝对幅度 $A_{ISP}$ 背离(340M GDN在12:1→6:1是20个比较中唯一的例外),解读时两指标不可互换。
独立分析的弱点
弱点一:相关性有余而因果性不足。论文展示了形态指标与下游任务的并列数据,但没有任何干预实验证明PAS/ISP本身对召回有益、有害或中性。改进方向:训练时用正则化直接压制PAS幅度(固定其他条件),再测NIAH/RULER等召回任务,把相关性升级为因果证据。弱点二:机制账户偏定性。'取消时机'由固定坐标的有符号更新图(Figure 6/7/14-16)支撑,但缺少定量预测——是什么决定某一层选择立即抵消还是延迟抵消?FFN层、线性层的遗忘门、还是训练动态?改进方向:对抵消更新做逐模块归因,建立抵消时机与门控参数(如GDN的 $F_t$)的定量联系。弱点三:输入覆盖窄。核心轨迹图多基于一句话,量化分析用每域100条短样本;长文档、代码仓库级上下文中sink可能不在首token,PAS/ISP是否保持同样形态未验证。改进方向:补充4K–128K长度扫描与真实检索场景。弱点四:追踪方法在纯线性极限失效。共识沉降依赖全注意力层的注意力矩阵,$\rho=\infty$ 时无锚点可用,导致'从纯线性到全注意力'的连续谱故事在最左端缺乏直接观测。改进方向:为纯线性模型设计替代锚点(如梯度归因或状态范数)。弱点五:干预手段单一。只测试了输出门控,未对比其他已知去MA手段(如QK归一化、softmax off-by-one修复、值态裁剪),无法判断'全注意力中心性'是输出门控特有的还是架构层面的普遍规律。
未来方向
作者明确提出的方向:(1)确定调控抵消时机的因素——哪些模块、参数或训练信号决定离群值被立即抵消还是持续存在;(2)判明瞬态PAS与持续ISP是否支持不同的计算角色,例如PAS是否专为全注意力sink服务、ISP是否在线性层间传递某种全局状态。基于本文成果可以延伸的方向:(3)把PAS/ISP定位用于工程优化——混合模型的KV cache只来自全注意力层,沉降锚点+尖峰定位或可指导混合模型的cache量化/驱逐策略(作者团队在KVSink、RotateKV上的前期工作可自然衔接);(4)用ISP密度作为混合比率选择的诊断指标:既然ISR随 $\rho$ 单调上升且架构间存在差异(DeltaNet在6:1即成平台、GDN需3:1),可探索用小规模模型的形态指标预测不同混合比下的召回性能,为3:1、7:1等工业配置提供理论依据;(5)长上下文因果研究:在RULER式任务上干预PAS/ISP并测召回,回答'形态是否功能性的';(6)扩展形态学普查到其他混合范式(块级混合Jamba、MoE+线性注意力如LongCat、每k层一注意力的非均匀调度)并追踪形态随规模至百B+级;(7)结合gated attention文献设计既保留线性层效率、又在架构层面控制MA负效应(如对量化的破坏)的混合层变体。
复现评估
复现条件总体较好。开源情况:分析代码托管在 GitHub(StartLuxLabs/Massive-Activations-HLA);受控预训练的检查点发布在 HuggingFace(startlux-models/Massive-Activations-HLA);推理时分析依赖的M-A-P混合模型套件(340M/1.3B,覆盖5架构×多混合比)本身公开,预训练用Flash Linear Attention框架、语料为FineWeb-Edu(340M用20B tokens/50K batch,1.3B用100B tokens/1M batch,AdamW+cosine)。评估数据(WikiText-103、Scientific Papers、GSM8K、CodeSearchNet、FLORES-200、NIAH/SWDE/FDA等)全部公开,指标式(5)–(9)定义明确,附录给出检查点清单、层调度、超参与协议。算力门槛:推理时分析只需前向hook,340M–1.3B模型单卡即可,12个开源检查点中最大397B(Kimi K3级)需要多卡高显存环境,是一般复现者的主要障碍;受控预训练340M×10B与1.3B×50B tokens约需数十至数百GPU天,属中等规模、可负担。风险点:固定坐标分析中'代表性坐标'的选取带有一定主观性(尽管附录E.2提供了跨模型复现),以及下游小样本评估(NIAH每项单值、方差未报告)的稳定性。综合判断:推理时验证为低-中难度,完整复现受控预训练为中高难度但远低于前沿大模型训练。
论文图表
以GDN为代表作出的总览图:横轴为层、纵轴为激活幅度,展示从纯线性注意力、24:1、12:1、6:1、3:1到全注意力共六种配置下首token巨激活的跨层轨迹。所有配置中尖峰都精确出现在全注意力层(金色框)之前一层(PAS);随全注意力变密,尖峰间的谷底逐步抬升连成平台(ISP),在 $\rho=1$ 极限合并为全注意力LLM典型的全深度稳定形态。
这是全文核心发现的一图总览,PAS与ISP的定义、随混合比率的形态演化以及全注意力极限的衔接都在这张图里,读懂它就掌握了论文的主线。