FFN -＞ GLU -＞ GAU

最新推荐文章于 2024-03-13 15:02:27 发布

zhurui_xiaozhuzaizai

最新推荐文章于 2024-03-13 15:02:27 发布

阅读量2.3k

点赞数

分类专栏：自然语言处理文章标签：深度学习神经网络

本文链接：https://blog.csdn.net/weixin_36378508/article/details/127362694

版权

自然语言处理专栏收录该内容

78 篇文章 25 订阅

订阅专栏

1 GLU

GLU的起源是2016年由Yann N. Dauphin在
论文:Language Modeling with Gated Convolutional Networks

在语言模型的建模方法上相比于循环神经网络更具有竞争力，提出了一种简单的线性门控单元来堆叠卷积层从而使得文本中的token可以并行化处理来获得上下文的语义特征。
而且与循环神经网络相比，其复杂度从O(N)降低到O(N/k)，其中的k为卷积核的宽度，N为文本的上下文集合。这里的循环神经网络不单指RNN，还有其变种LSTM、GRU等等。论文中整个模型的结构如下图所示：输入层(Input Sentence+Lookup Table)、中间层(Convolution+Gate)以及输出层(Softmax)

输入层(Input Sentence+Lookup Table)
W=[w1,w2,……，wn],wn代表输入token,通过lookuptable得到embedding
E=[e_w1,e_w2,……，e_wn]

中间层(Convolution+Gate)
首先输入到两个卷积层Conv1和Conv2 ，得到两个输出C_con1和C_con2，然后将C_con2利用sigmoid函数进行激活得到h(C_con2)，然后通过Hadamard积逐位相乘得到相应的隐层向量h，更一般的可表示为：

X表示输入的向量，N表示输入文本中token的集合长度，m表示向量的维度，k表示卷积核的宽度，n表示输出的特征图大小(输出维度)，V,W是可训练的参数矩阵，b,c表示对应的偏置项。
可以看出，门控线性单元与LSTM的门控本质上是一样的，只不过在计算隐层向量时不需要依赖上一个时间步。通过堆叠多层卷积就可以得到文本的上下文信息

GTU的门控机制，其实也就是将GLU的前一项利用函数进行激活，但引入一个激活函数就代表梯度在反向传播过程中就多了一项衰减项，因此，作者认为GLU优于GTU。GTU表示为：

对于采用sigmoid可能导致的梯度弥散问题，作者还在网络加入了残差。并且由于sigmoid在中间部分(近0端)表现近似于线性，所以整个模型的复杂度基本近似于线性。

输出层(Softmax)，

由于在语言模型建模过程中的词表V是相当大的，也就是在进行一次预测时会撸一遍词表，其复杂度就为O(V)，会严重影响模型的效率。所以作者采用了AdaptiveSoftmax作为归一化函数，将词表中的词分为高频词和低频词两组，将不同词频区间的词分为不同的clusters，按照词频高的cluster优先访问的原则，对cluster中的每个词进行softmax来预测，所以也要求词表需要按照频率从大到小进行排列。以此来加快模型训练时的效率。

门控线性单元（Gated Linear Unit, GLU），它是门控增强的改进版 MLP 变体。
GLU 已被证实在很多情况下都有效，并在 SOTA Transformer 中使用；
GLU Variants Improve Transformer

标准FFN

GLU

一般情况下的GLU是U不加激活函数而V加Sigmoid

GAU

论文《Transformer Quality in Linear Time》

其核心思路是将注意力和 GLU 作为一个统一层，并尽可能多地共享它们的计算，具体如下图所示。这样做不仅实现了更高的参数和计算效率，而且自然地赋能一个强大的注意力门控机制。

在这里插入图片描述

结合GLU。将Attention和GLU结合

在式(3)中，如果A等于单位阵I，那么它就是GLU式的FFN；而如果A是全1矩阵，那么它就是普通的注意力机制。所以说，(3)是Attention和FFN的一个简单而自然的融合，我们期望它能同时替换掉Attention和FFN，甚至有更好的表现。

Z是共享表示 (s<<d), 论文中s =128
当GAU只有一个头时，Wz的参数量就很少了，主要参数量在Wu,Wv,Wo上，所以GAU的参数量大约为3de；
而在标准的Transformer中，Attention的参数量为4d²，FFN的参数量为8d²（标准FFN中一般是e=4d），所以总参数量为12d²。
因此，从参数量看，当e=2d时，两层GAU大致上就等于原来的Attention+FFN。
“n层Attention+n层FFN”的标准Transformer模型，对应的就是“2n层GAU”的新模型，我们记为FLASH-Quad，其中Quad是“Quadratic”的简写，表明复杂度依然是二次的。

降低复杂度的方法
(1)将注意力计算稀疏化、即人为根据先验知识规定哪些token可以进行注意力计算(典型代表: Longformer、BigBird等)
(2)将注意力计算线性化。提出另外的方法，去逼近标准注意力的效果（典型代表: Linformer、Performer等），如下公式所示:

假设Q , K , V 的维度都为:( m , d )
右边是正常计算：m∗d∗m + m∗d∗m = 2dm²,跟序列长度m成平方正比.
左边：先计算K^TV, d∗m∗d+d∗m∗d,即:2md².

第二种方法随着序列的边长，效率会远高于第一种方法

分块注意力的计算
假设序列长度为n，每个块的维度为c，则可分成n/c个块(默认可整除)。

块内注意力：
每个块的token内部自行交互，本质上也算是“稀疏化”的一种，其复杂度大致是𝒪(n/c×c²)=𝒪(nc)，正比于n。实现时相当于头数为n/c、序列长度为c的多头注意力，可以充分地并行，
而如果想要做Decoder，那么mask掉注意力矩阵的上三角部分即可

全局注意力：
注意，这个操作跟直接用完整矩阵Q^lin,K^lin∈ℝ^n×s与V做线性Attention是完全等价的，写成这样只是更好地体现跟分块的联系。
如果是做Decoder，那么要防止泄漏未来信息，所以要改为cumsum形式：

这种情况下，为了保持并行性，我们只需要b(n/c)se的空间复杂度，而如果不分块直接用线性Attention，那么是bns²（要是原始的用法还要加上多头，那就是bhns²），在当前参数设置下有e/c≪s，所以是更省显存了。
最终GAU：基于线性版本GAU搭建的Transformer模型