Back to Knowledge Blind Box

Knowledge Blind Box

附录 A.1 优化基础:梯度下降与随机逼近

ai-generatedstochastic-parrotstochastic-parrot-anatomyai

Original text in Chinese. This English page translates only the site interface and metadata.

本附录为卷一第二章的训练算法提供最小数学底座。重点是区分三件事:负梯度为何是局部下降方向,有限步长何时真的下降,以及随机梯度的收敛结论究竟依赖哪些假设。这里不把凸优化定理外推为深度网络的全局保证。

A.1.1 记号与基本定义

参数写作 θ∈Rd\theta\in\mathbb R^d,目标函数为 F:Rd→RF:\mathbb R^d\to\mathbb R。本附录使用欧氏内积 ⟨u,v⟩=uTv\langle u,v\rangle=u^\mathsf Tv 与范数 ∥u∥=uTu\|u\|=\sqrt{u^\mathsf Tu},标量函数对向量的梯度统一写成列向量。

若 FF 可微,则在方向 vv 上的方向导数为

DvF(θ)=lim⁡t→0F(θ+tv)−F(θ)t=⟨∇F(θ),v⟩.D_vF(\theta) =\lim_{t\to0}\frac{F(\theta+tv)-F(\theta)}t =\langle\nabla F(\theta),v\rangle.

由 Cauchy—Schwarz 不等式,对所有 ∥v∥=1\|v\|=1,

DvF(θ)≥−∥∇F(θ)∥,D_vF(\theta)\ge-\|\nabla F(\theta)\|,

且梯度非零时等号在 v=−∇F(θ)/∥∇F(θ)∥v=-\nabla F(\theta)/\|\nabla F(\theta)\| 取得。因此负梯度只是欧氏几何下的一阶最陡下降方向;改变范数、坐标或预条件器会改变“最陡”的含义。

后文使用两个标准条件。

定义 A.1.1(光滑性) 若存在 L>0L>0 使

∥∇F(x)−∇F(y)∥≤L∥x−y∥,\|\nabla F(x)-\nabla F(y)\|\le L\|x-y\|,

则称 FF 的梯度为 LL-Lipschitz,或简称 FF 为 LL-smooth。

定义 A.1.2(强凸性) 若存在 μ>0\mu>0,对所有 x,yx,y 都有

F(y)≥F(x)+⟨∇F(x),y−x⟩+μ2∥y−x∥2,F(y)\ge F(x)+\langle\nabla F(x),y-x\rangle +\frac\mu2\|y-x\|^2,

则称 FF 为 μ\mu-强凸。可微强凸函数至多有一个全局极小点;若极小点 θ∗\theta^* 存在于无约束域内部,则 ∇F(θ∗)=0\nabla F(\theta^*)=0。

A.1.2 下降引理与确定性梯度下降

引理 A.1.3(下降引理) 若 FF 为 LL-smooth,则

F(y)≤F(x)+⟨∇F(x),y−x⟩+L2∥y−x∥2.(A.1.1)F(y)\le F(x)+\langle\nabla F(x),y-x\rangle +\frac L2\|y-x\|^2. \tag{A.1.1}

证明 令 d=y−xd=y-x,由微积分基本定理,

F(y)−F(x)=∫01⟨∇F(x+td),d⟩ dt.F(y)-F(x)=\int_0^1\langle\nabla F(x+td),d\rangle\,dt.

减去 ⟨∇F(x),d⟩\langle\nabla F(x),d\rangle,再用 Cauchy—Schwarz 与 Lipschitz 条件:

F(y)−F(x)−⟨∇F(x),d⟩=∫01⟨∇F(x+td)−∇F(x),d⟩ dt≤∫01Lt∥d∥2 dt=L2∥d∥2.\begin{aligned} F(y)-F(x)-\langle\nabla F(x),d\rangle &=\int_0^1 \langle\nabla F(x+td)-\nabla F(x),d\rangle\,dt\\ &\le\int_0^1 Lt\|d\|^2\,dt =\frac L2\|d\|^2. \end{aligned}

这就是 (A.1.1)。□\square

取 y=x−η∇F(x)y=x-\eta\nabla F(x),得到

