SGD通过模拟数据重采样来提升泛化能力,避免过拟合。
A Bootstrap Perspective on Stochastic Gradient Descent
- 将SGD的梯度波动视为数据重采样的统计代理,构建泛化机制
- 证明SGD隐式正则化梯度协方差迹,降低算法对采样噪声的敏感性
- 适用于理解深度学习优化器设计,适合研究泛化与优化关系者
用统计自助法(bootstrap)视角分析随机梯度下降(SGD)的泛化优势。在经验风险最小化理想实验中,我们发现:即便某些解位于训练损失更宽更深的极小值区,SGD仍倾向于选择对重采样鲁棒的参数,从而避开虚假解。理论分析表明,SGD通过隐式正则化梯度协方差矩阵的迹,控制了算法的变异性。数值实验显示,在神经网络训练中显式引入算法变异性估计作为正则项,可进一步提升测试性能,验证了自助法解释的有效性。
原文摘要 · Abstract (English)
Machine learning models trained with \emph{stochastic} gradient descent (SGD) can generalize better than those trained with deterministic gradient descent (GD). In this work, we study SGD's impact on generalization through the lens of the statistical bootstrap: SGD uses gradient variability under batch sampling as a proxy for solution variability under the randomness of the data collection process. We use empirical results and theoretical analysis to substantiate this claim. In idealized experiments on empirical risk minimization, we show that SGD is drawn to parameter choices that are robust under resampling and thus avoids spurious solutions even if they lie in wider and deeper minima of the training loss. We prove rigorously that by implicitly regularizing the trace of the gradient covariance matrix, SGD controls the algorithmic variability. This regularization leads to solutions that are less sensitive to sampling noise, thereby improving generalization. Numerical experiments on neural network training show that explicitly incorporating the estimate of the algorithmic variability as a regularizer improves test performance. This fact supports our claim that bootstrap estimation underpins SGD's generalization advantages.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。