Transformer 架构详解

大模型时代的基石架构，所有主流 LLM 的核心。

一句话总结

Transformer 是当今所有大模型（GPT、Claude、Gemini等）的核心架构，相当于大模型的”骨架”。

历史背景

提出时间: 2017 年
提出者: Google（论文《Attention Is All You Need》）
解决问题: RNN 处理序列数据的效率和长距离依赖问题

它解决了什么问题？

RNN 的困境

RNN 的处理方式：
"我 → 今天 → 去 → 北京 → 吃 → 了 → 烤鸭"
  ↓      ↓      ↓     ↓     ↓     ↓     ↓
 一个一个处理，像排队过安检，又慢又容易"忘事"

问题：

无法并行，训练慢
长序列容易”遗忘”前面的内容
梯度消失/爆炸问题

Transformer 的解决方案

Transformer 的处理方式：
"我 今天 去 北京 吃 了 烤鸭"
  ↓↓↓↓↓↓↓ 同时处理 ↓↓↓↓↓↓↓
 所有词一起看，像开会讨论，又快又能看到全局

核心机制：自注意力（Self-Attention）

直观理解

句子：“小明把苹果给了小红，因为她饿了”

问题：“她”指的是谁？

注意力机制的工作方式：

"她" 会去"询问"每个词：你和我有多相关？

  小明  →  相关度：0.1  （男的，不太可能）
  苹果  →  相关度：0.05 （物品，排除）
  小红  →  相关度：0.8  （女的，接收苹果的人，很可能！）
  饿了  →  相关度：0.3  （描述状态）

结论："她" = 小红 ✓

数学表示

注意力计算公式：

Attention(Q, K, V) = softmax(Q × K^T / √d_k) × V

其中：
- Q (Query): 查询向量，"我想找什么？"
- K (Key): 键向量，"我有什么特征？"
- V (Value): 值向量，"我的实际内容"
- √d_k: 缩放因子，防止点积过大

代码示例

import torch
import torch.nn.functional as F
 
def self_attention(Q, K, V):
    """
    自注意力计算
    Q, K, V: shape = (batch_size, seq_len, d_model)
    """
    d_k = Q.size(-1)
 
    # 计算注意力分数
    scores = torch.matmul(Q, K.transpose(-2, -1)) / torch.sqrt(torch.tensor(d_k))
 
    # softmax 归一化
    attention_weights = F.softmax(scores, dim=-1)
 
    # 加权求和
    output = torch.matmul(attention_weights, V)
 
    return output, attention_weights

Transformer 的整体结构

输入: "今天天气真好"
        ↓
   ┌─────────────┐
   │  词嵌入层    │  把文字变成数字向量
   │  + 位置编码  │  加入位置信息
   └─────────────┘
        ↓
   ┌─────────────┐
   │  多头注意力   │  ← 核心！让词与词互相"交流"
   │  + 残差连接  │     多个角度同时分析
   │  + LayerNorm │
   └─────────────┘
        ↓
   ┌─────────────┐
   │  前馈神经网络 │  进一步加工信息
   │  + 残差连接  │
   │  + LayerNorm │
   └─────────────┘
        ↓
   （重复 N 层，GPT-4 大约有 100+ 层）
        ↓
输出: 预测下一个词 / 理解语义

关键组件

1. 词嵌入（Word Embedding）

将离散的文字转换为连续的向量表示。

2. 位置编码（Positional Encoding）

因为 Transformer 并行处理，需要额外注入位置信息。

# 正弦余弦位置编码
PE(pos, 2i)   = sin(pos / 10000^(2i/d_model))
PE(pos, 2i+1) = cos(pos / 10000^(2i/d_model))

💡 现代大模型基本不再用上面的正弦余弦位置编码，主流方案是 RoPE（旋转位置编码）。 RoPE 还可以通过 YaRN 等技术外推到比训练时更长的上下文（如 32K → 128K）。详见 Dense 与 MoE 架构对比 § YaRN。

3. 残差连接（Residual Connection）

output = LayerNorm(x + Sublayer(x))

缓解深层网络的梯度问题。

4. 前馈网络（FFN）

FFN(x) = ReLU(x × W1 + b1) × W2 + b2

两层全连接网络，中间维度通常是隐藏层的 4 倍。

大模型家族的分化

Transformer (2017)
     │
     ├── Encoder-only → BERT（理解型）
     │                   - 双向注意力
     │                   - 适合分类、问答、NER
     │
     ├── Decoder-only → GPT、Claude（生成型）
     │                   - 单向注意力（causal mask）
     │                   - 适合文本生成、对话
     │
     └── Encoder-Decoder → T5、BART
                          - 适合翻译、摘要

为什么 Transformer 如此成功？

特点	说明
并行计算	所有位置同时处理，GPU 利用率高
长距离依赖	任意两个位置可以直接交互
可扩展性	模型越大、数据越多，效果越好（Scaling Law）
通用性	文本、图像、音频、视频都能用

类比总结

把大模型比作一个超级学霸：

Transformer = 学霸的大脑结构
注意力机制 = 学霸的”抓重点”能力
参数量 = 脑容量大小
训练数据 = 读过的所有书

参考资料

原始论文: Attention Is All You Need
Jay Alammar 的可视化讲解: The Illustrated Transformer

创建时间: 2024-12-10

与 AI 交流 AI

探索

Transformer