二、AI大模型（Transformer架构）

news2024/9/30 16:22:36

Transformer架构

Transformer架构是目前大多数AI大模型的基础，它通过**自注意力机制（Self-Attention Mechanism）**解决了序列处理中的长距离依赖问题。相比传统的递归神经网络（RNN）和长短期记忆网络（LSTM），Transformer能更高效地处理长序列输入，并极大提升了并行计算能力。

Transformer的核心组件

Transformer由两大部分组成：编码器（Encoder）和解码器（Decoder），但最常见的GPT、BERT等模型通常只使用编码器或解码器的单一部分来处理任务。

编码器（Encoder）：将输入序列转换为一组嵌入向量（contextual embeddings），这些嵌入向量表示输入中每个词汇的语义信息。
解码器（Decoder）：基于编码器输出的上下文信息，生成目标序列（如翻译任务中的输出句子）。

Transformer整体结构

在这里插入图片描述

典型的Transformer结构如下图所示：

+-------------------------+
|         输入序列         |
+-------------------------+
         |
         v
+-------------------------+
|   多层编码器（Encoder）   |
+-------------------------+
         |
         v
+-------------------------+
|   多层解码器（Decoder）   |
+-------------------------+
         |
         v
+-------------------------+
|       输出序列          |
+-------------------------+

在具体任务中，例如文本生成、翻译等场景，输入序列经过编码器处理为向量表示，随后解码器基于这个向量和自身的输入，生成所需的输出。

自注意力机制（Self-Attention Mechanism）

Transformer架构最核心的部分是自注意力机制，也叫Scaled Dot-Product Attention。它通过计算序列中每个词与其他所有词之间的关联（注意力权重），来捕捉句子中重要的上下文关系。

自注意力机制的流程

输入序列：假设输入一个序列 ( X = [x_1, x_2, x_3, …, x_n] )，每个 ( x_i ) 表示输入序列中的一个词。

生成三个矩阵：对于每个输入词，生成三个向量矩阵：
- Query（查询向量）：用于寻找与其他词的相关性。
- Key（键向量）：代表输入序列中的特征，用于与Query计算相关性。
- Value（值向量）：保存词的实际信息，最终用于输出。
用公式表示为：
[
Q = XW_Q, \quad K = XW_K, \quad V = XW_V
]
其中 ( W_Q, W_K, W_V ) 是训练过程中学习到的权重矩阵。
计算注意力分数：通过点积操作计算Query和Key之间的相似性（即相关性），用来衡量每个词对其他词的“注意力”。然后将结果进行缩放，防止数值过大：
[
\text{Attention}(Q, K, V) = \text{softmax}\left(\frac{QK^T}{\sqrt{d_k}}\right)V
]
其中 ( d_k ) 是Query或Key的维度，( \sqrt{d_k} ) 用来缩放点积结果，防止梯度消失或爆炸。
加权输出：计算出的注意力分数用于加权Value矩阵，这样每个词会根据与其他词的相关性重新组合自己的表征。

多头注意力机制（Multi-Head Attention）

为了捕捉更多的特征和信息，Transformer使用多头注意力机制，即将自注意力机制并行执行多次，得到多组不同的注意力输出。然后将这些结果拼接起来，通过线性变换得到最终的表示。

公式：
[
\text{MultiHead}(Q, K, V) = \text{Concat}(\text{head}_1, \text{head}_2, …, \text{head}_h)W^O
]
其中 ( \text{head}_i = \text{Attention}(QW_i^Q, KW_i^K, VW_i^V) )，( W_i^Q, W_i^K, W_i^V ) 是不同的权重矩阵，( W^O ) 是输出的线性变换矩阵。

多头注意力机制能够让模型在不同的空间维度上关注不同的语义关系，提升了模型的表达能力。

Transformer中的位置编码（Positional Encoding）

Transformer虽然能并行处理序列中的词汇，但与RNN等序列模型不同，Transformer不具备自然的序列顺序感知能力。因此，模型需要加入额外的位置编码来为序列中的每个词引入位置信息。

位置编码通常是通过正弦和余弦函数生成的，具有周期性，能够为不同长度的输入序列提供位置信息。这帮助模型理解序列中的词汇顺序。

位置编码公式：
[
PE(pos, 2i) = \sin\left(\frac{pos}{10000^{\frac{2i}{d_{model}}}}\right)
]
[
PE(pos, 2i+1) = \cos\left(\frac{pos}{10000^{\frac{2i}{d_{model}}}}\right)
]
其中 ( pos ) 是词的位置，( i ) 是维度索引，( d_{model} ) 是嵌入向量的维度。