★ FEATURED ARTICLE 简介这份144页PPT配套视频课程面向希望系统掌握Transformer架构的NLP学习者、算法工程师与高校师生从序列建模的经典方法讲起逐步深入到注意力机制、大语言模型与高效Transformer等前沿内容。包内共1个PDF文件约5.55MB以图文并茂的幻灯片形式呈现便于对照视频逐页学习与复习。课程由卡内基梅隆大学讲师Jean de Dieu Nyandwi主讲内容涵盖Transformer出现前的MLP、一维CNN与RNN/LSTM/GRU及其局限重点剖析注意力机制与架构核心组件并延伸至注意力可视化、LLM、不同框架下的实现、当代Transformer进展与参数化、知识蒸馏等效率优化技巧还附有趣味推文与可选练习。目前已有495人学习适合想从原理到实践全面吃透Transformer、并能将其应用于真实复杂问题的读者。1. 一份 144 页的 Transformer 讲义到底该按什么顺序啃如果你手里正躺着一份 144 页的 Transformer 讲义第一反应大概率是「先收藏回头再看」然后它就再也没被打开过。这份材料的价值不在于页数而在于它把注意力机制、多头结构、位置编码、训练技巧和推理优化串成了一条线——但前提是你得按正确的顺序拆它。我见过太多人从第一页的公式推导硬啃到第三十页就放弃了也见过有人直接跳到代码实现结果连 Q、K、V 的维度为什么那样切都没搞明白。这份讲义真正适合的是已经写过几行 PyTorch、能跑通一个分类模型但对 Transformer 的内部机制始终觉得像黑匣子的工程师。接下来的内容我会按「先立住直觉、再动手复现、最后踩坑排查」的路径把这份材料里最值得反复看的部分拆成可执行的步骤。你不需要 144 页全看完但下面这几段值得你打开编辑器跟着敲一遍。2. 从注意力公式到可运行的最小模块先让张量跑起来2.1 为什么缩放点积注意力要先除根号 d_k讲义里关于注意力的部分通常从一张图开始Query 和 Key 做点积除以根号 d_k再经过 softmax最后和 Value 加权求和。很多人背下了这个流程但说不清那个根号 d_k 到底在防什么。我一般会这样解释给团队里的新人听假设 Q 和 K 的每个分量都是独立同分布、均值 0 方差 1 的随机变量那么它们点积之后的方差就是 d_k。当 d_k 等于 64 或 512 时点积结果的数值范围会拉得很大softmax 之后就会变成接近 one-hot 的分布梯度几乎消失。除以根号 d_k 就是把方差拉回 1 附近让 softmax 的输出保持在一个有梯度的区间里。这不是玄学是数值稳定性的基本操作。下面这段代码是我平时用来验证这个结论的最小实现你可以直接复制到本地跑import torch import torch.nn.functional as F import math def scaled_dot_product_attention(Q, K, V, maskNone): Q: (batch, heads, seq_len, d_k) K: (batch, heads, seq_len, d_k) V: (batch, heads, seq_len, d_v) d_k Q.size(-1) # 核心一步点积后除以根号 d_k scores torch.matmul(Q, K.transpose(-2, -1)) / math.sqrt(d_k) if mask is not None: # mask 中为 0 的位置填一个极大的负数softmax 后趋近于 0 scores scores.masked_fill(mask 0, -1e9) attn_weights F.softmax(scores, dim-1) output torch.matmul(attn_weights, V) return output, attn_weights # 模拟一个 batch2, heads8, seq_len10, d_k64 的输入 Q torch.randn(2, 8, 10, 64) K torch.randn(2, 8, 10, 64) V torch.randn(2, 8, 10, 64) out, weights scaled_dot_product_attention(Q, K, V) print(out.shape) # torch.Size([2, 8, 10, 64]) print(weights.shape) # torch.Size([2, 8, 10, 10])这段代码里最值得注意的参数是d_k它决定了缩放系数。如果你把math.sqrt(d_k)去掉再打印一下attn_weights的最大值和最小值你会看到 softmax 后的分布变得极其尖锐大部分位置接近 0只有一个位置接近 1。这就是很多训练不稳定的源头之一。另一个参数是mask在解码器里用来遮住未来位置填的是-1e9而不是-inf因为-inf在某些后端上会导致 NaN。讲义里通常不会强调这个细节但实际写代码时这是血泪经验。2.2 多头注意力里「头」到底在切什么讲义翻到多头注意力那一节通常会画一张图Q、K、V 分别经过线性层然后切成 h 份每一份独立做注意力最后拼接再过一个线性层。很多人看到这里会问这个「切」是在哪个维度上切答案是特征维度也就是 d_model 被切成 h 个 d_k其中 d_k d_model / h。比如 d_model512h8那么每个头的 d_k 就是 64。切完之后每个头在自己的 64 维子空间里做注意力互不干扰。这样做的目的是让模型在不同的表示子空间里关注不同的模式——有的头可能关注语法结构有的头可能关注语义相似度。下面是一个多头注意力的最小实现我刻意把维度变换的过程写得很直白import torch import torch.nn as nn class MultiHeadAttention(nn.Module): def __init__(self, d_model512, num_heads8): super().__init__() assert d_model % num_heads 0, d_model 必须能被 num_heads 整除 self.d_model d_model self.num_heads num_heads self.d_k d_model // num_heads # 三个线性层分别生成 Q、K、V self.W_q nn.Linear(d_model, d_model) self.W_k nn.Linear(d_model, d_model) self.W_v nn.Linear(d_model, d_model) # 输出投影 self.W_o nn.Linear(d_model, d_model) def forward(self, x, maskNone): batch_size, seq_len, _ x.size() # 1. 线性变换后切多头: (batch, seq_len, d_model) - (batch, heads, seq_len, d_k) Q self.W_q(x).view(batch_size, seq_len, self.num_heads, self.d_k).transpose(1, 2) K self.W_k(x).view(batch_size, seq_len, self.num_heads, self.d_k).transpose(1, 2) V self.W_v(x).view(batch_size, seq_len, self.num_heads, self.d_k).transpose(1, 2) # 2. 缩放点积注意力 scores torch.matmul(Q, K.transpose(-2, -1)) / (self.d_k ** 0.5) if mask is not None: scores scores.masked_fill(mask 0, -1e9) attn torch.softmax(scores, dim-1) context torch.matmul(attn, V) # (batch, heads, seq_len, d_k) # 3. 拼接多头并投影 context context.transpose(1, 2).contiguous().view(batch_size, seq_len, self.d_model) output self.W_o(context) return output # 测试 mha MultiHeadAttention(d_model512, num_heads8) x torch.randn(2, 10, 512) out mha(x) print(out.shape) # torch.Size([2, 10, 512])这里的关键参数是num_heads和d_k。num_heads决定了子空间的个数d_k决定了每个子空间的维度。讲义里可能会提到「头数不是越多越好」原因在于当 d_model 固定时头数越多每个头的 d_k 就越小单个头的表达能力会下降。常见的配置是 d_model512 配 8 个头d_model768 配 12 个头基本保持 d_k 在 64 左右。如果你在复现时发现训练 loss 震荡得厉害可以先检查一下 d_k 是不是太小了比如 d_model256 却设了 16 个头d_k 只有 16这时候注意力的区分度会明显不够。2.3 位置编码为什么正弦波能表达相对位置讲义里位置编码那一页通常会给出一组公式用不同频率的正弦和余弦函数生成一个和词嵌入同维度的矩阵然后直接加到输入上。很多人会问为什么是正弦波而不是直接用一个可学习的 Embedding我的理解是正弦波的设计让模型能够通过线性变换表达相对位置关系。具体来说对于任意固定的偏移量 k位置 posk 的编码可以表示为位置 pos 编码的线性函数。这个性质让模型在训练时没见过那么长的序列也能在一定程度上泛化到更长的序列。下面这段代码可以帮你直观感受一下位置编码的数值分布import numpy as np import matplotlib.pyplot as plt def positional_encoding(max_len, d_model): 生成正弦位置编码 max_len: 最大序列长度 d_model: 模型维度 pe np.zeros((max_len, d_model)) position np.arange(0, max_len).reshape(-1, 1) # 分母项10000^(2i/d_model) div_term np.exp(np.arange(0, d_model, 2) * -(np.log(10000.0) / d_model)) pe[:, 0::2] np.sin(position * div_term) # 偶数维度用 sin pe[:, 1::2] np.cos(position * div_term) # 奇数维度用 cos return pe pe positional_encoding(100, 512) print(pe.shape) # (100, 512) # 打印前 5 个位置的前 8 维 print(pe[:5, :8])这段代码里div_term的计算方式决定了不同维度的波长。低维度的波长很短高维度的波长很长这样组合起来就能给每个位置一个独特的「指纹」。如果你把10000改成100波长会整体变短位置编码的区分度会下降。讲义里通常不会提这个参数可以调但在处理非常长的序列时有人会把这个底数调大让高维度的波长更长从而覆盖更长的位置范围。不过对于大多数中小规模任务默认的 10000 已经够用了。3. 把编码器和解码器拼起来训练一个玩具翻译模型3.1 编码器层的标准结构残差、LayerNorm、前馈网络讲义翻到编码器部分通常会给出一个标准层多头注意力 → 残差连接 → LayerNorm → 前馈网络 → 残差连接 → LayerNorm。这个顺序在原始论文里是 Post-LN也就是先做残差再加 LayerNorm。后来很多实现改成了 Pre-LN先做 LayerNorm 再做子层。两者的区别在于训练稳定性Pre-LN 通常更容易训练不需要 warmup 也能收敛但最终效果可能略逊于调好的 Post-LN。如果你在复现时发现模型很难训起来可以先换成 Pre-LN 试试。下面是一个编码器层的实现我用了 Pre-LN 结构class EncoderLayer(nn.Module): def __init__(self, d_model512, num_heads8, d_ff2048, dropout0.1): super().__init__() self.self_attn MultiHeadAttention(d_model, num_heads) self.feed_forward nn.Sequential( nn.Linear(d_model, d_ff), nn.ReLU(), nn.Dropout(dropout), nn.Linear(d_ff, d_model) ) self.norm1 nn.LayerNorm(d_model) self.norm2 nn.LayerNorm(d_model) self.dropout nn.Dropout(dropout) def forward(self, x, maskNone): # Pre-LN: 先归一化再进子层 attn_out self.self_attn(self.norm1(x), mask) x x self.dropout(attn_out) # 残差连接 ff_out self.feed_forward(self.norm2(x)) x x self.dropout(ff_out) # 残差连接 return x这里的关键参数是d_ff也就是前馈网络的中间维度。原始论文里设的是 d_model 的 4 倍比如 d_model512 时 d_ff2048。这个比例不是随便定的前馈网络的作用是对每个位置独立做非线性变换扩大中间维度可以增加模型的容量。如果你把 d_ff 设得太小比如和 d_model 一样模型的表达能力会明显受限训练 loss 下降得很慢。但设得太大也会导致参数量暴涨显存吃紧。我一般会从 4 倍开始试如果显存不够就降到 2 倍同时观察验证集效果有没有明显下降。3.2 解码器的掩码注意力为什么需要两种 mask解码器比编码器多了一个交叉注意力层而且自注意力层需要加掩码防止当前位置看到未来的 token。讲义里通常会把这两种 mask 画在一起一种是 padding mask用来遮住填充位置另一种是 causal mask用来遮住未来位置。在实际实现时这两种 mask 需要合并成一个否则会互相干扰。下面是一个解码器层的实现重点看 mask 的处理class DecoderLayer(nn.Module): def __init__(self, d_model512, num_heads8, d_ff2048, dropout0.1): super().__init__() self.self_attn MultiHeadAttention(d_model, num_heads) self.cross_attn MultiHeadAttention(d_model, num_heads) self.feed_forward nn.Sequential( nn.Linear(d_model, d_ff), nn.ReLU(), nn.Dropout(dropout), nn.Linear(d_ff, d_model) ) self.norm1 nn.LayerNorm(d_model) self.norm2 nn.LayerNorm(d_model) self.norm3 nn.LayerNorm(d_model) self.dropout nn.Dropout(dropout) def forward(self, x, enc_output, src_maskNone, tgt_maskNone): # 自注意力需要 causal mask attn_out self.self_attn(self.norm1(x), tgt_mask) x x self.dropout(attn_out) # 交叉注意力Q 来自解码器K、V 来自编码器 cross_out self.cross_attn(self.norm2(x), enc_output, enc_output, src_mask) x x self.dropout(cross_out) ff_out self.feed_forward(self.norm3(x)) x x self.dropout(ff_out) return x def generate_causal_mask(seq_len): 生成下三角矩阵遮住未来位置 mask torch.tril(torch.ones(seq_len, seq_len)).bool() return mask # 测试 decoder_layer DecoderLayer() x torch.randn(2, 10, 512) enc_out torch.randn(2, 15, 512) tgt_mask generate_causal_mask(10) out decoder_layer(x, enc_out, tgt_masktgt_mask) print(out.shape) # torch.Size([2, 10, 512])这里最容易翻车的地方是 mask 的维度。generate_causal_mask生成的是 (seq_len, seq_len) 的矩阵但实际传入注意力时需要广播到 (batch, heads, seq_len, seq_len)。如果你发现模型在训练时 loss 正常下降但推理时生成的序列完全乱套大概率是 causal mask 没有正确应用。检查方法是打印一下注意力权重的上三角部分如果非零说明 mask 没生效。3.3 用一个小数据集跑通训练循环讲义里通常会给出完整的模型定义但不会告诉你用什么数据跑通。我一般会用一个简单的复制任务来验证输入一个序列让模型输出同样的序列。这个任务足够简单可以在几分钟内看到 loss 下降适合用来检查模型结构有没有写错。# 构造一个简单的复制任务数据集 def generate_copy_data(batch_size, seq_len, vocab_size): 输入随机序列目标就是输入本身 src torch.randint(1, vocab_size, (batch_size, seq_len)) tgt src.clone() return src, tgt # 简化的训练循环 model nn.Transformer( d_model128, nhead8, num_encoder_layers2, num_decoder_layers2, dim_feedforward512, dropout0.1 ) optimizer torch.optim.Adam(model.parameters(), lr1e-3) criterion nn.CrossEntropyLoss(ignore_index0) vocab_size 50 for step in range(200): src, tgt generate_copy_data(32, 10, vocab_size) tgt_input tgt[:, :-1] # 解码器输入去掉最后一个 token tgt_output tgt[:, 1:] # 目标去掉第一个 token # 生成 causal mask tgt_mask generate_causal_mask(tgt_input.size(1)) output model(src, tgt_input, tgt_masktgt_mask) loss criterion(output.reshape(-1, vocab_size), tgt_output.reshape(-1)) optimizer.zero_grad() loss.backward() optimizer.step() if step % 50 0: print(fStep {step}, Loss: {loss.item():.4f})这个训练循环里ignore_index0是为了忽略 padding 位置的 loss。如果你发现 loss 一直不下降先检查一下tgt_input和tgt_output的错位是否正确——解码器输入应该是目标序列去掉最后一个 token目标应该是去掉第一个 token。这个错位是很多新手容易搞混的地方一旦错了模型永远学不会。4. 避坑与排查复现 Transformer 时最容易翻车的五个地方4.1 Loss 变成 NaN先查学习率和 mask现象训练刚开始几步loss 突然变成 NaN梯度也爆了。原因通常有两个一是学习率太大二是 mask 里用了-inf而不是-1e9。解决方法是先把学习率降到 1e-4 甚至 1e-5同时检查 mask 的填充值。如果用的是 PyTorch 的nn.Transformer它内部已经处理了 mask但如果你自己写注意力一定要确保masked_fill用的是一个大负数而不是无穷。4.2 模型输出重复 token检查位置编码和注意力掩码现象推理时生成的序列不断重复同一个 token比如「好好好好好」。原因可能是位置编码没有正确加上或者 causal mask 没有生效导致模型在解码时看到了未来位置。解决方法是打印位置编码的数值确认它被加到了输入上同时检查 causal mask 的上三角部分是否被正确遮住。4.3 训练 loss 下降但验证 loss 上升先看 dropout 和权重衰减现象训练集 loss 稳步下降验证集 loss 先降后升典型的过拟合。原因可能是模型太大、数据太少或者 dropout 设得太小。解决方法是增大 dropout比如从 0.1 调到 0.3加权重衰减比如 1e-4或者直接减小模型维度。讲义里通常不会强调这些正则化手段但在小数据集上它们比调结构更有效。4.4 显存不够从 batch size 和 d_ff 下手现象训练时显存溢出报 CUDA out of memory。原因可能是 batch size 太大、d_ff 太大或者序列太长。解决方法是先把 batch size 减半如果还不够把 d_ff 从 4 倍降到 2 倍。另外梯度累积可以在小显存上模拟大 batch代价是训练速度变慢。4.5 推理速度慢检查是否用了 KV Cache现象自回归生成时每生成一个 token 都要重新计算整个序列的注意力速度随序列长度线性增长。原因是解码器没有缓存之前计算的 K 和 V。解决方法是在推理时实现 KV Cache把已经算过的 K、V 存下来每次只计算新 token 的注意力。这个优化在讲义里通常不会提但在实际部署时是必做的。5. 从能跑通到跑得好三个值得反复调的参数5.1 学习率预热为什么前 4000 步不能直接用大学习率原始论文里用了一个 warmup 策略学习率在前 4000 步线性增加然后按步数的平方根倒数衰减。这个策略的目的是让模型在训练初期不要因为随机初始化的噪声而更新过猛。我一般会把这个策略直接抄过来尤其是用 Post-LN 结构时没有 warmup 很容易训崩。实现起来很简单class WarmupScheduler: def __init__(self, optimizer, d_model, warmup_steps4000): self.optimizer optimizer self.d_model d_model self.warmup_steps warmup_steps self.step_num 0 def step(self): self.step_num 1 lr self.d_model ** (-0.5) * min( self.step_num ** (-0.5), self.step_num * self.warmup_steps ** (-1.5) ) for param_group in self.optimizer.param_groups: param_group[lr] lr这里的warmup_steps可以根据数据集大小调整。数据量小的时候4000 步可能太多了可以降到 1000 甚至 500。判断标准是如果训练初期 loss 震荡得很厉害就增大 warmup如果 loss 下降得太慢就减小 warmup。5.2 标签平滑一个小改动让 BLEU 涨了一个点标签平滑是把硬标签从 1 变成 0.9剩下的 0.1 平均分给其他类别。这个技巧在翻译任务里几乎是无脑加的因为它能防止模型对某个 token 过度自信提升泛化能力。在 PyTorch 里实现起来很简单class LabelSmoothingLoss(nn.Module): def __init__(self, vocab_size, smoothing0.1): super().__init__() self.vocab_size vocab_size self.smoothing smoothing self.confidence 1.0 - smoothing def forward(self, pred, target): log_probs F.log_softmax(pred, dim-1) # 构造平滑后的标签分布 smooth_target torch.full_like(log_probs, self.smoothing / (self.vocab_size - 1)) smooth_target.scatter_(1, target.unsqueeze(1), self.confidence) loss (-smooth_target * log_probs).mean() return losssmoothing这个参数一般设 0.1设太大反而会欠拟合。我在一个中等规模的翻译任务上试过从 0 调到 0.1验证集 BLEU 大概涨了 0.8 到 1.2 个点代价是训练 loss 看起来没那么低但实际效果更好。5.3 梯度裁剪防止偶尔的梯度爆炸即使有 warmup训练过程中偶尔还是会出现梯度突然变大的情况。梯度裁剪就是把梯度的范数限制在一个阈值内超过就按比例缩小。这个操作几乎零成本但能避免很多训练中断。我一般设max_norm1.0在loss.backward()之后、optimizer.step()之前调用torch.nn.utils.clip_grad_norm_(model.parameters(), max_norm1.0)如果你发现裁剪前的梯度范数经常超过 10说明学习率可能偏大或者模型结构有问题。这时候不要只靠裁剪硬压回头检查一下学习率和初始化方式。这份 144 页的讲义我翻过好几遍每次都能在之前忽略的角落找到新的细节。但真正让我把 Transformer 从「能跑」变成「跑得好」的不是把每一页都看完而是把上面这几个最小模块亲手敲一遍然后在一个小任务上反复调参。我自己的习惯是每学一个新结构先写一个能在 5 分钟内跑完的玩具实验确认每一步的输入输出维度都对得上再去上大规模数据。这个习惯帮我省下了大量排查维度错误的时间。希望帮到你。本文还有配套的精品资源点击获取
阅读完成 · 觉得有帮助？
