跳转至

CS336 2026 Lecture 11:Scaling Case Study and Details

LaTeX 源码 · 观看视频

字段 内容
作者/整理 基于 Stanford CS336 Spring 2026 官方幻灯片重新整理
来源 Stanford CS336
日期 2026 年春季

CS336 2026 Lecture 11:Scaling Case Study and Details

本讲主线:scaling laws 在真实模型里的落地

上一讲建立了 data/model/compute scaling 的基本形式,本讲转向更困难的落地问题:真实模型并不只改变参数量,learning rate、batch、schedule、optimizer、normalization 与架构细节也会变化。核心任务是区分哪些规律可跨 scale 迁移,哪些结论只是某个 recipe 和实验区间的局部拟合。

讲义补充:源 Slide 1 如何接续 Lecture 9

Lecture 9 讲 scaling laws 的基本形式和 Kaplan/Chinchilla/IsoFLOPS 等方法;Lecture 11 则进入“真实公开模型如何做 scaling”。重点不再是公式本身,而是公开论文里的 recipes:如何调 batch/LR,如何用 WSD 降低 sweep 成本,muP 是否真的稳定,optimizer scaling 如何验证。

\singleslide{slides-images/slide-002.jpg}{本讲动机:Chinchilla 是否可落地、如何降低 scaling-fit 成本、哪些 parametrization 更稳定。}{2} 这页把抽象 scaling law 转成 recipe 审计:不仅预测 loss,还要预测 LR、batch、schedule 与 architecture 在目标规模是否仍有效。

讲义补充:源 Slide 2 的三个现实问题

第一,Chinchilla 的 compute-optimal 方法在真实模型论文中是否还能工作。第二,完整从头训练网格非常贵,是否能用 WSD、早停或复用曲线节省计算。第三,架构、optimizer、learning rate、batch size 这些 hyperparameters 是否随 scale 改变,能否通过小模型可靠外推。

\singleslide{slides-images/slide-003.jpg}{Scaling in practice:从 2022 经典结果转向 2024--2026 公开模型的真实 recipes。}{3} Slide 3 强调研究对象已经变化:近期论文披露 WSD、muP、optimizer 与 MoE scaling,提供比单一 Chinchilla exponent 更丰富的证据。

讲义补充:源 Slide 3 的历史意义

Lecture 9 主要围绕 Kaplan、Chinchilla 等早期 scaling 研究;这页提醒我们,2024-2026 的公开模型开始公布更多 scaling 细节。Scaling law 已经从研究论文变成训练报告里的工程流程:MiniCPM、DeepSeek、Qwen、Kimi、Hunyuan、LLaMA 3、MiniMax 都在展示不同风格的 scaling recipe。

Slide 4:Initialization、optimizers、learning rate、batch 等 hyperparameters 都可能 scale-sensitive。
Slide 4:Initialization、optimizers、learning rate、batch 等 hyperparameters 都可能 scale-sensitive。 查看原图

术语消化:scale-sensitive hyperparameters

对象 为什么 scale-sensitive 课程中的检验方式
Initialization 宽度、深度改变会改变 activation 和 update magnitude。 muP/SP 对比、activation/update 条件。
Learning rate 最优 LR 可能随参数量、batch、数据量变化。 小模型 grid search 或 muP 外推。
Batch size critical batch 随 loss 和数据规模变化。 batch-loss 曲线、StepFun/DeepSeek/Qwen fits。
Optimizer 不同 optimizer 需要不同超参和 scaling rule。 Muon、AdamC、exotic optimizers 的 scale study。

\singleslide{slides-images/slide-005.jpg}{公开案例路线:先读 MiniCPM,再用 DeepSeek 检查规律能否跨团队复现。}{5} 选择这两个案例是为了比较相似问题的不同实验设计,而不是做模型排名。

讲义补充:源 Slide 5 的案例选择

MiniCPM 是小模型高性能路线,公开了细致的 scaling computation 和 muP 使用;DeepSeek 是更大规模高性能开源模型,展示了不使用 muP 时如何直接估计 batch/LR 和做 IsoFLOP 分析。二者形成对照:一个强调稳定参数化,一个强调直接经验拟合。

MiniCPM:小模型高性能与 muP 稳定 scaling

为了观察 scaling law 如何真正指导模型训练,本部分先用 MiniCPM 作为完整案例。它把 muP 初始化、learning-rate transfer、optimal batch、WSD schedule 与 Chinchilla-style model/data tradeoff 放进同一条实验链,优点是每个假设都有小规模 sweep 和最终模型结果对应。

\singleslide{slides-images/slide-006.jpg}{MiniCPM:用密集 scaling experiments 与 muP 构造高性能小模型。}{6} 它不是当年最大模型,却公开了 LR、batch、WSD 与 data/model trade-off,适合作为方法案例。

讲义补充:源 Slide 6 为什么选 MiniCPM

MiniCPM 不是因为最大或最 SOTA 而重要,而是因为它公开展示了小模型如何通过 careful scaling、muP、batch/LR/token-to-size ratio tuning 得到强性能。它给课程提供了“scaling recipes 如何写进真实模型训练”的案例。

