arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.01203cs.CLcs.LG
收藏

注意力汇聚在注意力层中锻造原生MoE:针对头部坍塌的汇聚感知训练

Attention Sink Forges Native MoE in Attention Layers: Sink-Aware Training to Address Head Collapse

Institute for Artificial Intelligence, Peking University, Beijing(人工智能研究院,北京大学,北京);School of Integrated Circuits, Peking University, Beijing(集成电路学院,北京大学,北京)

Zizhuo Fu, Wenxuan Zeng, Runsheng Wang, Meng Li

AI总结:

本文通过理论和实证证明注意力汇聚自然构建了注意力层内的混合专家机制,并提出汇聚感知训练算法以缓解头部坍塌问题,提升模型性能。

摘要:

大型语言模型(LLMs)通常将不成比例的注意力分配给第一个标记,这种现象称为注意力汇聚。最近的几种方法旨在解决这个问题,包括GPT-OSS中的汇聚注意力和Qwen3-Next中的门控注意力。然而,缺乏对这些注意力机制之间关系的全面分析。在这项工作中,我们提供了理论和实证证据,表明普通注意力和汇聚注意力中的汇聚自然地在注意力层内构建了混合专家(MoE)机制。这一见解解释了先前工作中观察到的头部坍塌现象,即只有固定子集的注意力头对生成有贡献。为了缓解头部坍塌,我们提出了一种汇聚感知训练算法,该算法带有专为注意力层设计的辅助负载平衡损失。大量实验表明,我们的方法在普通注意力、汇聚注意力和门控注意力上实现了有效的头部负载平衡,并提高了模型性能。我们希望这项研究能为注意力机制提供新的视角,并鼓励进一步探索注意力层内固有的MoE结构。

英文摘要:

Large Language Models (LLMs) often assign disproportionate attention to the first token, a phenomenon known as the attention sink. Several recent approaches aim to address this issue, including Sink Attention in GPT-OSS and Gated Attention in Qwen3-Next. However, a comprehensive analysis of the relationship among these attention mechanisms is lacking. In this work, we provide both theoretical and empirical evidence demonstrating that the sink in Vanilla Attention and Sink Attention naturally construct a Mixture-of-Experts (MoE) mechanism within attention layers. This insight explains the head collapse phenomenon observed in prior work, where only a fixed subset of attention heads contributes to generation. To mitigate head collapse, we propose a sink-aware training algorithm with an auxiliary load balancing loss designed for attention layers. Extensive experiments show that our method achieves effective head load balancing and improves model performance across Vanilla Attention, Sink Attention, and Gated Attention. We hope this study offers a new perspective on attention mechanisms and encourages further exploration of the inherent MoE structure within attention layers.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