arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04545cs.LGmath.STstat.TH
收藏

分支缩放表现为隐式架构正则化以改善过参数化ResNet的泛化能力

Branch Scaling Manifests as Implicit Architectural Regularization for Improving Generalization in Overparameterized ResNets

Huawei Large Model Data Technology Lab, Shenzhen(华为大模型数据技术实验室,深圳);Tsinghua University, Beijing(清华大学,北京);Kyoto University, Kyoto(京都大学,京都)

Zixiong Yu, Guhan Chen, Jianfa Lai, Bohan Li, Songtao Tian

AI总结:

本文研究残差网络中分支缩放因子对过参数化ResNet泛化性能的影响,通过理论分析证明快速深度衰减的缩放因子结合早停可实现极小极大最优泛化率,并利用神经正切核(NTK)近似解释其机制。

摘要:

残差分支中的缩放因子已成为提升神经网络性能的流行方法,特别是在无归一化架构中。虽然先前的工作主要从优化角度研究缩放效应,本文通过泛化理论的视角探讨其在残差架构中的作用。具体来说,我们证明具有恒定缩放因子的宽残差网络(ResNet)随着深度增加渐近地变得不可学习。相反,当缩放因子表现出快速的深度方向衰减并结合早停时,过参数化ResNet实现了极小极大最优泛化率。为了建立这一结论,我们证明宽ResNet的泛化能力可以通过与神经正切核(NTK)相关的核回归来近似。我们的理论发现通过合成数据和真实世界分类任务(包括MNIST和CIFAR-100)的实验得到验证。

英文摘要:

Scaling factors in residual branches have emerged as a prevalent method for boosting neural network performance, especially in normalization-free architectures. While prior work has primarily examined scaling effects from an optimization perspective, this paper investigates their role in residual architectures through the lens of generalization theory. Specifically, we establish that wide residual networks (ResNets) with constant scaling factors become asymptotically unlearnable as depth increases. In contrast, when the scaling factor exhibits rapid depth-wise decay combined with early stopping, over-parameterized ResNets achieve minimax-optimal generalization rates. To establish this, we demonstrate that the generalization capability of wide ResNets can be approximated by kernel regression associated with the Neural Tangent Kernel (NTK). Our theoretical findings are validated through experiments on synthetic data and real-world classification tasks, including MNIST and CIFAR-100.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