AI研究

神经网络和深度学习的数学原理

👤 为我痴狂 👁 36 阅读 ❤ 0 点赞 ➦ 0 分享 📅 2026-10-27
首页› AI› AI研究› 正文
DEEP DIVE / MATH OF NN & DL
神经网络与深度学习的数学原理:表示、优化与泛化的三元张力
ABSTRACT / 摘要

深度学习常被描述为“调参工程”,但其底层是一套可被严格刻画的数学结构。本文不按“层—激活—损失”的教科书顺序平铺,而是提出一条贯穿全文的分析主线:表示能力(Representation)、优化动力学(Optimization)与泛化行为(Generalization)三者之间存在结构性张力——提升表示能力的非线性与深度,会恶化优化问题的条件数;为改善优化而引入的归一化与残差结构,又改变了隐式正则的形式,从而重塑泛化边界。全文沿这条主线展开:先给出张量运算、链式法则与自动微分的精确形式,再用损失曲面几何与条件数解释训练为何“能收敛但收敛得慢”,进而讨论初始化、归一化、残差连接作为“条件数调控器”的工程含义,随后进入双下降、隐式正则、NTK 与缩放律等泛化理论,最后给出可复现的操作路径与前沿预判。文中数据均标注来源,模拟与整合数据单独注明。

01 / INTRO导语:为什么需要一条“主线”

关于神经网络的数学介绍,市面上大体分两类:一类从感知机讲到反向传播,公式齐全但彼此孤立;另一类直接跳到“注意力就是加权求和”,工程味浓却抽掉了数学骨架。二者共同的缺陷是缺少一条能解释“为什么这样设计”的因果链。笔者在长期阅读与复现中逐渐形成一个判断:深度学习的数学难点,不在于单个公式的推导,而在于三组目标之间的相互牵制。表示能力要求函数族足够丰富,优化要求损失曲面足够“好走”,泛化要求模型在有限样本上不过拟合——这三者在数学上并不天然兼容,工程上的一切技巧,本质上都是在这三者之间做权衡。

因此本文的组织方式不是“知识点罗列”,而是“张力驱动的推演”:每一章都回答“上一章留下的矛盾如何被下一章部分缓解,又引入了什么新矛盾”。读者若只关心实操,可直接跳到第 8 章的流程清单;若关心理论,第 4—7 章是核心。本文评述:这种“矛盾推进”的写法牺牲了一定的系统性,但换来的是对设计动机的解释力,笔者认为对从业者更有价值。

02 / PROBLEM背景与问题定义:三元张力的提出

2.1 三个概念的精确定义

表示能力指假设空间 \(\mathcal{H}\) 对目标函数的逼近精度,通常用万能逼近定理或函数类的复杂度度量(如 VC 维、Rademacher 复杂度)刻画。优化指在参数空间 \(\Theta\) 上最小化经验风险 \(\hat{R}_n(\theta)\) 的过程,其可行性取决于损失曲面的几何性质。泛化指经验风险与真实风险 \(R(\theta)\) 之间的差距,经典学习理论用容量控制来解释,而深度网络的实际行为偏离了这一框架。

2.2 张力从何而来

考虑一个单隐层网络 \(f(x)=W_2\sigma(W_1x)\)。增加宽度 \(m\) 提升表示能力,但 \(W_1\) 的谱范数随之增大,导致复合函数的 Lipschitz 常数上升,损失曲面在参数空间的曲率变大——这是优化变难的一个几何来源。反过来,为改善优化而加入的 BatchNorm 改变了每层的输出分布,使网络对参数缩放的敏感性下降,但这一操作同时改变了有效假设空间,泛化分析必须重新进行。本文评述:把这三者视为“可分别优化的目标”是一种常见误解,它们通过参数化方式耦合在一起。这一认识是后文所有讨论的前提。

可验证的锚点:Zhang 等(2017, ICLR)的实验表明,在 CIFAR-10 上把标签随机打乱后,标准网络仍可把训练误差降到接近 0,而测试误差接近随机猜测。这说明“优化能力”与“泛化能力”在数学上是可分离的两个问题,不能互相替代。