\singleslide{slides-images/slide-007.jpg}{MiniCPM 的性能定位:1--2.5B 模型可匹配部分 7B,说明 recipe 会移动参数效率曲线。}{7} Slide 7 的结论不是“小模型永远更好”,而是参数量之外的数据、schedule 与 scaling calibration 会显著改变同规模表现。

讲义补充:源 Slide 7 不能只看模型大小

这页展示小模型通过训练策略和数据/超参优化达到强性能。它提醒我们:参数量不是性能唯一解释变量。若 scaling recipe 更好、数据更有效、训练 token 更多,小模型可能超过粗糙训练的大模型。

Slide 8:MiniCPM 技术 1,使用 muP 稳定 scaling,并给出 scale_emb、scale_depth、init_std、lr 等值。
Slide 8:MiniCPM 技术 1,使用 muP 稳定 scaling,并给出 scale_emb、scale_depth、init_std、lr 等值。 查看原图

读图:Slide 8 中 muP 的作用

muP 的目标是让不同宽度模型的训练动力学更可比,使小模型上找到的 learning rate 和初始化规则更稳定迁移到大模型。这里的具体数值不是通用处方,而是 MiniCPM architecture 和训练 recipe 下的标定结果。

\singleslide{slides-images/slide-009.jpg}{MiniCPM strategy:muP 初始化、固定 aspect ratio,并拟合 batch/LR/token-to-size 后扩大模型。}{9} 最大 sweep 模型与最终模型仍相差约 5 倍,因此成功依赖外推;muP 的作用是减少 width 改变时重新调参。

讲义补充:源 Slide 9 的 strategy 为什么省事

固定 aspect ratio 意味着不在每个规模都重新搜索 depth/width/head 等比例;muP 则希望 LR 和初始化跨宽度稳定。这样 scaling sweep 的变量更少,外推更容易。但图中也提醒:最大小模型和最终模型之间仍有约 5 倍 gap,外推并非无风险。

MiniCPM 的 LR、batch 和 WSD

前面的 muP 只承诺宽度变化时训练动力学更稳定,本节检查这是否真的让最优 learning rate 更可迁移,并进一步拟合 batch 随目标 loss 或数据量的变化。WSD 把 warmup、stable 与 decay 分开,使一次长训练可以在多个中间点比较 model/data 配比,降低从头训练大量候选的成本。

Slide 10:Optimal LR:根据 muP,最优 learning rate 应大致稳定;MiniCPM 检验这一点。
Slide 10:Optimal LR:根据 muP,最优 learning rate 应大致稳定;MiniCPM 检验这一点。 查看原图

读图:Slide 10 应看“最优点是否横向稳定”

若不同模型规模的 loss-vs-LR 曲线最小点接近同一个 LR,说明 muP 达到了 scale-invariant tuning 的目标。若最优点随规模漂移,说明小模型 LR 外推到大模型仍不可靠。

Slide 11:Optimal batch:三个模型大小在数据量、batch 和 loss 上形成曲面,红线标出近似最小 loss。
Slide 11:Optimal batch:三个模型大小在数据量、batch 和 loss 上形成曲面,红线标出近似最小 loss。 查看原图

读图:Slide 11 的三维 batch 曲面

横轴是 batch,纵轴或多条曲线代表数据/训练进度,颜色表示 loss。垂直列是一条固定 batch 的训练曲线。红线追踪不同数据量下的最优 batch。读这页时要看最优 batch 如何随 loss 下降而增大,而不是只找某个全局最小点。

Slide 12:Optimal batch size:沿 Kaplan 2020 分析,最优 batch 随最终 loss 降低按多项式增大。
Slide 12:Optimal batch size:沿 Kaplan 2020 分析,最优 batch 随最终 loss 降低按多项式增大。 查看原图

读图:Slide 12 与 critical batch 的关系

最终 loss 越低,训练越接近后期,critical batch 通常越大。图中的多项式趋势说明 batch size 也可被 scaling law 拟合。它不是说 batch 越大越好,而是说目标 loss 变化时,最优 batch 也应随之 schedule。

\singleslide{slides-images/slide-013.jpg}{剩余问题:模型大小与数据量 trade-off;完整 Chinchilla 网格需要从头训练,成本可从 \(n\) 增至 \(n^2\)。}{13} 早停曲线不能直接代表不同 token budgets 的最终 loss;WSD 尝试复用训练轨迹以减少独立 runs。

讲义补充:源 Slide 13 的成本爆炸

若每个模型大小都要从头训练多个数据量点,网格成本会快速变成 \(n_{\text{models}}\times n_{\text{data}}\)。这正是 scaling in practice 的核心痛点:拟合 scaling law 本身也要花 compute,必须寻找复用曲线或降低 sweep 成本的方法。

Slide 14:MiniCPM 的部分解法是 WSD learning rate:warmup、stable、decay 三阶段。
Slide 14:MiniCPM 的部分解法是 WSD learning rate:warmup、stable、decay 三阶段。 查看原图

术语消化:什么是 WSD learning rate

WSD 是 warmup-stable-decay。先 warmup 到学习率平台,在 stable phase 长时间训练;需要 Chinchilla-style 不同 token budget 时,可以从 stable 末尾 restart decay,而不是每个数据量都从头训练。它用训练曲线复用降低 scaling sweep 成本。

