AB
AiBoss
Wiki

什么是并行注意力与前馈(Parallel Attention and Feed-Forward)?一种把前馈网络与注意力块并列排布的 Transformer 层设计

并行注意力与前馈(Parallel Attention and Feed-Forward,PAF)是一种 Transformer 层内排布方式:把前馈网络块与注意力块放在同一层中并列计算,而不是像串行注意力与前馈(Series Attention and Feed-Forward,SAF)那样依次串联。arXiv 论文 2305.13297 用这一设计作为研究工具,检验前馈网络在 Transformer 中的作用。

并行注意力与前馈(Parallel Attention and Feed-Forward,PAF)是一种 Transformer 层内部的排布设计:在同一层里,前馈网络(Feed-Forward Network,FFN)块与注意力(attention)块以并列的方式组织,而不是按传统做法前后串联。它要解决的问题是:在标准 Transformer 中,前馈网络与自注意力总是被绑在一条串行通路上,研究者很难单独观察前馈网络究竟承担了什么职责;把两者改为并列之后,就可以在受控条件下检验关于前馈网络功能的假设。

需要说明的是,PAF 在 arXiv 论文《Investigating the Role of Feed-Forward Networks in Transformers Using Parallel Attention and Feed-Forward Net Design》中主要作为一种研究手段出现,作者是 Shashank Sonkar 与 Richard G. Baraniuk。该论文用它来对比串行注意力与前馈(Series Attention and Feed-Forward,SAF)结构,从而讨论前馈网络在 Transformer 中的作用,而不是把它作为一项通用的性能改进方案来推荐。

为什么重要

在常见的 Transformer 结构里,每一层通常先做自注意力,再做前馈网络,两者沿同一条残差通路依次叠加,这种排布在论文中被称为串行注意力与前馈(SAF)。这种串联方式在工程上很自然,但给分析带来了困难:前馈网络和注意力块的输出被混在同一条通路上,观察某一层的中间状态时,很难判断某个现象是由注意力造成的还是由前馈网络造成的。

前馈网络在参数中占比很大,但它的具体功能长期缺少清晰的解释。如果只是继续在 SAF 结构上做消融,改动前馈网络往往会同时改变注意力块的输入分布,因果关系不容易分离。该论文提出的思路是换一种层内排布:让前馈网络块与注意力块并列,使两者的输入来自同一处,从而让「前馈网络做了什么」这个问题变得可以单独考察。论文明确指出,其目的是加深对 Transformer 架构中前馈网络与自注意力机制各自角色及相互作用的理解。

工作机制

按照该论文的表述,PAF 设计的有效性建立在两条关于层内前馈块与注意力块的主要假设之上:

  1. 前馈块的主要功能是维持各 token 嵌入的各向同性(isotropy),防止它们退化。也就是说,前馈网络的作用被理解为一种对表示分布的调节,使不同 token 的嵌入不至于坍缩到少数方向上。
  2. 注意力块中计算出的残差范数显著小于输入 token 嵌入的范数。这意味着在并列结构下,注意力分支对主干表示的改动相对温和,主干表示主要由输入嵌入本身决定。

这两条假设共同构成了 PAF 能够成立的前提:如果前馈网络主要负责维持表示的分布形态,而注意力分支的残差又相对较小,那么把两者并列、让它们共享同一份输入,就不会破坏层内信息的流动,反而能把前馈网络的作用从注意力通路中「拆」出来单独观察。

从对比的角度看,SAF 与 PAF 的差别在于层内数据流的组织方式:SAF 中后一个块读取的是前一个块修改过的表示,误差与效应会沿通路累积;PAF 中两个块读取的是同一份输入表示,各自的贡献更容易被分别度量。论文正是利用这一点,把 PAF 当作实验装置,去检验上述两条假设是否在实际训练中成立。

典型例子

该论文给出的实证做法是:训练两个大型语言模型的 PAF 变体,分别是 RoBERTa-large 与 bert-large-uncased,并将它们与对应的 SAF 结构进行比较。论文报告的结果是,在 PAF 设计中上述两条假设都得到了验证。也就是说,在这些模型上,前馈块维持 token 嵌入各向同性、以及注意力块残差范数远小于输入嵌入范数这两点,都与论文的预期一致。

这个例子的意义在于它展示了 PAF 的典型用法:不是把它当作一个新的骨干网络去刷榜,而是把它当作一个可对照的层结构变体,用来回答「前馈网络在 Transformer 里到底做什么」这类机制性问题。论文的贡献也定位在理解层面,即加深对前馈网络与自注意力机制角色与相互作用的认识。

边界与常见误解

第一,容易把 PAF 误解为一种公认的、普遍更优的架构改进。就素材所及,该论文把 PAF 作为研究前馈网络角色的设计来使用,其结论是关于两条假设在 PAF 变体中成立,而不是宣称 PAF 在各类任务上优于 SAF。把它当成通用最佳实践,超出了论文的主张范围。

第二,容易把论文中的两条假设当成已被广泛证实的行业共识。这两条假设是该论文提出的、用于解释 PAF 为何有效的出发点,其验证范围限于论文所训练的两个模型变体(RoBERTa-large 与 bert-large-uncased)。其他规模、其他结构、其他训练目标的模型是否同样满足,素材中没有给出结论。

第三,PAF 的适用前提本身构成边界。它的分析价值依赖于「注意力残差范数相对输入嵌入范数较小」这一条件;若在某些设置下该条件不成立,并列排布与串行排布之间的可比性就会下降。论文并未讨论这种情形下的表现。

第四,已知代价方面,素材没有给出 PAF 在计算量、显存占用、训练稳定性或下游指标上的具体数字,因此不能据此推断它的效率或效果优劣。可以确定的只是:采用 PAF 意味着偏离主流 Transformer 的层内串联惯例,需要重新训练模型变体才能开展比较,这也是该类机制研究通常的成本。

参考资料