Abstract:It is well known that neural networks with many more parameters than training examples do not overfit. Implicit regularization phenomena, which are still not well understood, occur during optimization and 'good' networks are favored. Thus the number of parameters is not an adequate measure of complexity if we do not consider all possible networks but only the 'good' ones. To better understand which networks are favored during optimization, we study the geometry of the output set as parameters vary. When the inputs are fixed, we prove that the dimension of this set changes and that the local dimension, called batch functional dimension, is almost surely determined by the activation patterns in the hidden layers. We prove that the batch functional dimension is invariant to the symmetries of the network parameterization: neuron permutations and positive rescalings. Empirically, we establish that the batch functional dimension decreases during optimization. As a consequence, optimization leads to parameters with low batch functional dimensions. We call this phenomenon geometry-induced implicit regularization.The batch functional dimension depends on both the network parameters and inputs. To understand the impact of the inputs, we study, for fixed parameters, the largest attainable batch functional dimension when the inputs vary. We prove that this quantity, called computable full functional dimension, is also invariant to the symmetries of the network's parameterization, and is determined by the achievable activation patterns. We also provide a sampling theorem, showing a fast convergence of the estimation of the computable full functional dimension for a random input of increasing size. Empirically we find that the computable full functional dimension remains close to the number of parameters, which is related to the notion of local identifiability. This differs from the observed values for the batch functional dimension computed on training inputs and test inputs. The latter are influenced by geometry-induced implicit regularization.

On the Effect of Initialization: The Scaling Path of 2-Layer Neural Networks

Initialization Matters: On the Benign Overfitting of Two-Layer ReLU CNN with Fully Trainable Layers

How to Initialize your Network? Robust Initialization for WeightNorm & ResNets

Compelling ReLU Networks to Exhibit Exponentially Many Linear Regions at Initialization and During Training

Scaling ResNets in the Large-depth Regime

Deconstructing the Goldilocks Zone of Neural Network Initialization

On the Convergence of Gradient Descent for Large Learning Rates

Convergence Analysis of Two-layer Neural Networks with ReLU Activation

How Implicit Regularization of ReLU Neural Networks Characterizes the Learned Function -- Part I: the 1-D Case of Two Layers with Random First Layer

Multilevel Initialization for Layer-Parallel Deep Neural Network Training

Deep linear networks for regression are implicitly regularized towards flat minima

Provable Benefit of Orthogonal Initialization in Optimizing Deep Linear Networks

A Sober Look at Neural Network Initializations

Principles for Initialization and Architecture Selection in Graph Neural Networks with ReLU Activations

From Activation to Initialization: Scaling Insights for Optimizing Neural Fields

Geometry-induced Implicit Regularization in Deep ReLU Neural Networks

Regularization Matters: Generalization and Optimization of Neural Nets v.s. their Induced Kernel

On weight initialization in deep neural networks

Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstruction

Towards Understanding the Condensation of Two-layer Neural Networks at Initial Training.