Abstract:We analyze the dynamics of finite width effects in wide but finite feature learning neural networks. Starting from a dynamical mean field theory description of infinite width deep neural network kernel and prediction dynamics, we provide a characterization of the O ( 1 / width ) fluctuations of the dynamical mean field theory order parameters over random initializations of the network weights. Our results, while perturbative in width, unlike prior analyses, are non-perturbative in the strength of feature learning. We find that once the mean field/μP parameterization is adopted, the leading finite size effect on the dynamics is to introduce initialization variance in the predictions and feature kernels of the networks. In the lazy limit of network training, all kernels are random but static in time and the prediction variance has a universal form. However, in the rich, feature learning regime, the fluctuations of the kernels and predictions are dynamically coupled with a variance that can be computed self-consistently. In two layer networks, we show how feature learning can dynamically reduce the variance of the final tangent kernel and final network predictions. We also show how initialization variance can slow down online learning in wide but finite networks. In deeper networks, kernel variance can dramatically accumulate through subsequent layers at large feature learning strengths, but feature learning continues to improve the signal-to-noise ratio of the feature kernels. In discrete time, we demonstrate that large learning rate phenomena such as edge of stability effects can be well captured by infinite width dynamics and that initialization variance can decrease dynamically. For convolutional neural networks trained on CIFAR-10, we empirically find significant corrections to both the bias and variance of network dynamics due to finite width.

Dynamical and stationary properties of on-line learning from finite training sets

Mean-field theory of learning: from dynamics to statics

Dynamics of Gradient-Based Learning and Applications to Hyperparameter Estimation

Mean-square convergence analysis of ADALINE training with minimum error entropy criterion.

Learning Time-Scales in Two-Layers Neural Networks

Machine learning in and out of equilibrium

Imitating Deep Learning Dynamics via Locally Elastic Stochastic Differential Equations

Multi-scale Feature Learning Dynamics: Insights for Double Descent

The Limiting Dynamics of SGD: Modified Loss, Phase Space Oscillations, and Anomalous Diffusion

Finite-Time Error Bounds For Linear Stochastic Approximation and TD Learning

Statistical Mechanics of On-line Learning When a Moving Teacher Goes around an Unlearnable True Teacher

Training Dynamics of Nonlinear Contrastive Learning Model in the High Dimensional Limit

Stationary Behavior of Constant Stepsize SGD Type Algorithms: An Asymptotic Characterization

A General Framework for Analyzing Stochastic Dynamics in Learning Algorithms

Understanding Short-Range Memory Effects in Deep Neural Networks

Dynamics of Finite Width Kernel and Prediction Fluctuations in Mean Field Neural Networks

The High Line: Exact Risk and Learning Rate Curves of Stochastic Adaptive Learning Rate Algorithms

Dynamics of finite width Kernel and prediction fluctuations in mean field neural networks *

Fast and Strong Convergence of Online Learning Algorithms.

Reconciling Modern Deep Learning with Traditional Optimization Analyses: The Intrinsic Learning Rate