小批量训练让大模型更泛化,因优化路径隐含正则作用
Implicit Regularization and Generalization in Overparameterized Neural Networks
- 通过小批量SGD和损失曲面分析,发现优化路径影响泛化能力
- 小批量解的最高赫斯特征值仅为大批量的1/11.8,测试准确率高1.61个百分点
- 仅保留10%参数的稀疏子网络重训练后性能接近完整模型
经典统计学习理论预测过度参数化模型会严重过拟合,但现代深度神经网络参数远超训练样本却仍能良好泛化。本文通过控制实验探究优化动态与隐式正则在过度参数化网络中的作用。实验涵盖不同批量大小下的随机梯度下降(SGD)、通过赫斯特征值估计和权重扰动分析平坦与尖锐极小值、宽网络下的神经正切核(NTK)区间、跨模型规模的双下降现象,以及基于迭代幅度剪枝的彩票猜想验证。所有实验均在CIFAR-10和MNIST数据集上使用PyTorch完成,包含多个随机种子。结果表明,泛化能力受网络结构、优化算法和损失景观几何的交互影响显著:小批量训练始终带来更低的测试误差和更平坦的极小值;小批量解的最高赫斯特征值比大批量解低11.8倍,对应测试准确率高出1.61个百分点;仅保留10%参数的稀疏子网络在重新训练后性能仅比全模型低1.15个百分点。这些发现凸显了需构建新的学习理论框架以解释高维模型下的泛化现象。
原文摘要 · Abstract (English)
Classical statistical learning theory predicts that overparameterized models should exhibit severe overfitting, yet modern deep neural networks with far more parameters than training samples consistently generalize well. This contradiction has become a central theoretical question in machine learning. This study investigates the role of optimization dynamics and implicit regularization in enabling generalization in overparameterized neural networks through controlled experiments. We examine stochastic gradient descent (SGD) across batch sizes, the geometry of flat versus sharp minima via Hessian eigenvalue estimation and weight perturbation analysis, the Neural Tangent Kernel (NTK) regime through wide-network experiments, double descent across model scales, and the Lottery Ticket Hypothesis through iterative magnitude pruning. All experiments use PyTorch on CIFAR-10 and MNIST with multiple random seeds. Our findings demonstrate that generalization is strongly influenced by the interaction between network architecture, optimization algorithms, and loss landscape geometry. Smaller batch sizes consistently produced lower test error and flatter minima, with an 11.8x difference in top Hessian eigenvalue between small-batch and large-batch solutions corresponding to 1.61 percentage points higher test accuracy. Sparse subnetworks retaining only 10% of parameters achieved within 1.15 percentage points of full model performance when retrained from their original initialization. These results highlight the need for revised learning-theoretic frameworks capable of explaining generalization in high-dimensional model regimes.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。