Do Deep Networks Forget Initialization? A Forgetting-Time View of Practical Inductive Bias
Not on the map. Why?
Abstract (arXiv)
Randomly initialized neural networks induce a prior over functions, but the predictor used in practice is produced only after training. We ask how much of this initial bias survives the training pipeline. To make the question measurable, we introduce initialization memory: the dependence of the validation-selected predictor on the scale of the random initialization. We perform controlled CIFAR-10 experiments on ResNets where initialization memory already sharply separates training regimes. Low-learning-rate SGD can interpolate while still remembering its initialization: on ResNet-9 with batch size b=128, test accuracy varies by 26.5 percentage points across initialization scales despite ≥99.5% training accuracy. This is not undertraining: extending the same low-learning-rate regime to 5,000 epochs leaves the spread essentially unchanged. In contrast, Adam-family methods largely erase the dependence. SGD can also be made to forget when larger learning rates are paired with explicit L2 norm control. We interpret these findings in terms of the time scale of forgetting: gradient-flow-like dynamics can preserve initialization memory, whereas stochastic finite-step effects, explicit norm decay, and adaptive preconditioning erase it on scales governed by the size of explicit or implicit regularization. The practical inductive bias of a trained network is therefore not the architectural prior alone, but the architectural prior after being filtered by the forgetting dynamics of the training pipeline; and the same regularizers that improve generalization are precisely those that erase memory of initialization.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 84 of its 113 references that are in the map. First posted on arXiv 27 May 2026.
Builds on
84 of 113 references.
References · 84
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2018On the Spectral Bias of Neural Networkscited by 1,341
- 2017A Closer Look at Memorization in Deep Networkscited by 942
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
- 2017Spectrally-normalized margin bounds for neural networkscited by 686
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
38 more
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2017Deep Neural Networks as Gaussian Processescited by 622
- 2017Exploring Generalization in Deep Learningcited by 600
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 552
- 2019Benign Overfitting in Linear Regressioncited by 549
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 533
- 2017Sharp Minima Can Generalize For Deep Netscited by 466
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
- 2020Fine-Tuning Pretrained Language Models: Weight Initializations, Data Orders, and Early Stoppingcited by 384
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
- 2017A PAC-Bayesian Approach to Spectrally-Normalized Margin Bounds for Neural Networkscited by 314
- 2018Stronger generalization bounds for deep nets via a compression approachcited by 307
- 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 300
- 2015Norm-Based Capacity Control in Neural Networkscited by 299
- 2022Tensor Programs V: Tuning Large Neural Networks via Zero-Shot Hyperparameter Transfercited by 288
- 2017Train longer, generalize better: closing the generalization gap in large batch training of neural networkscited by 279
- 2023Physics of Language Models: Part 3.1, Knowledge Storage and Extractioncited by 270
- 2016Exponential expressivity in deep neural networks through transient chaoscited by 261
- 2018Implicit Bias of Gradient Descent on Linear Convolutional Networkscited by 251
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 250
- 2019SGD on Neural Networks Learns Functions of Increasing Complexitycited by 204
- 2016Deep Information Propagationcited by 195
- 2018Sensitivity and Generalization in Neural Networks: an Empirical Studycited by 189
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 184
- 2018Deep learning generalizes because the parameter-function map is biased towards simple functionscited by 183
- 2018Dynamical Isometry and a Mean Field Theory of CNNs: How to Train 10,000-Layer Vanilla Convolutional Neural Networkscited by 176
- 2019Complexity of Linear Regions in Deep Networkscited by 167
- 2019Deep ReLU Networks Have Surprisingly Few Activation Patternscited by 161
- 2019Tensor Programs I: Wide Feedforward or Recurrent Neural Networks of Any Architecture are Gaussian Processescited by 151
- 2020Implicit Gradient Regularizationcited by 146
- 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 142
- 2017Resurrecting the sigmoid in deep learning through dynamical isometry: theory and practicecited by 132
- 2018Non-Vacuous Generalization Bounds at the ImageNet Scale: A PAC-Bayesian Compression Approachcited by 126
- 2018Stochastic Modified Equations and Dynamics of Stochastic Gradient Algorithms I: Mathematical Foundationscited by 114
- 2018Predicting the Generalization Gap in Deep Networks with Margin Distributionscited by 113
- 2018How to Start Training: The Effect of Initialization and Architecturecited by 109
- 2022Self-Consistent Dynamical Field Theory of Kernel Evolution in Wide Neural Networkscited by 105
- 2018Which Neural Net Architectures Give Rise To Exploding and Vanishing Gradients?cited by 94
50 most cited of 84 All 84 (JSON)
Built on by
Not cited yet.