Transformer 深度学习演示

从零到一理解现代大模型——覆盖分词、三大范式架构、位置编码、反向传播、微调与学习率调度的完整学习路径。
每个模块均含理论文档、可运行代码和交互式可视化网页。

11
知识模块
18
可视化网页
13
理论文档
4
可运行代码

🔤 基础:分词与理论

1
子词分词器
分词的精准程度直接影响下游模型效果。本模块用 BPE 与 WordPiece 两种算法实现从零构建分词器,并用可视化直观展示同一句子在字符级/BPE/WordPiece 三种切分方式下的差异。
BPEWordPiece Python 实现交互可视化
2
理论速查
三大范式(Encoder-Only / Decoder-Only / Encoder-Decoder)的数学差异一表对照,掩码类型、位置编码、嵌入缩放、输出层横向速查。配套各模块详读时用作索引。
Math Reference对照表

🏗️ 架构范式

3
原版 Transformer · Encoder-Decoder
掌握完整 Encoder-Decoder 架构。包含 PyTorch 从零实现、交互式架构图、所有层级的张量维度流动可视化、以及推理时的自回归解码 + KV-Cache 机制。这是理解其他范式的基础。
Encoder-DecoderMHA+FFN Notebook3 个可视化
4
GPT · Decoder-Only
自回归语言模型——现代大模型的根基。因果掩码、自回归生成、权重绑定三大核心机制。支持三种解码策略对比(贪心/温度采样/Top-K),以及 KV-Cache 加速推理。
Decoder-OnlyPython 实现 KV-Cache4 个可视化
5
BERT · Encoder-Only
双向理解——MLM 掩码语言模型 + NSP 下一句预测。与 GPT 共享同样的 Transformer Block 结构,唯独注意力是完全双向的。与 GPT 对照学习,理解一个掩码方向差异如何造就完全不同的模型行为。
Encoder-OnlyPython 实现 2 个可视化
13
ViT · Vision Transformer
Transformer 跨模态的第一步——把图像切成 patch 当"视觉词"输入,同样的 Encoder 做图像分类。CNN 的归纳偏置 vs Transformer 的全局视野:小数据 CNN 胜,大数据 ViT 反超。
Patch EmbeddingCLS Token 图像分类交互可视化
12
T5 · Text-to-Text 统一框架
把所有 NLP 任务统一为 text→text 格式。翻译、分类、摘要、问答都用同一套 Encoder-Decoder。Span Corruption 预训练 + 哨兵 token 机制。三范式对照中补齐 Encoder-Decoder 的现代实践。
Span CorruptionSentinel Tokens 交互可视化

🔬 深入:位置编码、注意力、归一化与反向传播

13
位置编码 · 四种方案横向对比
Transformer 是排列不变的——位置编码是模型理解顺序的唯一途径。四种方案并排:Sinusoidal / Learned / RoPE / ALiBi。重点剖析 RoPE 的外推能力(为什么现代 LLM 都用它)和 ALiBi 的线性偏置机制。
SinusoidalRoPEALiBi交互可视化
12
注意力变体 · MQA / GQA / Flash Attention
现代大模型的实际用件。MHA 的 KV-Cache 内存瓶颈如何被 MQA(共享一套 K/V)和 GQA(分组共享 K/V)解决;Flash Attention 如何通过分块计算和 IO 优化让注意力在长序列上不再爆炸。Llama / Mistral / Qwen 全在用。
MQAGQAFlash Attention交互可视化
13
反向传播 · 梯度流动全链路
补齐"模型如何学习"的关键一环。前向/反向双通道时间线动画,链式法则逐层展开,残差连接与 LayerNorm 的梯度稳定机制,梯度消失/爆发交互演示,参数更新下山动画。
Chain RuleGradient Flow 交互动画
11
归一化对比 · BN / LN / RMSNorm
初学者常混淆的三种归一化。BatchNorm 沿 batch 维度、LayerNorm 沿 feature 维度、RMSNorm 去掉均值的简化版。为什么 Transformer 用 LN?为什么 Llama 用 RMSNorm?一图看明白。
BatchNormLayerNormRMSNorm 交互可视化

⚙️ 实践:微调、学习率与优化器

12
微调 · 从预训练到下游任务
预训练学通用知识,微调适配具体任务。BERT 四种下游头架构(分类/NER/句对/问答)可视化,GPT 的 Prompt-based 微调方法。重点覆盖 LoRA 参数高效微调——冻结原始权重、低秩增量更新的参数量对比动画。
Fine-TuningLoRAPEFT 2 个可视化
13
学习率调度器 · 工程细节
训练优化的最后一公里。HuggingFace 原生调度器源码级模拟——Linear / Cosine / Cosine with Restarts / Polynomial / Constant 六种策略并排对比,以及不同 kwargs 参数组合的全景探索。
SchedulerWarmup 2 个可视化
14
优化器 · SGD → Adam → AdamW
梯度下降的演进路线:SGD 震荡 → Momentum 惯性平滑 → Adam 自适应学习率 → AdamW 解耦权重衰减。AdamW 是几乎所有现代 Transformer 训练的标准。配合 learning_rate 模块形成完整的训练优化闭环。
SGDMomentumAdamAdamW 交互可视化