跳转至

Transformer

Transformer 是以注意力和前馈网络为主要计算单元的序列模型架构。

基本块

一个典型块包含:

  • 多头注意力;
  • 逐位置前馈网络;
  • 残差连接;
  • LayerNorm 或 RMSNorm 等归一化。

三种常见结构

  • Encoder-only:双向读取上下文,常用于理解、分类和表示学习。
  • Decoder-only:使用因果掩码预测下一个 token,是生成式 LLM 的常见结构。
  • Encoder–Decoder:Encoder 编码输入,Decoder 通过 Cross-Attention 读取编码结果,适合条件生成。

Pre-Norm 与 Post-Norm

差别在归一化相对残差分支的位置。Pre-Norm 通常更容易训练很深的网络;Post-Norm 的优化行为不同, 往往需要更谨慎的初始化、学习率和 warmup。选择应视具体架构和训练配方,而不是把二者简单排序。

核心限制

标准全注意力对长序列成本高;自回归解码又具有逐 token 串行性。因此训练和推理系统发展出了 FlashAttention、稀疏注意力、KV Cache、PagedAttention 和投机解码等优化。