arXiv:2605.18855cs.LGcs.CV2026-05被引 4

用层间变化量替代累积状态,提升注意力残差的路由选择能力

Delta Attention Residuals

论文配图:Delta Attention Residuals
图 1 · 摘自论文原文
  • 以各子层的输出增量为注意力输入,避免冗余信息干扰
  • 最大注意力权重达0.6,显著高于标准方法的0.2,路由更精准
  • 适用于220M到7.6B参数模型,可直接微调预训练模型使用

注意力残差通过学习软最大值注意力实现跨层选择性路由,但传统方法对前层累积隐藏状态进行注意力计算,存在严重冗余。我们发现这导致深层路由坍缩:注意力权重对比度低且趋近均匀(最大权重≈0.2),限制了对关键信息的选择能力。由此提出关键问题:注意力残差应路由哪些层内表示?本文提出Delta注意力残差,不再关注累积状态,而是关注每子层引入的增量(Δ = h_{i+1} - h_i)。增量表示结构差异大,生成更高对比度的注意力分布(最大权重≈0.6),实现更有效的跨层选择。该机制在子层与模块粒度均有效。在所有测试规模(220M–7.6B)下,相比标准残差和注意力残差,验证困惑度降低1.7%–8.2%。此外,可通过标准微调将预训练检查点转换为Delta注意力残差。代码已开源。

原文摘要 · Abstract (English)

Attention Residuals replace standard additive residual connections with learned softmax attention over previous layer outputs, enabling selective cross-layer routing. However, standard Attention Residuals still attend over cumulative hidden states in previous layers, which are highly redundant. We show that this redundancy leads to routing collapse in deeper layers: attention weights become low-contrast and closer to uniform (max weight ${\approx}$0.2), limiting the model's ability to select informative states in previous layers. This raises a key but underexplored design question: what layer-wise representations should be routed in Attention Residuals? To answer this question, we propose Delta Attention Residuals, which attend over deltas -- the change introduced by each sublayer ($\mathbf{v}_i = \mathbf{h}_{i+1} - \mathbf{h}_i$) -- instead of cumulative states. Delta representations are structurally diverse and yield higher-contrast attention distributions (max weight ${\approx}$0.6), enabling more selective and effective routing across layers. This principle applies at both per-sublayer and block granularity. Across all tested scales (220M--7.6B), Delta Attention Residuals consistently outperform both standard residuals and Attention Residuals, with 1.7--8.2\% validation perplexity gains. Delta Attention Residuals also enables converting pretrained checkpoints into Delta Attention Residuals via standard fine-tuning. Code is available at https://github.com/wdlctc/delta-attention-residuals-code.

注意力机制残差连接模型优化深度学习

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