arXiv:2602.11534cs.LGcs.AI2026-02被引 1

用局部互动替代全局竞争,让注意力更稳定高效

Krause Synchronization Transformers

  • 基于距离的局部稀疏交互,避免全局竞争
  • 序列长度线性复杂度,效率提升显著
  • 多场景验证有效,适合大模型与长序列

Transformer 的自注意力依赖全局归一化 softmax 权重,导致所有标记在每层都相互竞争,深层叠加后引发强同步动态,易导致表示坍缩和注意力黑洞。本文提出 Krause Attention,受有限信心共识动态启发,将基于相似性的全局聚合替换为基于距离的局部、选择性稀疏交互,促进结构化局部同步而非全局混合。该机制与将 Transformer 视为粒子系统的理论相呼应,证明有限信心交互可自然缓解注意力集中问题。限制交互范围使计算复杂度从二次降至线性。实证在视觉(ViT on CIFAR/ImageNet)、自回归图像生成(MNIST/CIFAR-10)、大语言模型(Llama/Qwen)及从头训练的语言模型(100M/200M)中均取得一致性能提升,且效率更高,表明有限信心动力学是注意力的有效可扩展归纳偏置。

原文摘要 · Abstract (English)

Self-attention in Transformers relies on globally normalized softmax weights, causing all tokens to compete for influence at every layer. When composed across depth, this interaction pattern induces strong synchronization dynamics that favor convergence toward a dominant mode, a behavior associated with representation collapse and attention sink phenomena. We introduce Krause Attention, a principled attention mechanism inspired by bounded-confidence consensus dynamics. Krause Attention replaces similarity-based global aggregation with distance-based, localized, and selectively sparse interactions, promoting structured local synchronization instead of global mixing. We relate this behavior to recent theory modeling Transformer dynamics as interacting particle systems, and show how bounded-confidence interactions naturally moderate attention concentration and alleviate attention sinks. Restricting interactions to local neighborhoods also reduces runtime complexity from quadratic to linear in sequence length. Empirically, we validate Krause Attention across diverse settings, including vision (ViT on CIFAR/ImageNet), autoregressive image generation (MNIST/CIFAR-10), large language models (Llama/Qwen), and language models trained from scratch at multiple scales (100M/200M). Across these domains, Krause Attention achieves consistent performance gains while improving computational efficiency, highlighting bounded-confidence dynamics as a scalable and effective inductive bias for attention.

注意力机制Transformer局部交互高效计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。