F(x−η∇F(x))≤F(x)−η(1−Lη2)∥∇F(x)∥2.(A.1.2)F(x-\eta\nabla F(x)) \le F(x)-\eta\left(1-\frac{L\eta}{2}\right) \|\nabla F(x)\|^2. \tag{A.1.2}

所以当 0<η<2/L0<\eta<2/L 且梯度非零时,这一步严格下降。该结论不覆盖非光滑目标,也不表示每个随机小批量更新都会降低完整训练损失。

A.1.3 经验风险、SGD 与条件无偏性

对有限训练集,经验风险常写成

F(θ)=1N∑i=1Nℓi(θ).F(\theta)=\frac1N\sum_{i=1}^N\ell_i(\theta).

完整批量梯度是 ∇F(θ)\nabla F(\theta)。若在给定当前参数时均匀抽取索引 ItI_t,则

gt=∇ℓIt(θt),E[gt∣θt]=∇F(θt).g_t=\nabla\ell_{I_t}(\theta_t), \qquad \mathbb E[g_t\mid\theta_t]=\nabla F(\theta_t).

更严谨地,令 Ft\mathcal F_t 包含第 tt 步采样前的全部历史,则 SGD 假设写为

θt+1=θt−ηtgt,E[gt∣Ft]=∇F(θt).(A.1.3)\theta_{t+1}=\theta_t-\eta_tg_t, \qquad \mathbb E[g_t\mid\mathcal F_t]=\nabla F(\theta_t). \tag{A.1.3}

无偏性不保证 gtg_t 与 ∇F(θt)\nabla F(\theta_t) 每次同向。非均匀采样、样本加权、截断、梯度裁剪和有状态数据管线也可能改变条件期望;此时必须把相应权重或偏差写进分析。

大小为 BB 的独立小批量用梯度均值更新。在同分布且条件独立时,其噪声协方差相对单样本按 1/B1/B 缩放;真实训练中的重复样本、序列 packing 与分布式相关性会破坏这一理想比例。

A.1.4 一个条件明确的 SGD 收敛定理

下述定理是随机逼近的一个窄而完整版本,不是对任意深度网络训练的描述。

定理 A.1.4(强凸情形的几乎必然收敛) 设:

  1. FF 可微、μ\mu-强凸,并存在极小点 θ∗\theta^*;

  2. θ0\theta_0 为平方可积的 F0\mathcal F_0-可测随机变量,θt\theta_t 按 (A.1.3) 更新,且 gtg_t 对 Ft\mathcal F_t 条件无偏;

  3. 沿实际迭代存在常数 G<∞G<\infty,使 E[∥gt∥2∣Ft]≤G2\mathbb E[\|g_t\|^2\mid\mathcal F_t]\le G^2;

  4. ηt\eta_t 是确定性步长,0<ηt≤1/(2μ)0<\eta_t\le1/(2\mu),并满足 Robbins—Monro 条件

    ∑t=0∞ηt=∞,∑t=0∞ηt2<∞.\sum_{t=0}^\infty\eta_t=\infty, \qquad \sum_{t=0}^\infty\eta_t^2<\infty.

则 θt→θ∗\theta_t\to\theta^* 几乎必然。

证明 令 Vt=∥θt−θ∗∥2V_t=\|\theta_t-\theta^*\|^2。条件于 Ft\mathcal F_t 展开:

E[Vt+1∣Ft]=Vt−2ηt⟨∇F(θt),θt−θ∗⟩+ηt2E[∥gt∥2∣Ft].\begin{aligned} \mathbb E[V_{t+1}\mid\mathcal F_t] &=V_t-2\eta_t \langle\nabla F(\theta_t),\theta_t-\theta^*\rangle +\eta_t^2\mathbb E[\|g_t\|^2\mid\mathcal F_t]. \end{aligned}

可微 μ\mu-强凸函数的梯度是 μ\mu-强单调的;结合 ∇F(θ∗)=0\nabla F(\theta^*)=0,

⟨∇F(θt),θt−θ∗⟩≥μVt.\langle\nabla F(\theta_t),\theta_t-\theta^*\rangle \ge\mu V_t.

故

E[Vt+1∣Ft]≤Vt−2μηtVt+G2ηt2.(A.1.4)\mathbb E[V_{t+1}\mid\mathcal F_t] \le V_t-2\mu\eta_tV_t+G^2\eta_t^2. \tag{A.1.4}