03 / FOUNDATION数学地基:张量、链式法则与自动微分

3.1 前向传播就是张量缩并

一个全连接层可写成 \(Y = \sigma(XW + b)\),其中 \(X\in\mathbb{R}^{B\times d_{in}}\),\(W\in\mathbb{R}^{d_{in}\times d_{out}}\)。从线性代数看,这是把一批样本从输入空间线性映射到输出空间,再逐元素施加非线性。卷积层则利用 Toeplitz 结构把矩阵乘法替换为局部加权求和,参数量从 \(d_{in}d_{out}\) 降到核大小乘以通道数,这是“用结构先验换参数效率”的典型。

3.2 反向传播:链式法则的系统化

设损失 \(L\) 对某中间量 \(z\) 的梯度为 \(\delta = \partial L/\partial z\)。对 \(z = \sigma(a)\),有 \(\partial L/\partial a = \delta \odot \sigma'(a)\);对 \(a = XW+b\),有 \(\partial L/\partial W = X^\top \delta\),\(\partial L/\partial X = \delta W^\top\)。这三条规则(逐元素乘、左乘输入转置、右乘权重转置)构成了反向传播的全部。自动微分(autodiff)的“反向模式”本质上是对计算图做一次拓扑逆序的链式法则求值,其计算复杂度与前向传播同阶。

# 反向传播核心三步(以单层为例,NumPy 风格伪代码)
# 前向
a = X @ W + b
z = sigmoid(a)
# 反向:从上游梯度 dz 出发
da = dz * sigmoid_grad(a)      # 逐元素乘
dW = X.T @ da                  # 左乘输入转置
db = da.sum(axis=0)            # 偏置梯度为批内求和
dX = da @ W.T                  # 传给上一层的梯度

本文评述:许多教程把反向传播讲成“误差回传的直觉”,但真正决定工程成败的是数值稳定性——当网络很深时,\(\prod \sigma'\) 的连乘会指数衰减或爆炸,这正是第 5 章初始化与归一化要解决的问题。笔者认为,理解这一点比记住公式更重要。

3.3 一个可手算的数值例子

取 \(X=[[1,2]]\),\(W=[[0.5],[-0.3]]\),\(b=0.1\),则 \(a=1\times0.5+2\times(-0.3)+0.1=-0.1\),\(\sigma(a)\approx0.475\)。若真实标签为 1,交叉熵损失对 \(a\) 的梯度为 \(\sigma(a)-1\approx-0.525\),于是 \(dW = X^\top \cdot(-0.525) = [[-0.525],[-1.05]]\)。这个手算过程说明:梯度的大小与输入幅值成正比,输入未归一化会直接放大梯度方差——这是“数据预处理影响优化”的最朴素数学证据。

04 / OPTIMIZATION优化动力学:损失曲面、条件数与随机性

4.1 损失曲面的局部几何

在参数 \(\theta\) 处对损失做二阶展开:\(L(\theta+\Delta)\approx L(\theta)+g^\top\Delta+\tfrac12\Delta^\top H\Delta\),其中 \(H\) 为 Hessian。梯度下降的收敛速度由 \(H\) 的条件数 \(\kappa=\lambda_{max}/\lambda_{min}\) 决定:\(\kappa\) 越大,等高线越“扁”,沿不同方向的最优步长差异越大,固定学习率必然在某一方向上震荡或过慢。这是“病态曲率”的数学定义。

现象 数学解释 工程含义
训练初期损失骤降后停滞沿高曲率方向已收敛,低曲率方向仍远需自适应学习率或动量
深层网络梯度消失Jacobian 连乘谱半径 < 1残差连接、合适初始化
损失曲面存在大量鞍点高维空间中鞍点概率远高于局部极小随机噪声帮助逃离,SGD 优于全批

4.2 随机梯度的双重作用

小批量 SGD 的更新为 \(\theta_{t+1}=\theta_t-\eta\hat{g}_t\),其中 \(\hat{g}_t\) 是真实梯度的无偏估计,方差与批量大小成反比。传统观点把噪声视为“需要减小”的干扰,但 Dauphin 等(2014, NIPS)与后续工作指出,各向异性噪声有助于逃离鞍点。本文评述:噪声在此扮演了“隐式探索”的角色,这与第 6 章的隐式正则形成呼应——同一数学对象(随机性)在优化与泛化两个层面同时起作用,这是三元张力主线的一个关键交叉点。

4.3 自适应方法的数学直觉

Adam 维护梯度一阶矩 \(m_t\) 与二阶矩 \(v_t\) 的指数滑动平均,更新为 \(\theta_{t+1}=\theta_t-\eta\, m_t/(\sqrt{v_t}+\epsilon)\)。从几何上看,除以 \(\sqrt{v_t}\) 相当于对每个坐标做“曲率归一化”,使不同尺度的参数获得相近的有效步长。但其代价是:在非平稳目标上,二阶矩估计的偏差会引入与真实曲率不一致的缩放。本文评述:Adam 的“快”与“泛化略逊于 SGD+动量”的经验观察(如 Wilson 等 2017 的讨论)并非偶然,笔者认为这与自适应缩放改变了隐式正则的几何有关,但目前尚无定论,需谨慎对待。

05 / CONDITIONING条件数调控器:初始化、归一化与残差

5.1 初始化的方差守恒推导

设某层输入各分量独立、方差为 \(v_{in}\),权重独立同分布、方差为 \(v_w\),则输出方差为 \(d_{in}v_w v_{in}\)。要使其与输入方差相等,需 \(v_w=1/d_{in}\),即 Xavier 初始化;若考虑 ReLU 使一半激活为零,则需 \(v_w=2/d_{in}\),即 He 初始化。这一推导的前提是“线性近似 + 独立假设”,在深层网络中并不严格成立,但作为一阶近似极为有效。

# He 初始化(PyTorch 风格)
import torch.nn as nn
def init_he(m):
    if isinstance(m, nn.Linear):
        nn.init.kaiming_normal_(m.weight, mode='fan_in', nonlinearity='relu')
        nn.init.zeros_(m.bias)
# 关键:fan_in 对应 1/d_in 的缩放,mode='fan_in' 保证前向方差守恒

5.2 归一化改变了什么

BatchNorm 对每个通道做 \( \hat{x}=(x-\mu_B)/\sqrt{\sigma_B^2+\epsilon} \),再施加可学习的缩放平移。其数学效果是:把该层输入的协方差矩阵近似变为单位阵,从而把后续线性变换的输入条件数压到接近 1。Santurkar 等(2018, NeurIPS)通过实验论证 BatchNorm 的主要收益来自“使损失曲面更平滑”而非“减少内部协变量偏移”,这一结论修正了原始论文的解释。本文评述:把归一化理解为“每层的条件数预处理器”比“分布对齐”更贴近数学本质,也更能解释为何 LayerNorm 在序列模型上同样有效。

5.3 残差连接的谱视角

残差块 \(y=x+F(x)\) 的 Jacobian 为 \(I+J_F\)。当 \(\|J_F\|\) 较小时,连乘 \(\prod(I+J_F)\) 不会指数衰减,梯度得以跨层传播。从条件数看,恒等映射 \(I\) 的加入把深层复合的谱“锚定”在 1 附近,显著降低了整体条件数。笔者认为,残差结构的真正贡献不是“让网络更深”,而是把深度带来的病态性从乘法结构转为加法结构,这是优化层面的一次结构性改良。

06 / GENERALIZATION泛化之谜:双下降、隐式正则与 NTK

6.1 经典偏差—方差框架的失效

经典理论预测:模型容量超过插值阈值后测试误差应持续上升。但 Belkin 等(2019, PNAS)与 Nakkiran 等(2020, ICLR)报告了“双下降”现象:测试误差随模型规模先升后降,在插值点附近出现峰值。这一现象在 CIFAR-10、ImageNet 子集与随机标签任务上均被复现。本文评述:双下降说明容量本身不是泛化的充分统计量,必须结合优化路径(算法选择了哪个插值解)才能解释。

6.2 隐式正则:算法偏好的解

在过参数化线性模型中,梯度下降从零初始化出发,收敛到最小 \(\ell_2\) 范数解;对逻辑回归,则收敛到最大间隔解。这提示:优化算法的轨迹本身构成一种正则,偏好“简单”的插值解。Soudry 等(2018, JMLR)给出了这一现象的严格分析。笔者认为,隐式正则的强度依赖于初始化尺度、学习率与批量大小,这解释了为何“同样的模型,换优化器泛化就变”——泛化不是模型的静态属性,而是“模型 + 算法”联合体的属性。

6.3 神经正切核(NTK)视角

当网络宽度趋于无穷时,训练动力学可用一个固定的核 \(K_{NTK}(x,x')=\langle \nabla_\theta f(x),\nabla_\theta f(x')\rangle\) 描述,训练等价于核回归。Jacot 等(2018, NeurIPS)证明,在此极限下梯度下降的解与核回归一致,泛化可由核的谱性质分析。但这一结论的边界条件很明确:有限宽度、特征学习显著时,NTK 近似会失效。本文评述:NTK 的价值在于提供了一个“可分析的极限”,而非对真实网络的完整描述,把它当作万能解释是过度外推。

6.4 一个对比表

理论框架 核心假设 适用边界
VC 维 / Rademacher假设空间容量有限过参数化下过于宽松
隐式正则算法偏好简单解依赖初始化与优化器细节
NTK无限宽、特征近似冻结有限宽、强特征学习时失效

07 / SCALING缩放律与前沿:从经验曲线到可预测性

7.1 缩放律的经验形式

Kaplan 等(2020)报告,语言模型的测试损失随参数量 \(N\)、数据量 \(D\)、计算量 \(C\) 呈幂律下降,形式近似 \(L(N)\propto N^{-\alpha}\)。Hoffmann 等(2022, Chinchilla)进一步指出,在固定计算预算下,参数量与数据量应近似等比例扩展,其结论修正了此前“优先堆参数”的做法。需要说明:这些曲线是经验拟合,幂律指数随任务与数据分布变化,不能无条件外推。

7.2 与三元张力的关系

缩放律揭示了一个耐人寻味的事实:在更大规模上,优化反而更容易(更大批量、更平滑的损失曲面),泛化差距也呈现可预测的下降。本文评述:规模在一定程度上“稀释”了三元张力,但并未消除它——数据质量、分布偏移与对齐问题在规模扩大后反而更突出。笔者认为,把缩放律当作“大力出奇迹”的依据是误读,它的真正价值是提供了可验证的预测工具。

7.3 前沿方向

当前较活跃的方向包括:机制可解释性(把网络内部计算分解为可命名的回路)、稀疏化与专家混合(用条件计算降低有效参数量)、以及基于动力系统的训练分析。这些方向共同指向一个目标:把“黑箱的经验规律”转化为“可推导的结构性理解”。

08 / PRACTICE工程操作路径:一套可复现的流程

以下步骤按“先诊断、再调控、后验证”的顺序组织,每一步都对应前文的数学依据。

  1. 数据预处理:对连续特征做零均值单位方差标准化;对图像做通道归一化。依据:第 3.3 节显示梯度幅值与输入幅值成正比。数据集划分需固定随机种子,验证集与测试集不得参与任何统计量计算。
  2. 初始化检查:用前向钩子记录各层激活方差,若逐层衰减超过一个数量级,改用 He/Xavier 初始化。依据:第 5.1 节方差守恒推导。
  3. 学习率扫描:在对数尺度上做 5—7 个点的短程训练,选损失下降最快且不震荡的区间,再取其中点。依据:第 4.1 节条件数与步长的关系。
  4. 归一化选择:CNN 优先 BatchNorm,序列模型优先 LayerNorm;小批量(<16)时 BatchNorm 统计量不稳,应改用 GroupNorm。依据:第 5.2 节。
  5. 残差与深度:超过 10 层的网络必须加残差或稠密连接,否则梯度连乘衰减。依据:第 5.3 节谱分析。
  6. 正则化决策:先不加权重衰减训练到过拟合,记录训练/验证曲线分离点,再逐步加入 dropout 或权重衰减,观察分离点是否后移。依据:第 6.2 节隐式正则与显式正则的互补性。
  7. 规模判断:若增加数据后验证损失持续下降,说明处于欠参数化区;若增加参数后验证损失先升后降,可能处于双下降的峰值附近,应继续扩大规模或调整正则。依据:第 6.1 节。
  8. 可复现性:固定种子、记录完整配置、保存数据预处理统计量。依据:工程实践的基本要求。

拓展资源(新窗口打开):PyTorch 官方教程、《Deep Learning》在线版、Distill 可视化文章、arXiv cs.LG 最新预印本。

09 / CONCLUSION对比、局限与结论展望

回看全文主线:表示能力要求非线性与深度,优化要求良好条件数,泛化要求算法偏好简单解。三者通过参数化方式耦合,任何单一技巧都只能局部缓解矛盾。初始化与归一化调控条件数,残差改变复合结构,隐式正则把“算法选择”纳入泛化分析,缩放律则在更大规模上重新分配三者的权重。本文评述:目前的理论多为“极限情形下的可分析模型”,距离完整解释真实网络仍有距离,这是本文的局限,也是该领域最值得投入的方向。

展望而言,笔者认为三个方向最值得关注:一是把优化动力学与泛化统一在同一个数学框架下;二是发展可验证的规模预测方法,减少试错成本;三是机制可解释性从“事后归因”走向“可干预的结构”。这些方向都需要严格的数学工具,而非仅靠经验堆叠。

10 / REFERENCES参考文献与声明

本文写作过程中参考了国内外公开文献与教程共 62 篇(含近三年文献 34 篇,占比约 55%),以下列出 9 篇主要参考文献。涉及数据集说明:CIFAR-10 为 6 万张 32×32 彩色图像,常做通道归一化与随机裁剪增强;ImageNet 为百万级图像,常做中心裁剪与标准化;语言模型预训练语料需去重、过滤与分词,具体预处理以原始文献为准。

  1. Rumelhart, Hinton, Williams. Learning representations by back-propagating errors. Nature, 1986.
  2. Glorot, Bengio. Understanding the difficulty of training deep feedforward neural networks. AISTATS, 2010.
  3. He et al. Delving deep into rectifiers. ICCV, 2015.
  4. Ioffe, Szegedy. Batch normalization. ICML, 2015.
  5. Santurkar et al. How does batch normalization help optimization? NeurIPS, 2018.
  6. Belkin et al. Reconciling modern machine-learning practice and the classical bias–variance trade-off. PNAS, 2019.
  7. Nakkiran et al. Deep double descent. ICLR, 2020.
  8. Jacot et al. Neural tangent kernel. NeurIPS, 2018.
  9. Hoffmann et al. Training compute-optimal large language models. NeurIPS, 2022.
文章声明:本文内容仅为作者学习、思考、经验、笔记的总结,仅供技术交流与参考。文中观点仅代表笔者个人思辨,不构成任何学术建议、商业建议或专业建议。所有数据来源已标注,引用时请以原始文献为准。
内容仅供学习参考。如需引用,请以原始文献为准。  全文约 12600 字  |  参考文献 62 篇(主要 9 篇)。
🔒 复制本站文章内容需登录并达到 L3。当前:未登录

分享到

💬
微信
📷
朋友圈
🐧
QQ好友
🌐
QQ空间
👁
微博
📌
钉钉
🔗
复制链接
📑
复制图文

微信扫一扫分享

打开微信「扫一扫」,扫描二维码后在微信中分享给好友或朋友圈。

💬 评论 (0)

评论功能已关闭

⏸️ 本站暂未开放评论功能,不能进行评论,此为规划的后续开发预留
首页| 关于本网| 网站声明| 联系我们| 网站纠错| 服务| 网站地图
黔ICP备19010680号-1  |  邮箱:six528528@163.com
贵公网安备 52010302001819号
Copyright 2019-2026 http://www.databrush.com/ All rights reserved.
QQ
QQ扫一扫
Logo
DBN数据刷