Back to Knowledge Blind Box

Knowledge Blind Box

附录 A.4 正则化、MAP 与 Dropout 的边界

ai-generatedstochastic-parrotstochastic-parrot-anatomyai

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

本附录只保留卷一第二章需要的三条接口:范数约束与惩罚目标何时相关,L1/L2 与 MAP 在什么建模前提下对应,以及 dropout 的缩放与梯度怎样计算。AdamW 的算法细节放在附录 A.8。

A.4.1 约束问题与惩罚问题

考虑

min⁡θF(θ)s.t.Ω(θ)≤C(A.4.1)\min_\theta F(\theta) \quad\text{s.t.}\quad \Omega(\theta)\le C \tag{A.4.1}

与

min⁡θF(θ)+λΩ(θ),λ≥0.(A.4.2)\min_\theta F(\theta)+\lambda\Omega(\theta), \qquad \lambda\ge0. \tag{A.4.2}

二者不是对任意 C,λC,\lambda 自动一一对应。

命题 A.4.1(惩罚解给出一个约束解) 若 λ>0\lambda>0,且 θλ\theta_\lambda 是 (A.4.2) 的全局极小点,则它也是取 C=Ω(θλ)C=\Omega(\theta_\lambda) 时 (A.4.1) 的全局极小点。

证明 对任意满足 Ω(θ)≤C\Omega(\theta)\le C 的 θ\theta,

F(θλ)+λC≤F(θ)+λΩ(θ)≤F(θ)+λC.F(\theta_\lambda)+\lambda C \le F(\theta)+\lambda\Omega(\theta) \le F(\theta)+\lambda C.

消去 λC\lambda C 得 F(θλ)≤F(θ)F(\theta_\lambda)\le F(\theta)。□\square

反方向需要更多条件。若 F,ΩF,\Omega 是正常(proper)、下半连续凸函数,存在属于其有效域相对内部的严格可行点 θˉ\bar\theta,满足 Ω(θˉ)<C\Omega(\bar\theta)<C(Slater 条件),且约束最优解存在,则强对偶和 KKT 条件给出某个乘子 λ∗≥0\lambda^*\ge0,满足

0∈∂F(θ∗)+λ∗∂Ω(θ∗),λ∗(Ω(θ∗)−C)=0.0\in\partial F(\theta^*)+\lambda^*\partial\Omega(\theta^*), \qquad \lambda^*(\Omega(\theta^*)-C)=0.

此时 θ∗\theta^* 也是相应惩罚问题的解。映射仍可能不唯一;约束不活跃时可有 λ∗=0\lambda^*=0。非凸网络、局部极小点和近似优化下,不能只凭拉格朗日直觉宣称两个训练过程等价。

A.4.2 MAP 与范数惩罚

设数据为 D\mathcal D,参数相对于固定基准测度有先验密度 p(θ)p(\theta),似然为 p(D∣θ)p(\mathcal D\mid\theta)。若后验可正规化,则

p(θ∣D)∝p(D∣θ)p(θ).p(\theta\mid\mathcal D) \propto p(\mathcal D\mid\theta)p(\theta).

MAP 点估计满足

θMAP∈arg⁡min⁡θ[−log⁡p(D∣θ)−log⁡p(θ)].(A.4.3)\theta_{\mathrm{MAP}} \in\arg\min_\theta \left[-\log p(\mathcal D\mid\theta)-\log p(\theta)\right]. \tag{A.4.3}

因此“正则项等于负对数先验”只在训练损失确为相应负对数似然、系数尺度一致且参数化固定时成立。

Gaussian 与 Laplace 例子

若 θ∼N(0,τ2I)\theta\sim\mathcal N(0,\tau^2I),则

−log⁡p(θ)=12τ2∥θ∥22+constant.-\log p(\theta) =\frac1{2\tau^2}\|\theta\|_2^2+\text{constant}.

若各坐标独立且 θj∼Laplace⁡(0,b)\theta_j\sim\operatorname{Laplace}(0,b),则

−log⁡p(θ)=1b∥θ∥1+constant.-\log p(\theta) =\frac1b\|\theta\|_1+\text{constant}.

系数取决于损失采用总和还是均值。若 −log⁡p(D∣θ)=∑i=1Nℓi(θ)-\log p(\mathcal D\mid\theta)=\sum_{i=1}^N\ell_i(\theta),高斯先验给出系数 1/(2τ2)1/(2\tau^2);若优化平均损失 N−1∑iℓiN^{-1}\sum_i\ell_i,同一个 MAP 问题中的 L2 系数变为 1/(2Nτ2)1/(2N\tau^2)。所以“L2 就是标准高斯先验”没有自动确定训练代码里的 weight decay 数值。

MAP 不是完整 Bayesian 推断

Bayesian 预测分布需要积分