Slide 15:WSD 在 MiniCPM 中表现良好,stable 阶段慢,decay 阶段 loss 快速下降,decay 约 10%。
Slide 15:WSD 在 MiniCPM 中表现良好,stable 阶段慢,decay 阶段 loss 快速下降,decay 约 10%。 查看原图

读图:Slide 15 的曲线说明什么

WSD 的好处不是让 stable 阶段最快,而是让同一条训练轨迹在多个候选 token budget 上可复用。Decayed tail 给出类似“如果在这里结束训练”的 loss 点,从而支持 lower-envelope 或 joint fit。

MiniCPM 的 Chinchilla 分析

\singleslide{slides-images/slide-016.jpg}{MiniCPM 的 Chinchilla-type analysis:借助 WSD 使用 lower envelope 与 joint fit。}{16} 两种方法分别从观测最优点和全网格函数出发;一致性比单个拟合常数更重要。

讲义补充:源 Slide 16 的方法选择

Method 1 是 lower envelope:看所有训练曲线在不同 compute 下的最小点。Method 3 是 joint fit:拟合参数量和数据量到 loss 的曲面。MiniCPM 同时使用两者,是为了在降低成本的同时增强外推可信度。

\singleslide{slides-images/slide-017.jpg}{Chinchilla Method 1:不同模型 training curves 的 lower envelope 与数据边际收益。}{17} 颜色代表模型规模;envelope 是否直线决定 power-law 外推是否可信,图中轻微曲率应保留为不确定性。

讲义补充:源 Slide 17 的颜色和趋势

不同颜色代表不同模型规模。若 lower envelope 呈直线或平滑曲线,就可以外推 compute-optimal 关系。图中趋势清晰但不完美,说明公开 scaling recipe 在真实数据中会有噪声,不能只看拟合线的漂亮程度。

Slide 18:Chinchilla method 3,MiniCPM 的主要 scaling approach 是 joint fit,并得到很高 data-model ratios。
Slide 18:Chinchilla method 3,MiniCPM 的主要 scaling approach 是 joint fit,并得到很高 data-model ratios。 查看原图

读图:Slide 18 的高 data-model ratio 意味着什么

如果 joint fit 推荐很高 token-to-parameter ratio,训练策略会偏向“小一些模型 + 多训练数据”。这与 Chinchilla 以来的过训趋势一致,也解释为什么小模型通过更多 token 和精细 tuning 能表现很强。

DeepSeek 与近期 scaling recipes

MiniCPM 展示了依赖 muP 的路线,接下来 DeepSeek 提供另一种工程选择:不要求所有超参理论上 scale-invariant,而是直接通过小规模 runs 估计 optimal batch、learning rate 与 IsoFLOP model/data 配比。比较两者能看出,scaling recipe 的关键是可验证预测,而不是统一采用某个参数化流派。

\singleslide{slides-images/slide-019.jpg}{DeepSeek:另一套公开 scaling analysis,用 7B/67B 高性能模型检验 MiniCPM 规律。}{19} 跨团队复现能区分普遍规律与单一实现细节,尤其是 LR、batch 与 data/model allocation。

讲义补充:源 Slide 19 的 DeepSeek 对照

DeepSeek 与 MiniCPM 的区别在于它不依赖 muP,而是直接估计 batch/LR scaling,并使用 WSD-style learning rate 和 IsoFLOP 分析。它代表另一条路线:用足够多小规模实验证明超参趋势,而不是依赖参数化理论稳定性。

Slide 20:DeepSeek scaling strategy:不用 muP,直接估计 optimal batch 和 LR。
Slide 20:DeepSeek scaling strategy:不用 muP,直接估计 optimal batch 和 LR。 查看原图

读图:Slide 20 的风险

不用 muP 意味着最优 LR 可能随 scale 漂移,因此必须更认真地做 LR/batch sweep。直接经验拟合很实用,但外推风险也更依赖实验覆盖和函数形式选择。

Slide 21:DeepSeek 的 learning rate scaling,用小规模 runs 收集 near-optimal 模型;LR fit 看起来有些可疑。
Slide 21:DeepSeek 的 learning rate scaling,用小规模 runs 收集 near-optimal 模型;LR fit 看起来有些可疑。 查看原图

读图:Slide 21 的“questionable fit”怎么看

Near-optimal 定义为距离最小 loss 0.25% 以内,这能缓解噪声,但 LR 的最优区间可能很平。若点分布宽、拟合斜率不稳,就不能过度相信外推公式。公开论文里的 scaling fit 也需要质疑。

Slide 22:DeepSeek 使用 WSD-style learning rate,fast warmup 加两次 10% decay,性能接近 cosine。
Slide 22:DeepSeek 使用 WSD-style learning rate,fast warmup 加两次 10% decay,性能接近 cosine。 查看原图

读图:Slide 22 的 WSD 共性

MiniCPM 和 DeepSeek 都使用 WSD 类思想,说明 WSD 的价值不仅是曲线形状,而是可复用训练轨迹、支持 scaling 分析。它让同一段 stable training 可以接多个 decay endings,减少重复从头训练。

Slide 23:DeepSeek 的 data-size tradeoff 使用 Chinchilla method 2,即 IsoFLOP 风格分析。
Slide 23:DeepSeek 的 data-size tradeoff 使用 Chinchilla method 2,即 IsoFLOP 风格分析。 查看原图

