环境

conda create -n torch python=3.12
conda activate torch

# Install PyTorch (CPU version)
pip install torch torchvision
# Install PyTorch with CUDA (version <= nvidia-smi shown)
#  https://pytorch.org/get-started/locally
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu130

运行

  • 关键:Decoder-only, RoPE, GQA, MoE, SwiGLU
cd start-deep-learning/practice/Transformer
pip install -r requirements.txt
python build_transformer.py

BATCH_SIZE 4, 可以改 32,如果显存够大。我呢,训不太动 😢

进化线

之后是概括,给一些关键词。基此问一下 AI,能得到更好的回答呢。

例如,DeepSeek 归纳了 5 条核心进化线,

如果把大模型比作一辆车:

  • Attention 是 引擎(GQA/MLA 是省油技术);
  • Normalization 是 底盘悬挂(RMSNorm 保证行驶稳定);
  • 激活函数 是 燃油标号(SwiGLU 是高标号汽油);
  • 位置编码 是 导航系统(RoPE 让你知道该去哪);
  • 架构 是 车身设计(Decode-Only 最终统一了赛道)。

这些进化线相互配合,才有了今天又长、又快、又强的 LLM。

架构进化

Encoder-Decoder: 原始,适合 Seq2Seq 任务

Decoder-only (Causal): 当前,经过 Scaling Laws 验证

Attention 进化

KV Cache 优化,

注意力机制 KV Cache 量级 表达能力 核心思想
MHA (多头注意力) 最大 (H) 最强 每个头独立看世界
MQA (多查询注意力) 最小 (1) 较弱 所有人用同一套记忆
GQA (分组查询注意力) 中等 (G) 可控 分组共享,折中之道
MLA (多头潜在注意力) 极小 (≈1) 压缩记忆,解耦计算
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