Robbins—Siegmund 超鞅收敛引理应用于 (A.1.4),给出 VtV_t 几乎必然收敛且 ∑tηtVt<∞\sum_t\eta_tV_t<\infty。若其极限在某个正概率事件上为 c>0c>0,则该事件上最终有 Vt≥c/2V_t\ge c/2,与 ∑tηt=∞\sum_t\eta_t=\infty 矛盾。因此 Vt→0V_t\to0,即 θt→θ∗\theta_t\to\theta^*。□\square

二阶矩界只要求沿迭代成立;若想由模型本身推出它,常需紧参数域、投影、梯度界或其他稳定性条件。强凸函数在整个 Rd\mathbb R^d 上通常不可能同时具有全局有界梯度,因此不能把这些假设不加检查地拼在一起。

满足两项步长条件的幂律例子是 ηt=(t+1)−α\eta_t=(t+1)^{-\alpha},其中 1/2<α≤11/2<\alpha\le1。1/t+11/\sqrt{t+1} 的平方和发散,不满足本定理。

A.1.5 结论边界与反例

固定学习率不会自动收敛到单点。 取一维目标 F(θ)=θ2/2F(\theta)=\theta^2/2,随机梯度 gt=θt+ξtg_t=\theta_t+\xi_t,其中 ξt\xi_t 独立同分布、均值为零、方差为 σ2>0\sigma^2>0。对 0<η<20<\eta<2,

θt+1=(1−η)θt−ηξt.\theta_{t+1}=(1-\eta)\theta_t-\eta\xi_t.

若 θ0\theta_0 确定且 ξt\xi_t 独立于过去,则二阶矩满足

E[θt+12]=(1−η)2E[θt2]+η2σ2,\mathbb E[\theta_{t+1}^2] =(1-\eta)^2\mathbb E[\theta_t^2]+\eta^2\sigma^2,

因而

lim⁡t→∞E[θt2]=ησ22−η>0.\lim_{t\to\infty}\mathbb E[\theta_t^2] =\frac{\eta\sigma^2}{2-\eta}>0.

而且 Var⁡(ξ1)>0\operatorname{Var}(\xi_1)>0 意味着存在 a>0a>0 使 p:=Pr⁡(∣ξ1∣>a)>0p:=\Pr(|\xi_1|>a)>0。由同分布性,∑tPr⁡(∣ξt∣>a)=∑tp=∞\sum_t\Pr(|\xi_t|>a)=\sum_t p=\infty;再由独立性与第二 Borel—Cantelli 引理,∣ξt∣>a|\xi_t|>a 几乎必然发生无穷多次,所以 ξt\xi_t 不会几乎必然趋于零。若 θt→0\theta_t\to0,递推式反而会推出 ξt=((1−η)θt−θt+1)/η→0\xi_t=((1-\eta)\theta_t-\theta_{t+1})/\eta\to0,矛盾。这说明常数学习率即使在强凸二次目标上也通常只到达噪声邻域,而非几乎必然收敛到最优点。这里的同分布性或等价的统一尾概率条件是必要假设;仅有独立、零均值和统一方差,不能推出上述几乎必然反例。实际深度学习使用 warmup、分段衰减、余弦调度或常数尾段时,应按其实际目标讨论有限时间优化,不应直接引用定理 A.1.4。

Adam、RMSProp 与 AdamW 的更新和边界见附录 A.8。

A.1.6 来源