读图:Slide 23 的 IsoFLOP 逻辑

固定 FLOP budget,扫 model size 和 data size,取 loss 最小点。这样得到每个 compute 下的 optimal model size。它比直接 joint fit 更直观,也更容易发现某些点是否异常。

Slide 24:DeepSeek 的 fitted scaling models 通常能准确预测 final model losses。
Slide 24:DeepSeek 的 fitted scaling models 通常能准确预测 final model losses。 查看原图

读图:Slide 24 是 scaling recipe 的验收

Scaling law 最终要看外推预测是否命中大模型 final loss。若小规模 fitted curve 能预测 7B/67B 等最终 loss,说明超参、数据、训练 recipe 在外推区间内足够稳定。反之,漂亮的小规模曲线也可能没有工程价值。

更多公开模型 scaling 片段

为了判断 DeepSeek 式 recipe 是否只是个例,本节横向查看 Qwen、Kimi、Hunyuan、LLaMA 与 MiniMax 等公开材料。不同团队拟合的对象包括 dense/MoE 参数量、sparsity、batch、learning rate 与 downstream metric;阅读重点是实验自变量、active parameter 定义和预测是否经过最终规模验证。

\singleslide{slides-images/slide-025.jpg}{Qwen 2.5/3 的 scaling evidence:batch 与 hyperparameter fits 仍是近期 recipe 的常见组成。}{25} 这页说明 scaling practice 没有在 Chinchilla 后停止,而是扩展到优化超参与不同模型族。

讲义补充:源 Slide 25 的 Qwen 证据

Qwen 的公开信息说明 batch 和 LR scaling 已经成为主流模型训练报告的重要组成。不同团队可能用不同函数形式,但共同目标都是减少大模型调参风险。

\singleslide{slides-images/slide-026.jpg}{Kimi K2:用 sparsity scaling law 选择 MoE 的稀疏程度。}{26} MoE 不只决定总参数,还要选择 activated experts/parameters;scaling law 可比较质量收益与路由、通信和存储成本。

讲义补充:源 Slide 26 把 scaling 扩展到 sparsity

Scaling 不只决定模型大小和 token 数,也可以决定 sparse/MoE 模型里 active parameters、expert 数、稀疏比例。Kimi K2 这类模型说明“参数价值”在稀疏模型里更复杂,必须把 active compute 和 total parameters 分开。

\singleslide{slides-images/slide-027.jpg}{Hunyuan MoE IsoFLOPS:拟合 data 与 active-parameter 的最优约 96:1 比例。}{27} 对 MoE 应使用每 token active parameters 解释 compute,同时另记 total parameters 的 memory/communication 成本。

讲义补充:源 Slide 27 的 active param ratio

MoE 的 total parameters 远大于每 token active parameters,因此 compute-optimal ratio 应以 active param 或 activated FLOPs 理解。96:1 这样的比例提醒我们,Chinchilla dense 模型口径不能直接套到 MoE。

Slide 28:LLaMA 3 scaling laws,包括 IsoFLOPS-style scaling、39:1 ratio 和 compute-to-downstream scaling。
Slide 28:LLaMA 3 scaling laws,包括 IsoFLOPS-style scaling、39:1 ratio 和 compute-to-downstream scaling。 查看原图

读图:Slide 28 的两个层次

预训练 loss 的 IsoFLOPS 只是第一层;compute-to-downstream scaling 关注下游任务如何随 compute 改善。大模型训练真正关心的是产品能力和 benchmark,而不仅是 cross entropy。因此 downstream scaling 是更难也更有价值的外推。

\singleslide{slides-images/slide-029.jpg}{MiniMax-01:把 architecture choices 也纳入 scaling law,并结合 Chinchilla Method 1。}{29} 当 attention、depth、width 或 context 机制改变时,不能假设旧 exponent 不变;需要模型族级对照实验。

讲义补充:源 Slide 29 的 architecture scaling

Architecture scaling laws 用来比较不同结构随规模变化的 loss 或能力曲线。MiniMax-01 的例子说明,架构选择也可以进入 scaling law pipeline,而不只是凭最大模型结果判断。

Slide 30:近期 scaling recipe 总结,以 DeepSeek 为代表:假设大部分 Transformer hypers scale-invariant,重点拟合 batch/LR 和 model sizing。
Slide 30:近期 scaling recipe 总结,以 DeepSeek 为代表:假设大部分 Transformer hypers scale-invariant,重点拟合 batch/LR 和 model sizing。 查看原图

读图:Slide 30 的 recipe 可执行版本

实践中常先固定大多数 architecture hypers,只做 batch/LR scaling,再用 IsoFLOP 选择模型大小和 token 数。这样把指数级超参搜索压缩成少数关键轴。代价是:如果固定的 architecture hypers 实际随 scale 变化,recipe 会漏掉更优方案。

本章小结

MiniCPM 和 DeepSeek 展示了两种 scaling 实践:一类用 muP 稳定迁移,另一类用直接 empirical sweep 拟合 batch/LR 和 IsoFLOP。近期模型则说明 scaling law 的对象正在扩展到 MoE sparsity、architecture、downstream 能力和 active parameters。