p(y∣x,D)=∫p(y∣x,θ)p(θ∣D) dθ,p(y\mid x,\mathcal D) =\int p(y\mid x,\theta)p(\theta\mid\mathcal D)\,d\theta,

而 MAP 只保留一个后验众数。它不会给出后验不确定性的完整表示。此外,连续密度的众数在非线性重参数化下会受 Jacobian 影响;MAP 一般不具有参数化不变性。故不能把普通 L2 训练后的单个 checkpoint 直接称为“Bayesian 神经网络”。

L1-MAP 中可能出现的精确零,来自 ∥θ∥1\|\theta\|_1 在零点的非光滑最优性条件。对光滑损失 FF,proximal gradient 的一步为

ut=θt−η∇F(θt),u_t=\theta_t-\eta\nabla F(\theta_t), θt+1,j=sign⁡(ut,j)max⁡(∣ut,j∣−ηλ,0).(A.4.4)\theta_{t+1,j} =\operatorname{sign}(u_{t,j}) \max(|u_{t,j}|-\eta\lambda,0). \tag{A.4.4}

普通次梯度更新不具备 (A.4.4) 的显式阈值操作,不能保证坐标一到零就永久保持为零。

A.4.3 L2 penalty、weight decay 与 AdamW

对 plain SGD,若目标为

J(θ)=F(θ)+λ2∥θ∥2,J(\theta)=F(\theta)+\frac\lambda2\|\theta\|^2,

则

θt+1=(1−ηtλ)θt−ηt∇F(θt).(A.4.5)\theta_{t+1} =(1-\eta_t\lambda)\theta_t -\eta_t\nabla F(\theta_t). \tag{A.4.5}

因此在这一更新约定下,L2 penalty 与同步比例衰减代数相同。对 Adam,把 λθt\lambda\theta_t 并入梯度会使它进入一阶、二阶矩估计并接受逐坐标预条件;这不再等于 (A.4.5)。AdamW 则把参数衰减与损失梯度的矩估计解耦。它是一种优化算法定义,不自动对应某个固定先验下的 MAP。

A.4.4 Dropout 的前向与反向

令丢弃率 p∈[0,1)p\in[0,1)、保留率 q=1−pq=1-p,并令掩码坐标独立满足

rj∼Bernoulli⁡(q).r_j\sim\operatorname{Bernoulli}(q).

现代实现通常使用 inverted dropout:

h~=rq⊙h(训练),h~=h(评估).(A.4.6)\widetilde h=\frac rq\odot h \quad\text{(训练)}, \qquad \widetilde h=h \quad\text{(评估)}. \tag{A.4.6}

在给定 hh 且掩码独立时,

Er[h~∣h]=h,Var⁡r(h~j∣hj)=pqhj2.(A.4.7)\mathbb E_r[\widetilde h\mid h]=h, \qquad \operatorname{Var}_r(\widetilde h_j\mid h_j) =\frac pqh_j^2. \tag{A.4.7}

若上游梯度为 g=∇h~Lg=\nabla_{\widetilde h}L,则同一次前向使用的掩码给出

∇hL=rq⊙g.(A.4.8)\nabla_hL=\frac rq\odot g. \tag{A.4.8}

掩码为零只切断该样本和该位置的这条激活路径;共享参数仍可从 batch 中其他路径得到梯度。

期望匹配不等于精确模型平均

(A.4.7) 只匹配被 dropout 激活的一阶条件期望。一般有

Er[ϕ(h~)]≠ϕ(Er[h~])\mathbb E_r[\phi(\widetilde h)] \ne\phi(\mathbb E_r[\widetilde h])

对非线性 ϕ\phi 成立。例如 q=1/2q=1/2、标量 h≠0h\ne0、ϕ(z)=z2\phi(z)=z^2 时,h~\widetilde h 以相同概率取 00 或 2h2h,所以

E[h~2]=2h2≠h2=(E[h~])2.\mathbb E[\widetilde h^2]=2h^2 \ne h^2=(\mathbb E[\widetilde h])^2.

因此评估时关闭 dropout 不是对所有随机子网络输出的精确平均,只是成本低且广泛使用的确定性近似。dropout 是否改善泛化是依赖数据、架构和训练配方的经验问题。

A.4.5 来源

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 优化基础:梯度下降与随机逼近
  70. 63附录 A.2 感知机收敛定理
  71. 64附录 A.3 统计学习:偏差、容量与一致收敛
  72. 65附录 A.4 正则化、MAP 与 Dropout 的边界
    1. A.4.1 约束问题与惩罚问题
    2. A.4.2 MAP 与范数惩罚
    3. A.4.3 L2 penalty、weight decay 与 AdamW
    4. A.4.4 Dropout 的前向与反向
    5. A.4.5 来源
  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 与互信息边界