Transformer 深度学习演示

从零到一理解现代大模型——覆盖分词、三大范式架构、位置编码、反向传播、微调、归一化、优化器与学习率调度的完整学习路径。
每个模块均含理论文档、可运行代码和交互式可视化网页。新增现代模型架构横向对比(GPT / Llama / Qwen / DeepSeek-V3 / GLM-5)、Mixtral MoE 架构交互与 Muon 优化器实战。

15
知识模块
28
可视化网页
20
理论文档
7
可运行代码

🔤 基础:分词与理论

1
子词分词器
分词的精准程度直接影响下游模型效果。本模块用 BPE 与 WordPiece 两种算法实现从零构建分词器,并用可视化直观展示同一句子在字符级/BPE/WordPiece 三种切分方式下的差异。
BPEWordPiece Python 实现交互可视化
2
理论速查
三大范式(Encoder-Only / Decoder-Only / Encoder-Decoder)的数学差异一表对照,掩码类型、位置编码、嵌入缩放、输出层横向速查。新增现代 Decoder-Only 模型架构细分(GPT vs Llama vs Qwen vs DeepSeek-V3 vs GLM-5),包括 MLA/DSA/GQA 注意力机制分化与 MoE 对比。
Math Reference对照表

🏗️ 三大范式

3
原版 Transformer · Encoder-Decoder
掌握完整 Encoder-Decoder 架构。包含 PyTorch 从零实现、交互式架构图、所有层级的张量维度流动可视化、以及推理时的自回归解码 + KV-Cache 机制。这是理解其他范式的基础。
Encoder-DecoderMHA+FFN Notebook 5 个可视化
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 个可视化

🔬 核心机制

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

🚀 架构扩展

6
ViT · Vision Transformer
Transformer 跨模态的第一步——把图像切成 patch 当"视觉词"输入,同样的 Encoder 做图像分类。CNN 的归纳偏置 vs Transformer 的全局视野:小数据 CNN 胜,大数据 ViT 反超。
Patch EmbeddingCLS Token 图像分类Encoder-Decoder3 个可视化
7
T5 · Text-to-Text 统一框架
把所有 NLP 任务统一为 text→text 格式。翻译、分类、摘要、问答都用同一套 Encoder-Decoder。Span Corruption 预训练 + 哨兵 token 机制。三范式对照中补齐 Encoder-Decoder 的现代实践。
Span CorruptionSentinel Tokens 交互可视化
14
LLM 架构 · 混合专家(MoE)
大模型架构专题(当前聚焦稀疏混合专家 MoE,后续纳入稠密模型)。稀疏激活的核心理念:将 FFN 替换为多个并行「专家」子网络,由路由器动态选择 Top-K 个参与计算。以 Mixtral 8×7B 为例深度解析:Router Gate 门控 → TopK 稀疏选择 → SwiGLU 专家内部结构 → 加权求和 → 负载均衡。总参数 46.7B 但每 token 仅激活 ~12.9B。
MoEMixtralRouterSwiGLU 稀疏激活
15
Memory Decoder · 参数化长期记忆模块
把小模型的「长期记忆」从推理主干中拆出来,做成独立可预训练、可独立扩容的参数化记忆模块。以 Pythia-410M + 6.9B 记忆为例:冻结 Backbone 推理,记忆模块离线用分布式 Faiss kNN 检索出的稀疏目标分布做 KL 蒸馏训练,推理时按任务权重 α 融合两者输出,17 基准平均分反超 12B 底座且总参数量少 39%。
Memory DecoderkNNKL 蒸馏 参数化记忆交互可视化

⚙️ 训练实践

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 → Muon
梯度下降的演进路线:SGD 震荡 → Momentum 惯性平滑 → Adam 自适应学习率 → AdamW 解耦权重衰减 → Muon 矩阵正交化更新。AdamW 是几乎所有现代 Transformer 训练的标准,Muon 通过 Newton-Schulz 迭代实现更优的矩阵参数更新。配合 learning_rate 模块形成完整的训练优化闭环。
SGDMomentumAdamAdamWMuon 交互可视化
输入关键词开始搜索