Optimizer scaling:StepFun、Muon 与 scale dependence

\singleslide{slides-images/slide-031.jpg}{Optimizer scaling 路线:optimizer choice 与 tuning 都可能随规模变化。}{31} 比较 optimizer 不能只复用同一 LR/batch;每个方法需要公平的规模相关调参和等 compute 对照。

讲义补充:源 Slide 31 的问题

Optimizer 不是“选 AdamW 就结束”。不同 optimizer 的最优 LR、batch、weight decay、momentum 可能随 scale 变化;如果只在小模型上比较,可能因为没调好超参而误判算法。

Slide 32:StepFun scaling law study,核心问题是随着 scale 如何设置 LR/batch。
Slide 32:StepFun scaling law study,核心问题是随着 scale 如何设置 LR/batch。 查看原图

读图:Slide 32 的 StepFun 角色

StepFun 类研究把 optimizer hyperparameter scaling 当成主要对象,而不只是模型大小。它类似 DeepSeek/Qwen 路线:训练大量小模型,在 LR/batch 空间中找规律。

\singleslide{slides-images/slide-033.jpg}{LR 与 batch scaling 的变量选择:critical-batch(loss)、compute power law,或其他函数形式。}{33} 自变量决定外推语义;用 loss 表示训练状态通常比只用 step 更可迁移,但仍要验证不同模型族。

讲义补充:源 Slide 33 的变量选择

Critical batch 把 batch 写成 loss 的函数;DeepSeek 风格可能把超参写成 compute 的幂函数;也可能与数据量、模型大小分别相关。函数形式不是数学细节,而是决定外推是否可信的建模选择。

Slide 34:StepFun approach 是纯经验 grid search,训练模型来描绘 hyperparameter space。
Slide 34:StepFun approach 是纯经验 grid search,训练模型来描绘 hyperparameter space。 查看原图

读图:Slide 34 的 grid search 代价

纯经验方法的优势是不依赖强理论假设;劣势是成本高,且容易受数据范围影响。它要求每个模型规模和数据量上都有足够覆盖,否则拟合出的 scaling rule 只是局部规律。

Slide 35:Observation 1:pre-training loss over batch/LR 通常呈凸形,minimizer 可较清楚识别。
Slide 35:Observation 1:pre-training loss over batch/LR 通常呈凸形,minimizer 可较清楚识别。 查看原图

读图:Slide 35 为什么重要

若 loss surface 对 LR/batch 近似凸,hyperparameter tuning 就可被系统化:每个 scale 找最小点,再拟合最小点随 scale 的轨迹。若曲面多峰或噪声大,外推就危险得多。

Slide 36:Observation 2:batch 主要依赖 dataset size;固定模型时更大数据可能需要更高 LR,但这对 WSD 可能脆弱。
Slide 36:Observation 2:batch 主要依赖 dataset size;固定模型时更大数据可能需要更高 LR,但这对 WSD 可能脆弱。 查看原图

读图:Slide 36 的 fragile LR 结论

如果换成 WSD schedule、不同 warmup、不同 decay 或不同 optimizer,LR scaling 结论可能改变。Batch scaling 可能更稳定,LR scaling 更依赖训练 recipe。因此公开 LR 幂律要带着 recipe 一起读。

Slide 37:Observation 3:结果可能泛化到 MoE 和其他 datasets,但仍需谨慎。
Slide 37:Observation 3:结果可能泛化到 MoE 和其他 datasets,但仍需谨慎。 查看原图

读图:Slide 37 的 robustness 问题

Scaling rule 是否跨模型族、数据集、dense/MoE 架构泛化,是 optimizer scaling 的核心。若泛化好,小模型 tuning 价值很大;若不泛化,每个新数据集/架构都要重新 sweep。

本节核心观点

评估 optimizer 时,必须同时控制 compute、Chinchilla ratio、batch/LR scaling 和模型规模。否则一个 optimizer 看似更好,可能只是因为它在某个 scale 上超参调得更合适。

Slide 38:Optimizer and scale,optimization 是 LLM 的核心,但因 scale dependence 很棘手。
Slide 38:Optimizer and scale,optimization 是 LLM 的核心,但因 scale dependence 很棘手。 查看原图

\singleslide{slides-images/slide-039.jpg}{Optimizer comparison 陷阱:不同 optimizer 有不同最优超参和 scaling,默认配置会制造假赢家。}{39} 公平比较应为每个规模分别调 LR、batch、momentum/weight decay,并报告 tune budget。

讲义补充:源 Slide 39 的 optimizer 比较陷阱

若 AdamW 调得很好、Muon 或 SGD 没调好,比较没有意义。每个 optimizer 都需要自己的 learning-rate/batch/weight-decay scaling rule。公平比较不是同一套超参,而是各自最优超参。

Slide 40:Problem 2:显著 scale dependence;算法开发必须检查 compute 和 Chinchilla ratios。
Slide 40:Problem 2:显著 scale dependence;算法开发必须检查 compute 和 Chinchilla ratios。 查看原图

读图:Slide 40 对算法论文的提醒

一个 optimizer 在小模型或非 Chinchilla ratio 下胜出,不代表在大模型 compute-optimal setting 中胜出。算法开发需要 scaling with compute,而不是只报告一个固定规模实验。