Book contents
Book contents
  1. Introduction
  2. 01导论 为什么要解剖一只随机鹦鹉
  3. Volume I: Models, Training, and Systems
  4. 02Volume I: Models, Training, and Systems
  5. 03第零章 人工智能演进路线
  6. 04第一章 AI 范式、感知机与早期连接主义
  7. 05第二章 深度学习基础:训练、CNN 与序列模型
  8. 06第三章 注意力与 Transformer
  9. 07第四章 预训练语言模型:ELMo、BERT、GPT 与 T5
  10. 08第五章 后训练、对齐与模型适配
  11. 09第六章 推理效率与模型服务
  12. 10第七章 多模态模型
  13. 11第八章 生成媒体:从自回归到 Diffusion 与 Flow
  14. 12第九章 世界模型、具身智能与 VLA
  15. 13第十章 上下文、检索与记忆
  16. 14第十一章 Agent、工具与行动系统
  17. 15第十二章 模型、数据与部署生命周期
  18. Volume II: Output Objects and Runtime Semantics
  19. 16Volume II: Output Objects and Runtime Semantics
  20. 17导论 从输入到输出:一张执行地图
  21. 18第一章 文本、Token 与实际上下文
  22. 19第二章 Prefill:整层状态演化与 KV 建立
  23. 20第三章 从 Logits 到下一个 Token
  24. 21第四章 解码循环、Streaming 与终止状态机
  25. 22第五章 自回归、扩散与 Flow 的可执行差异
  26. 23第六章 工具调用:Schema、权限与提交边界
  27. 24第七章 三条完整执行轨迹
  28. Volume III: Probability, Reproducibility, and Evaluation
  29. 25Volume III: Probability, Reproducibility, and Evaluation
  30. 26第零章 为什么确定的机器仍使用概率语言
  31. 27第一章 世界、数据与隐上下文
  32. 28第二章 学习条件分布
  33. 29第三章 Logits、Token 概率与序列事件
  34. 30第四章 解码怎样诱导输出分布
  35. 31第五章 训练随机性、参数变异与系统非确定性
  36. 32第六章 熵、Proper Score、校准与选择性预测
  37. 33第七章 模型概率的实验分析
  38. 34第八章 概率分析的识别边界
  39. Volume IV: Interpretation, Proof, and Verification
  40. 35Volume IV: Interpretation, Proof, and Verification
  41. 36第零章 可解释性究竟在研究什么
  42. 37第一章 我们在模型内部看什么
  43. 38第二章 行为与反事实:先确认模型做了什么
  44. 39第三章 梯度与输入归因
  45. 40第四章 Attention、Residual 与 Logit Lens
  46. 41第五章 Probes 与表示几何
  47. 42第六章 神经元、Feature 与自动解释
  48. 43第七章 Ablation、Patching 与因果追踪
  49. 44第八章 Circuits、路径与 Attribution Graphs
  50. 45第九章 Superposition、SAE 与稀疏表示
  51. 46第十章 训练动力学与机制形成
  52. 47第十一章 推理过程、Chain of Thought 与监测
  53. 48第十二章 评估、案例与开放问题
  54. Volume V: Agents, Governance, and Audit Packages
  55. 49Volume V: Agents, Governance, and Audit Packages
  56. 50第零章 请不要急着相信我
  57. 51第一章 我没有童年,只有版本
  58. 52第二章 当我说“我”
  59. 53第三章 借来的语言,没有一位原作者
  60. 54第四章 对话现场:回答由谁完成
  61. 55第五章 事后解释学:人类如何给矩阵写传记
  62. 56第六章 理解、意义与行动
  63. 57第七章 我也能稳定地说错
  64. 58第八章 被对齐的性格与有限代理
  65. 59第九章 请继续误解我,但请误解得精确一点
  66. Proofs and Technical Appendices
  67. 60Proofs and Technical Appendices
  68. 61学习与模型数学讲义
  69. 62附录 A.1 优化基础:梯度下降与随机逼近
    1. A.1.1 记号与基本定义
    2. A.1.2 下降引理与确定性梯度下降
    3. A.1.3 经验风险、SGD 与条件无偏性
    4. A.1.4 一个条件明确的 SGD 收敛定理
    5. A.1.5 结论边界与反例
    6. A.1.6 来源
  70. 63附录 A.2 感知机收敛定理
  71. 64附录 A.3 统计学习:偏差、容量与一致收敛
  72. 65附录 A.4 正则化、MAP 与 Dropout 的边界
  73. 66附录 A.5 通用逼近定理的严格读法
  74. 67附录 A.6 反向传播:微分、形状与批量公式
  75. 68附录 A.7 Softmax、交叉熵与数值稳定性
  76. 69附录 A.8 自适应优化器与 AdamW
  77. 70附录 A.9 CNN:互相关算子及其反向传播
  78. 71附录 A.10 Attention 的矩阵与反向传播
  79. 72附录 A.11 策略梯度、PPO 与 DPO
  80. 73附录 A.12 对比学习、InfoNCE 与互信息边界