大规模学习理论
本章梳理统计学习理论的核心数学框架,从 PAC 学习、VC 维、Rademacher 复杂度出发,建立深度神经网络泛化能力的理论根基。重点覆盖神经网络学习理论、泛化与优化的交互(双下降现象)、算法稳定性分析以及信息论泛化界。
1. PAC 学习框架
1.1 PAC 学习定义
定义 1.1(PAC 学习)
设
其中
定义 1.2(ERM — Empirical Risk Minimizer)
经验风险最小化算法定义为
其中
1.2 有限假设空间下的泛化界
定理 1.1(Hoeffding 不等式 + 联合界)
设
对所有
解得
定理 1.2(有限假设空间泛化界)
以至少
证明:
对每个特定假设
由 Union Bound:
设此上界等于
取
推论 1.1
令
即 ERM 的 excess risk 被该上界控制。
2. VC 维与 Rademacher 复杂度
2.1 VC 维定义
定义 2.1(打散 Shattering)
设
定义 2.2(VC 维)
备注:本节 VC 维为二分类定义;深度学习回归任务常用伪维数(Pseudo-dimension),定义为实值函数类被”打散”的最多点数,概念上为二分类 VC 维的自然推广。
若对任意
定义 2.3(Growth 函数)
Growth 函数定义为
即
定理 2.1(Sauer-Shelah Lemma)
若
当
2.2 VC 维泛化界
定理 2.2(VC 维泛化界)
设
关键引理:一致收敛界
引理 2.1(Symmetrization)
对任意假设类
其中
引理 2.2(Massart’s Lemma)
设
定理 2.2 的证明思路:
由 Symmetrization 和 Massart’s Lemma,对任意
其中
代入即得 VC 维泛化界。
2.3 Rademacher 复杂度
定义 2.4(Rademacher 复杂度)
设
其中
定义 2.5(经验 Rademacher 复杂度)
引理 2.3(Rademacher 复杂度的期望界)
对函数类
且
定理 2.3(Rademacher 泛化界)
以至少
更精确地:
证明:
步骤 1:使用 Symmetrization
步骤 2:McDiarmid 不等式( bounded differences )
由于更换单个样本
令
步骤 3:结合ERM 形式
对ERM
推论 2.1(组合 VC 维与 Rademacher)
由 Massart’s Lemma 和 Sauer-Shelah Lemma:
因此 Rademacher 界在
2.4 函数空间的 Rademacher 复杂度
对于实值函数空间
引理 2.4(Contraction Lemma)
设
其中
3. 神经网络学习理论
3.1 深度网络的 VC 维上界
定理 3.1(经典 VC 维上界,Bartlett et al.)
设神经网络具有
更精确地,对于具有
证明思路:
- 每一层参数量为
,将该层的权重矩阵 分解为符号向量和幅度向量的乘积。 - 由 ReLU 的分段线性性质,整个网络将输入空间划分为至多
个线性区域。 - 每个线性区域对应一个线性分类器,因此整个网络产生的不同二值向量数目至多为
。 - 由 Sauer-Shelah 引理,
。结合总参数 的上界,整理得 。
推论 3.1(深度 vs. 宽度)
VC 维上界与网络深度
3.2 过参数化区域的行为
定义 3.1(过参数化)
设网络参数数量
现象 3.1(过参数化与可实现性)
在过参数化设定下,神经网络在训练集上可以零误差拟合(
关键发现:过参数化并不导致过拟合——这是深度学习”悖论”的核心,即高度过参数化的网络仍能泛化良好。
定理 3.2(过参数化网络的外推)
设
其中
3.3 Neural Tangent Kernel (NTK) 与无穷宽网络
定义 3.2(Neural Tangent Kernel)
设
其中
定理 3.3(Jacot et al., 2018 — NTK 极限)
当网络宽度
且
推论 3.2(无穷宽网络 = 核ridge回归)
在 NTK 极限下,神经网络的训练行为等价于在特征空间
对应的泛化风险为
定义 3.3(NTK 范数与复杂度)
定义 RKHS 空间
备注:该式仅对训练集核矩阵、RKHS 单位范数球成立,其中
3.4 表征复杂度(Representation Complexity)
定义 3.4(表征复杂度)
设
定理 3.4(深度网络的表征能力)
设有一个深度为
这意味着:深度指数级地提升表征效率(对
定理 3.5(功率定理,Power of Depth)
存在函数族
- 深度
网络需要 个参数才能以精度 逼近; - 深度
网络只需要 个参数。
即深度带来指数级的表征能力提升。
4. 泛化与优化的交互
4.1 双下降(Double Descent)现象
定义 4.1(双下降现象)
设模型族
- classical regime(
): 随 增加而减小(偏差减小),至某个临界点 后开始上升(方差增大),形成 classical bias-variance trade-off。 - interpolation regime(
): 达到峰值,此时模型刚好能零误差拟合训练数据(插值阈值)。 - double descent regime(
): 再次下降,在严重过参数化区域反而泛化良好。
定理 4.1(双下降的理论上界)
设
其中
图示(文字描述):
R(h)
| **** <- interpolation peak
| ** **
| ** ** -----> double descent (test risk)
| ** ** /
| ** **
| ** *** (overparameterized)
|**
+---------------------------------> m (parameters)
^ ^ ^
m_GB(under) m*(interp) m >> n (overparam)
4.2 奥卡姆剃刀定理(PAC 版本)
定理 4.2(奥卡姆剃刀 — PAC 形式)
设
更一般地,对非零风险的
直觉解释:最简单的(参数最少或假设最简洁的)能解释数据的假设,泛化风险最小。这正是奥卡姆剃刀”如无必要,勿增实体”在 PAC 框架下的形式化。
定义 4.2(描述长度)
设
4.3 一致收敛与自学(Uniform vs. Non-Uniform Convergence)
定义 4.3(一致收敛)
称假设类
即收敛速度与
定义 4.4(非一致收敛)
仅满足对每个固定的
定理 4.3(一致收敛的充分条件)
若
关键区别:
- 一致收敛:所有
同时与真实风险接近,收敛率统一。 - 非一致收敛:每个
单独收敛,但不同 的收敛率可以不同。
深度学习中的一个关键问题是:常用的神经网络假设类在大规模训练下是否满足一致收敛?答案是否定的——深度网络
5. 稳定性与泛化
统一假设:本章默认损失函数
5.1 随机梯度下降的稳定性分析
定义 5.1(Uniform Stability)
设算法
对泛化风险而言,通常考虑输出为函数时的差值:
其中
定理 5.1(稳定性 implies 泛化)
如果算法
以至少
证明思路:
定义
考虑泛化误差的期望:
通过对称化(用
5.2 SGD 的稳定性
定理 5.2(SGD 的均匀稳定性)
设目标函数
在典型设置
推论 5.1(步长与稳定性)
步长越大,算法稳定性越差(
定理 5.3(Hardt et al., 2016 — SGD 的非一致稳定性)
对于
5.3 稳定性与泛化的关系定理
定理 5.4(泛化上界 via 稳定性)
设
证明:
定义虚拟样本
由三角不等式和稳定性:
取期望并利用 Hoeffding 不等式可得
6. 信息论泛化界
6.1 压缩界(Compression Bounds)
定义 6.1(压缩学习)
称学习算法
定理 6.1(压缩界 — Russo & Zou, 2015)
设算法
其中
以至少
证明核心:
利用 Donsker-Varadhan 变换:
更具体地,由 Pinsker 不等式 和数据处理不等式,可建立
推论 6.1(压缩界形式)
若算法
从而
6.2 信息瓶颈与泛化
定义 6.2(信息瓶颈,IB)
对随机变量
其中
定理 6.2(IB 的泛化解释)
设
且泛化误差可被
关键引理:
右边第一项是
6.3 Stochastic Gradient Langevin Dynamics (SGLD) 的泛化性质
定义 6.3(SGLD)
SGLD 采样算法对参数
其中
定理 6.3(SGLD 的泛化界)
设损失函数
推论 6.2(对比 SGD 与 SGLD)
- SGD(无噪声):确定性映射
,输出是点估计, 相对较小(通常为 )。 - SGLD:输出是分布
,信息论复杂度 可以显著更小,因为噪声注入了”随机性”,减少了算法对具体训练数据的依赖。
定理 6.4(噪声作为正则化)
设
即 SGLD 通过添加噪声,等价于在信息论复杂度上施加了额外的惩罚,从而改善泛化。
附录:核心不等式汇总
| 不等式 | 内容 | 用途 |
|---|---|---|
| Hoeffding | 有限假设联合界 | |
| McDiarmid | 浓度不等式 | |
| Massart | Rademacher 复杂度(有限) | |
| Sauer-Shelah | VC 维 growth 函数 | |
| Donsker-Varadhan | 信息论界 |
参考文献
- Vapnik, V. N. (1998). Statistical Learning Theory. Wiley.
- Shalev-Shwartz, S., & Ben-David, S. (2014). Understanding Machine Learning: From Theory to Algorithms. Cambridge University Press.
- Bartlett, P. L., & Mendelson, S. (2002). Rademacher and Gaussian complexities: Risk bounds and structural results. JMLR.
- Bartlett, P. L., Harvey, N., Liaw, C., & Mehrabian, A. (2019). Nearly-tight VC-dimension and pseudodimension bounds for piecewise linear neural networks. JMLR.
- Jacot, A., Hongler, C., & Gabriel, F. (2018). Neural tangent kernel: Convergence and generalization in neural networks. NeurIPS.
- Belkin, M., Hsu, D., Ma, S., & Mandal, S. (2019). Reconciling modern machine learning practice and the bias-variance trade-off. PNAS.
- Hardt, M., Recht, B., & Singer, Y. (2016). Train faster, generalize better: Stability of stochastic gradient descent. ICML.
- Russo, D., & Zou, J. (2015). Controlling bias in adaptive data analysis using information theory. AISTATS.
- Tishby, N., & Zaslavsky, N. (2015). Deep learning and the information bottleneck principle. IEEE ITW.
- Chen, Z., Cao, Y., Zou, D., & Gu, Q. (2020). Maximum likelihood principle and generalization. NeurIPS.