\singleslide{slides-images/slide-041.jpg}{看似平滑的 hyperparameter scaling 仍可能在更大规模 blow up。}{41} 外推曲线需要在未参与拟合的中间规模做验证,并监控稳定性而非只看最终 loss;parametrization 或 optimizer 细节可让漂亮规律失效。

讲义补充:源 Slide 41 的 blow-up 教训

Scaling curves 在小范围内平滑,不代表在更大 compute 上稳定。某些参数化、batch-size LR scaling 或 optimizer tweak 可能在大 scale 才暴露不稳定。外推前必须留出中间 scale 验证点。

Muon

前面的 StepFun 结果说明 optimizer 比较必须随 scale 调参,本节用 Muon 检查一个更激进的矩阵更新方法。核心问题不是 Newton--Schulz 形式本身是否漂亮,而是它在相同 compute、数据配比和各自最优超参下,能否稳定跨越 NanoGPT、小规模 study 与大型 MoE 训练。

Slide 42:Muon 是面向 matrix-valued parameters 的 optimizer,用 Newton-Schulz 近似正交化更新矩阵。
Slide 42:Muon 是面向 matrix-valued parameters 的 optimizer,用 Newton-Schulz 近似正交化更新矩阵。 查看原图

读公式:Slide 42 的 Muon 直觉

若更新矩阵 \(B_t=USV^\top\),Muon 近似把它转成 \(UV^\top\),也就是去掉奇异值尺度,只保留方向上的正交化结构。它试图让矩阵参数的更新更均衡,但实际收益必须看 scaling 和超参调优。

Slide 43:Muon and scaling:NanoGPT speedrun、小规模 scaling study、Kimi K2 都显示 Muon 可在 scale 上工作。
Slide 43:Muon and scaling:NanoGPT speedrun、小规模 scaling study、Kimi K2 都显示 Muon 可在 scale 上工作。 查看原图

读图:Slide 43 的保守结论

Muon “works at scale” 不等于它在所有 scale、所有模型、所有 compute ratios 下都优于 AdamW。Scaling gains 很难测,因为要公平调参、控制 compute、控制数据和架构。课程更看重方法论:optimizer 改动必须做 scaling 验证。

本章小结

Optimizer scaling 的难点在于:不同 optimizer 有不同最优超参,而且这些超参随 scale、数据、compute 和训练 schedule 变化。任何 optimizer 结论都必须绑定 scaling study,而不是孤立单点。

muP in depth:为什么它想让超参跨宽度稳定

前面的案例多次使用 muP,但经验结果只有在机制清楚时才可迁移。本部分回到 maximum update parametrization 的两个量级条件:初始化 activation 保持 \(\Theta(1)\),一次更新引起的 activation change 也保持 \(\Theta(1)\);再检查现代 Transformer 组件如何破坏这些假设。

Slide 44:Maximum update parametrization 深入:scale-invariant hyperparameter tuning 很诱人,但它如何工作、是否实践有效?
Slide 44:Maximum update parametrization 深入:scale-invariant hyperparameter tuning 很诱人,但它如何工作、是否实践有效? 查看原图

术语消化:muP 与 SP

muP 是 maximum update parametrization,希望在宽度变化时保持 activation 和 update 的量级稳定,从而让小模型最优 learning rate 更可迁移。SP 是 standard parametrization,常见但宽度放大时训练动力学可能变化。Lecture 11 后半段问:现代 LM 里 muP 的理论条件哪些仍成立,哪些会被 RMSNorm、SwiGLU、exotic optimizers、weight decay 破坏。

Slide 45:CerebrasGPT 从 0.1B 到 13B 用 Chinchilla recipe,核心发现是 muP 让 scaling 更稳定。
Slide 45:CerebrasGPT 从 0.1B 到 13B 用 Chinchilla recipe,核心发现是 muP 让 scaling 更稳定。 查看原图

读图:Slide 45 的证据类型

CerebrasGPT 是 muP 实践证据之一:跨多个模型大小训练,比较稳定性和超参迁移。它说明 muP 不是纯理论玩具,但也不是所有现代架构自动适用的万能钥匙。

Slide 46:什么是 muP:要求初始化时 activations 保持 \(Θ(1)\),一次梯度更新后 changes 也保持 \(Θ(1)\)。
Slide 46:什么是 muP:要求初始化时 activations 保持 \(Θ(1)\),一次梯度更新后 changes 也保持 \(Θ(1)\)。 查看原图

读图:Slide 46 的两个条件

A1:随着层宽 \(n_l\) 改变,初始化时的 activations 不应爆炸或消失,应为 \(\Theta(1)\)。A2:一次梯度更新造成的 activation change 也应为 \(\Theta(1)\)。muP 的学习率和初始化规则就是围绕这两个量级条件设计的。

Deriving muP:condition A1

本节先处理初始化尺度。为了让宽度增加时每层 activation 不爆炸也不消失,需要根据输入输出维度缩放权重方差;推导的价值不是记住某个常数,而是学会从 producer/consumer 维度判断矩阵谱范数和输出范数如何随 width 改变。

Slide 47:推导 muP 条件 A1,用深线性网络 \(h_l=W_lh_l-1\) 和矩阵 concentration 控制 activation。
Slide 47:推导 muP 条件 A1,用深线性网络 \(h_l=W_lh_l-1\) 和矩阵 concentration 控制 activation。 查看原图

