// 算法学习路线
从零理解一个大模型
从分词到 Transformer、现代架构、预训练和推理,按 nanoGPT 的顺序搞懂一个大模型的每个零件。
难度
4 步19 个算法5 个配套课程单元
01步骤
每一步都以上一步为基础。点卡片看论文和资料。
文本如何变成数字
Transformer 核心
Transformer 与自注意力Transformer & Self-Attention 完全基于注意力、可并行训练的序列模型,是所有大模型的骨架。 2017核心 GPT-3 与上下文学习GPT-3 & In-Context Learning 足够大的语言模型只看几个例子就能完成新任务。 2020核心 BERT 双向 Transformer 编码器,通过“完形填空”预训练。 2018核心 旋转位置编码 RoPERotary Position Embedding (RoPE) 通过旋转 Q/K 向量来编码位置,使注意力天然感知相对距离。 2021核心 RMSNorm 只用均方根缩放、省掉减均值的简化版 LayerNorm。 2019核心 SwiGLUSwiGLU (Gated Linear Units) 给前馈层加一条“门控”支路,提升 Transformer 质量。 2020核心现代效率组件
分组查询注意力 GQAGrouped-Query Attention (GQA) 把注意力头分组,每组共享一份 K/V,在 MHA 与 MQA 间折中。 2023核心 多头潜在注意力 MLAMulti-head Latent Attention (MLA) 把 K/V 压缩成一个低维潜在向量缓存,再按需还原。 2024核心 FlashAttentionFlashAttention (v1–v3) 按 GPU 内存层级分块计算的精确注意力,快且省显存。 2022核心 KV Cache 把已生成 token 的 Key/Value 存起来,避免每步重复计算。 核心 温度与 Top-p 采样Temperature & Nucleus (Top-p) Sampling 控制生成随机性:只从累计概率达到 p 的候选里抽样。 2019核心规模化
缩放定律Scaling Laws 模型损失随参数、数据、算力按幂律平滑下降。 2020核心 Chinchilla 计算最优Chinchilla Compute-Optimal Scaling 同样算力下,参数量和训练 token 应同比增长(约 20 token/参数)。 2022核心 稀疏门控混合专家Sparsely-Gated Mixture of Experts 一个门控网络为每个输入只挑少数几个“专家”子网络。 2017核心 DeepSeekMoE 细粒度专家DeepSeekMoE (Fine-Grained & Shared Experts) 把专家切得更细并加入共享专家,让每个专家更专精。 2024核心 混合精度与 FP8 训练Mixed Precision & FP8 Training 用 16 位甚至 8 位浮点计算,用 32 位保存主权重。 2017核心 ZeRO / FSDP 数据并行ZeRO / FSDP (Sharded Data Parallelism) 把优化器状态、梯度和参数切片到各卡,消除数据并行的冗余。 2019核心
02配套课程单元
和这些算法配套的学习目标、动手项目和自测题。
U40 分词与嵌入 BPE 分词、token 计费、嵌入向量与语义相似度。 6 小时6 个算法2 篇论文已有正文 U41 Transformer 与注意力 自注意力、多头、位置编码、残差与归一化,从零实现 GPT。 16 小时6 个算法3 篇论文已有正文 U42 现代 LLM 架构:位置编码、GQA/MLA 与长上下文 RoPE、SwiGLU、RMSNorm、GQA/MLA、FlashAttention、滑动窗口与上下文扩展。 12 小时12 个算法5 篇论文 U44 预训练、数据与缩放定律 预训练数据流水线、缩放定律、Chinchilla、μP、MTP 与训练稳定性。 10 小时6 个算法4 篇论文 U60 解码、采样与 KV Cache 预填充与解码、温度/top-p、KV Cache、内存带宽瓶颈。 6 小时4 个算法2 篇论文已有正文