Transformer多层感知机

从端到端的角度来看，数据在Transformer中的流转可以概括为四个阶段：Embedding（嵌入）、Attention（注意力机制）、MLPs（多层感知机）和Unembedding（从模型表示到最终输出）。
图片
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）。

图片
Transformer的编码器和解码器结构
  • MLP在Transformer中的位置：MLP位于Self-Attention（或Masked Self-Attention）和Multi-Head Attention层之后。

图片

MLP在Transformer中的位置

MLPs（多层感知机）的模型架构：

  • MLPs模型架构：

    （1）通常包含两个线性变换层，即一个输入层、一个隐藏层和一个输出层。

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

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

MLP模型架构

神经网络算法 - 一文搞懂FFNN（前馈神经网络）

MLPs（多层感知机）在Transformer中的作用：

  • MLP在Transformer中的作用：

    （1）非线性变换：MLPs通过引入激活函数（如ReLU）提供非线性变换，这有助于模型捕获输入数据中的复杂模式。

    （2）特征提取与整合：MLPs进一步处理和转换注意力机制提取的特征，提取和整合更多有用的信息，使其能够学习更加复杂的函数关系。
  • MLP的两个线性变换层的作用：
    （1）第一个线性变换层：增加了输入的维度，以便网络能够学习更复杂的表示。
    （2）第二个线性变换层：将维度还原到与输入相同的大小，通过残差连接和层归一化，将MLP的输出与原始输入相加并进行归一化，以确保稳定的训练过程。
图片

MLP在Transformer中的作用

Transformer无限循环操作Attention + MLP：
  • Transformer操作的本质：海量的矩阵乘法。大模型比拼的就是算力，背后大佬就是万卡集群（GPU集群）。
图片
Transformer操作的本质
  • 无限循环Attention + MLP：Attention模块和多层感知机（MLP）模块之间不断切换，直到输出完全融入到序列的最后一个向量。 

图片

    无限循环Attention + MLP


参考资料

  • https://www.3blue1brown.com/lessons/gpt