读公式:Slide 47 的初始化尺度

\(W_l\sim \mathcal{N}(0,\sigma^2 I)\),矩阵谱范数会随输入/输出维度变化。要让 \(\|h_l\|\) 不随宽度爆炸,就要让 \(\sigma\)\(n_{l-1},n_l\) 缩放。标准 Xavier/He 初始化和 muP 都在解决“宽度变大时 activation 尺度如何保持”的问题。

Deriving muP:condition A2

Condition A1 只保证初始网络稳定,接下来还要保证训练更新不会随宽度变得过大或消失。本节把线性层梯度写成上游 loss gradient 与前层 activation 的 outer product,再选择 layer-specific learning-rate scaling,使 \(\Delta W_l h_{l-1}\) 保持同一量级。

\singleslide{slides-images/slide-048.jpg}{muP condition A2:约束权重更新诱导的 hidden-state change 在宽度扩展时保持 \(\Theta(\sqrt n)\)。}{48} 推导把 SGD rank-one update、operator norm 与 activation norm 连接起来,用来决定 layer-wise learning-rate scaling。

讲义补充:源 Slide 48 的 update 尺度

线性层更新可写为

\[ \Delta W_l = -\eta_l\, \nabla_{h_l}\ell\, h_{l-1}^{\top}. \]

这说明 update 尺度由学习率 \(\eta_l\)、上游 loss gradient 和前一层 activation 共同决定。宽度变化时,若 \(\eta_l\) 不随维度调整,\(\Delta W_l h_{l-1}\) 可能不再是 \(\Theta(1)\)

Slide 49:A2 part 2:选择 LR 使 \(\| W_l\|_* √n_l-1=Θ(√n_l)\)。
Slide 49:A2 part 2:选择 LR 使 \(\| W_l\|_* √n_l-1=Θ(√n_l)\)。 查看原图

读公式:Slide 49 的 learning-rate 缩放

这页的关键是把“update 后 activation change 保持常数”翻译成学习率缩放规则。不同层、不同参数类型可能需要不同 LR scaling;这也是 muP 比普通“所有参数同一 LR”更细的原因。

Slide 50:muP mini recap:通过 \(W\) 和 \( W\) 控制 activations 与 changes;给出初始化和 Adam LR 规则。
Slide 50:muP mini recap:通过 \(W\) 和 \( W\) 控制 activations 与 changes;给出初始化和 Adam LR 规则。 查看原图

读图:Slide 50 的 baby muP 总结

muP 不是一条单独公式,而是一组随宽度缩放的初始化和学习率规则。其目标是让 activation scale 和 update scale 在不同宽度下可比。这样小模型调参才有希望迁移到大模型。

现代 LM 中 muP 的适用性

\singleslide{slides-images/slide-051.jpg}{muP 细节表:embedding、attention、MLP 与 softmax 层需要不同 initialization/LR multipliers。}{51} muP 不是“全局 LR 不变”,而是按参数角色规定 width scaling;实现遗漏任一特殊层都可能破坏 transfer。

讲义补充:源 Slide 51 的细节意义

现代 Transformer 参数类型很多:embedding、attention projection、MLP matrix multiply、softmax/output linear。不同参数在前向和反向中的维度角色不同,因此 muP 需要按参数类型分别指定初始化和 LR scaling。

Slide 52:Replicating muP:当宽度 scale 时,最优 LR 是否真的保持常数?
Slide 52:Replicating muP:当宽度 scale 时,最优 LR 是否真的保持常数? 查看原图

读图:Slide 52 的核心验证

muP 的实证检验不是看训练是否能跑,而是看不同宽度模型的 optimal LR 曲线是否对齐。如果最优 LR 随宽度稳定,muP 才实现了 scale-invariant tuning;如果漂移,说明理论假设或实现细节被破坏。

\singleslide{slides-images/slide-053.jpg}{muP robustness 问题:SwiGLU、batch、初始化、RMSNorm、Lion 与 regularizers 都偏离简化理论。}{53} 每个新组件都应在多个 widths 上做 hyperparameter-transfer test,不能仅因 backbone 使用 muP 就假设完整 recipe 稳定。

讲义补充:源 Slide 53 是 muP 实践风险清单

muP 理论常在简化网络中推导,但现代 LM 包含 gating activations、normalization、attention details、复杂 optimizer 和大 batch。每个偏离都可能破坏 scale invariance。因此使用 muP 时要做 replication,而不是只引用理论。

Slide 54:muP 对 RMSNorm gain 不鲁棒;learnable RMSNorm gains 会破坏 muP,但移除后损失很小。
Slide 54:muP 对 RMSNorm gain 不鲁棒;learnable RMSNorm gains 会破坏 muP,但移除后损失很小。 查看原图

读图:Slide 54 的工程修正

若 RMSNorm gain 是可学习参数,它可能引入随宽度变化的额外尺度自由度,使 muP 预期的 activation/update scaling 失效。一个务实修正是移除 learnable gains,并验证性能损失是否可接受。

