Transformer 技术笔记:以 Gemma 2 27B 为例

less than 1 minute read

Published:

这是一份以 Gemma 2 27B 为解剖对象的 Transformer 技术笔记。

阅读全文 →

内容包括:

  • 模型骨架:d_model 与 head_dim、GQA、RoPE 与长度外推、ALiBi / NoPE、MLA、FFN 与 MoE、RMSNorm 与 Pre-LN
  • 训练:反向传播、优化器、训练显存账
  • 损失与信息论:交叉熵、信息论视角、模型能存多少知识
  • 推理工程:KV cache、带宽瓶颈、投机解码、FlashAttention、量化(GPTQ / AWQ / SmoothQuant)
  • 对齐:SFT、RLHF / PPO、DPO、GRPO 的训练流水线,LoRA / QLoRA
  • 评测统计:一个分数有多可信、多次采样的两种方差、怎样比较两个模型

本文由我与 Claude(Anthropic)多轮对话整理而成,具体数字和论文结论都对照原始论文、官方配置或数值实验核实过,不确定的地方标了”待核实”,发现的错误记在附录 A。欢迎指错。

Leave a Comment