Transformer动画讲解 - 多层感知机

技术

Transformer多层感知机

从端到端的角度来看,数据在Transformer中的流转可以概括为四个阶段:Embedding(嵌入)、Attention(注意力机制)、MLPs(多层感知机)和Unembedding(从模型表示到最终输出)。

picture.image

Embedding -> Attention -> MLPs -> Unembedding

下面对第三个阶段MLPs(多层感知机或前馈网络)进行详细介绍:

MLPs(多层感知机) 在Transformer中的位置:

  • Transformer的编码器和解码器结构:

(1)Transformer的编码器由多个相同的层堆叠而成,每个层包含两个主要的子层:

一个

多头自注意力(Multi-Head Self-Attention)机制和一个全连接的前馈神经网络(MLP)。

(2)Transformer的解码器也由多个相同的层堆叠而成,但每个层包含三个主要的子层:

一个Masked Multi-Head Self-Attention机制(用于编码器的输出),一个Multi-Head Encoder-Decoder Attention机制(用于结合编码器的输出和解码器的当前位置信息),以及一个全连接的前馈神经网络(MLP)。

picture.image

Transformer的编码器和解码器结构

  • MLP在Transformer中的位置: MLP位于Self-Attention(或Masked Self-Attention)和Multi-Head Attention层之后。

picture.image

MLP在Transformer中的位置

MLPs(多层感知机)的 模型架构:

  • MLPs模型架构

( 1)通常包含两个线性变换层,即一个输入层、一个隐藏层和一个输出层。

(2)在每个线性变换层之间,会应用一个激活函数(如ReLU),以增强模型的非线性表示能力。

(3)数学表达式为,FFN(x) = max(0, xW1 + b1)W2 + b2。

picture.image

MLP模型架构

神经网络算法 - 一文搞懂FFNN(前馈神经网络)

MLPs(多层感知机 )在Transformer中的作用

  • MLP在Transformer中的作用 :

(1)非线性变换:

MLPs通过引入激活函数(如ReLU)提供非线性变换,

这有助于模型捕获输入数据中的复杂模式。

(2)特征提取与整合:

MLPs进一步处理和转换注意力机制提取的特征,

提取和整合更多有用的信息, 使其能够学习更加复杂的函数关系。

  • MLP的两个线性变换层的作用 :

(1)第一个线性变换层:

增加了输入的维度,以便网络能够学习更复杂的表示。

(2) 第二个线性变换层

将维度还原到与输入相同的大小,通过残差连接和层归一化,将MLP的输出与原始输入相加并进行归一化,以确保稳定的训练过程。

picture.image

MLP在Transformer中的作用

Transformer无限循环操作Attention + MLP:

  • Transformer操作的本质:

海量的矩阵乘法。大模型比拼的就是算力,背后大佬就是万卡集群(GPU集群)。

picture.image

Transformer操作的本质

  • 无限循环Attention + MLP:

Attention模块和多层感知机(MLP)模块之间不断切换,直到输出完全融入到序列的最后一个向量。

picture.image

无限循环Attention + MLP

参考资料

0
0
0
0
关于作者

文章

0

获赞

0

收藏

0

相关资源
边缘云游戏行业解决方案
《“加速”游戏体验升级,火山引擎边缘云游戏行业解决方案》 许思安 | 火山引擎边缘云高级总监
相关产品
评论
未登录
看完啦,登录分享一下感受吧~
暂无评论