\singleslide{slides-images/slide-055.jpg}{muP 的边界:基于矩阵几何或特殊预条件的 exotic optimizers 可能需要新的 scaling rule。}{55} Muon 等 optimizer 改变更新的谱/矩阵结构,SGD/Adam 推导出的 width multiplier 不一定保持 feature-learning regime。

讲义补充:源 Slide 55 的 optimizer 交互

muP 给的是参数化和 LR scaling 规则;optimizer 改变 update 方向和尺度。若 optimizer 使用 sign、orthogonalization 或其他非 AdamW 机制,muP 条件 A2 的 update 尺度可能重新变化,需要单独验证。

Slide 56:muP 对强 weight decay 不鲁棒,0.1 级别 weight decay 可能是显著 failure。
Slide 56:muP 对强 weight decay 不鲁棒,0.1 级别 weight decay 可能是显著 failure。 查看原图

读图:Slide 56 的 weight decay 警告

Weight decay 直接改变权重尺度动态。强 decay 会和 muP 试图维持的尺度条件冲突,导致小模型 tuning 不再可迁移。做 muP 实验时,weight decay 不是次要超参,而是需要纳入 scaling check。

Slide 57:muP 是否有用?证据表明 muP 通常有用,SP 更不稳定,muP 参数化/初始化更容易调。
Slide 57:muP 是否有用?证据表明 muP 通常有用,SP 更不稳定,muP 参数化/初始化更容易调。 查看原图

读图:Slide 57 的保守结论

muP 的价值是降低小模型到大模型超参迁移的不稳定性,而不是保证一次调参永远正确。当前证据支持“muP generally useful”,但现代 LM 组件会引入例外,需要针对架构做验证。

总结与延伸

Slide 58:Scaling in the wild recap:实践挑战包括架构超参、optimizer 超参和拟合大 Chinchilla sweep 的 compute。
Slide 58:Scaling in the wild recap:实践挑战包括架构超参、optimizer 超参和拟合大 Chinchilla sweep 的 compute。 查看原图

读图:Slide 58 的最终框架

真实 scaling 的挑战有三类:模型架构超参如何设,optimizer/LR/batch 如何随 scale 变,拟合 compute-optimal sweep 本身如何不花掉太多 compute。解决方案对应三类:muP 或架构假设降低搜索空间,empirical LR/batch scaling 拟合超参,WSD/IsoFLOP/joint fit 降低 Chinchilla sweep 成本。

本章小结

Lecture 11 把 scaling laws 从干净公式带到真实模型训练报告中。MiniCPM 展示了 muP 和 WSD 如何降低 scaling 成本;DeepSeek 展示了直接 batch/LR 和 IsoFLOP 拟合;近期模型说明 scaling 的对象已经扩展到 sparsity、MoE active parameters、architecture 和 downstream 能力;optimizer scaling 和 muP 深入则提醒我们,初始化、LR、batch、optimizer 都可能随 scale 改变。

最终 takeaways

  1. Scaling in practice 的核心不是拟合一条线,而是降低大模型训练前的决策风险。
  2. MiniCPM 的路线是 muP 稳定超参迁移,WSD 降低从头训练网格成本。
  3. DeepSeek 的路线是不依赖 muP,直接做 batch/LR 和 IsoFLOP scaling。
  4. 近期模型把 scaling 扩展到 MoE sparsity、active parameters、architecture 和 downstream metrics。
  5. Optimizer 比较必须做 scale-aware tuning,否则很容易比较错。
  6. muP 有用但不万能;RMSNorm gains、exotic optimizers、strong weight decay 等都可能破坏其假设。

拓展阅读

阅读对象 带着什么问题读 与本讲的连接
MiniCPM technical report muP、WSD 与 scaling computation 分别节省了哪一类搜索成本? 对应 Slides 6–18 的完整 recipe。
DeepSeek scaling analysis 不依赖 muP 时,LR、batch 与 IsoFLOP 最优点如何直接拟合? 对应 Slides 19–24 的经验外推路线。
Qwen、Kimi、Hunyuan、LLaMA 3、MiniMax reports 不同团队选择了哪些 scaling 对象与验收指标? 对应 Slides 25–30,比较 dense、MoE 与 downstream scaling。
StepFun scaling study 公平比较 optimizer 时,为什么必须让 LR 与 batch 一起随 scale 重调? 对应 Slides 31–41 的 optimizer scaling 方法论。
muP 与 CerebrasGPT activation 与 update 的 \(Θ(1)\) 条件如何导出 layer-wise rules? 对应 Slides 44–57 的推导与现代组件反例。
Muon scaling studies 正交化矩阵更新在什么规模、compute 与 tuning 条件下仍有收益? 对应 Slides 42–43,避免把单点胜负当成普遍规律。

最小复现实验:先验证迁移,再相信外推

选择三个宽度递增、架构其余部分尽量固定的小模型。先在最小模型上搜索 learning rate 与 batch size,再分别用 SP 和 muP 将最优设置迁移到后两个模型;每个规模记录训练早期的 activation RMS、update-to-weight ratio、稳定训练步数与最终 validation loss。最后增加一个未参与拟合的中间宽度作为 hold-out:只有当它同时落在 loss 外推曲线内、训练动力学没有 blow up,且迁移超参优于重新粗搜的基线时,才有证据说明 scaling recipe 真正在该架构上成立。