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.

Kolmogorov Width Decay and Poor Approximators in Machine Learning: Shallow Neural Networks, Random Feature Models and Neural Tangent Kernels

Stochastic Neural Networks with Infinite Width are Deterministic

Linear approximability of two-layer neural networks: A comprehensive analysis based on spectral decay

Sharp Bounds on the Approximation Rates, Metric Entropy, and $n$-widths of Shallow Neural Networks

Efficient optimal Kolmogorov approximation of random variables

Dynamically Stable Infinite-Width Limits of Neural Classifiers

Wide stable neural networks: Sample regularity, functional convergence and Bayesian inverse problems

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

Neural networks: deep, shallow, or in between?

The Kolmogorov N-width for linear transport: Exact representation and the influence of the data

A Lipschitz spaces view of infinitely wide shallow neural networks

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

Neural networks in non-metric spaces

Universal approximation with complex-valued deep narrow neural networks

Deformed semicircle law and concentration of nonlinear random matrices for ultra-wide neural networks

Kernel regression, minimax rates and effective dimensionality: beyond the regular case

Feature Learning in Infinite-Width Neural Networks

On the infinite width limit of neural networks with a standard parameterization

New advances in universal approximation with neural networks of minimal width

Critical feature learning in deep neural networks

Infinite-dimensional generalization of Kolmogorov widths