Skip to content

常用的神经网络结构。

全连接(FC)

全连接层(FC层)MLP(Multilayer Perceptron)多层感知机的基本结构,本质上为权重矩阵与输入矩阵简单相乘并叠加非线性激活函数的运算形式。当前的神经网络机构的运算范式的基础。下面就是它的公式:其中WT代表该层的参数,X表示输入,这个输入必须是2维的(加上Batch的话是3维)。全连接层一般会组成一个前馈网络用作最后一层。

y=σ(WTX+B)

卷积

卷积可以可以看做是FC层在二维上的扩展。卷积层的输入是一个三维的数据C×H×W(不包括Batch)。相当于将全连接层权重进行了二维排布,组成一个卷积核。

且卷积神经网络拥有的权值共享的特性。也就是说对于单个通道是共用的(大白话就是使用一个卷积核卷积一整个图像)。

卷积的输出大小计算公式是:N表示输出的宽度,W表示图像的宽度,F表示卷积核的宽度,P表示填充的大小(一般为1),S表示卷积的步长

N=x(WF+2P)/S+1

RNN

普通RNN

RNN 相比与FCN的区别在于,RNN会保存当前一次的输出状态,并将该输出状态作为下一次的输入和新的数据输入一起参与新一轮的计算。它的公式为: Ht表示的是当前时刻的输出,[]表示的是拼接运算。初始时刻的Ht1可以为0。RNN的本质上与FCN是一样的,只不过多了将上一个时刻的输出作为一部分输入输入到了神经网络。

Ht=σ(WT[Xt1,Ht1]+B)

LSTM

普通RNN存在的缺点是当输入的时间序列信息特别长的时候由于不相关的信息太多会造成网络非常难以训练。LSTM ( Long Short-Term Memory)的提出就是为了解决这样的问题,LSTM通过引入更多的计算部件控制当前时刻的输入以及上一个时刻输入对于输出结果的影响。它的计算则相对更加复杂一点。计算分为三个门:

其中Wf,Wi,Wo,Wc均为可学习的参数矩阵,Bf,Bi,Bo,Bc为可学习的偏置参数。

遗忘门:

Ft=σ(WfT[Ht1,Xt]+Bf)

输入门:

It=σ(WiT[Ht1,Xt]+Bi)

输出门:

Ot=σ(WoT[Ht1,Xt]+Bo)

除了Ht需要记录,LSTM还需要维护多一个细胞状态作为下一次计算的输入:表示Element-Wise 的相乘。

C~t=tanh(WcT[Ht1,Xt]+Bc)Ct=FtC~t1+ItC~t

tanh 的函数图像为如下图,输出的范围为(1,1)

img

最终的输出为:

Ht=Ottanh(Ct)

GRU

GRU(Gated Recurrent Unit)可以视为简化版的LSTM。GRU只使用了两个门控单元:

更新门:

Zt=σ(WzT[Ht1,Xt]+Bz)

重置门:

Rt=σ(WrT[Ht1,Xt]+Br)

最终输出

H~t=tanh(WhT(Rt[Ht1,Xt])+Bh)Ht=(1Zt)Ht1+ZtH~t

Attention

Scaled Dot-Product Attention

注意力机制(Attention)相对于本质普通的神经网络还是有比较大的区别。它的计算过程设计得非常精细,对于长序列数据的处理能力相比RNN更加强大(主观判断)。 它的输入为一排 向量,输出也是一排向量,输出的向量是综合了序列中所有信息之后得到的表达。

首先需要先根据输入计算三个矩阵QKV, 分别对应Query、Key和Value,它们都是输入XRd×LX是一个由L个维度为d的向量组成的输入矩阵。计算三个矩阵的过程为:

Q=WQXK=WKXV=WVX

其中, WQ,WK,WVRd×d均为可学习的参数矩阵。

KV会组合在一起运算得到注意力分数Attention Score 矩阵ARL×L,它表示了其中第(i,j)个元素表示位置i与位置j之间的相关性得分,计算公式为:

Aij=softmax(QiTKjd)

其中 1d是为了环节是为了缓解点积得分可能较大的问题,同时可以使梯度更加稳定。A矩阵的每一列对应的就是输出的 AttentionScore ,因为过了softmax所以有

Aj=iAij=1

使用矩阵A对Value矩阵V进行加权求和j就得到了输出矩阵 O~Rd×L。这个输出再经过一个线性变换就得到了最终的输出O

O~=VAO=WOO~

Multi-head self-attention

GOOGLE关于Transformer的原始论文中使用的是多头注意力机制。多头注意力机制是单头注意力机制的改良版本,可以允许模型同时关注在不同位置的表达子空间上的不同信息。

它的公式是,h表示头的个数:

MultiHead(Q,K,V)=Concat(head1,head2,...,headh)

每个头的QKV计算和普通自注意力计算是一样,只是在还需要再经过没个头专属的线性映射矩阵后再进行每个头的注意力机制的计算:

headi=Attention(WQiQ,WKiK,WViV)

最终的输出为

O=WO×MultiHead(Q,K,V)=WO×Concat(head1,head2,...,headh)

因为 Concat得到的矩阵大小为hd×LWO的大小为d×hd。所以最终输出的O仍然是d×L

普通的注意力机制和多头注意力机制的架构区别

最新更新: