A Two-Phase Perspective on Deep Learning Dynamics
By: Robert de Mello Koch, Animik Ghosh
Potential Business Impact:
Helps computers learn better by forgetting some things.
We propose that learning in deep neural networks proceeds in two phases: a rapid curve fitting phase followed by a slower compression or coarse graining phase. This view is supported by the shared temporal structure of three phenomena: grokking, double descent and the information bottleneck, all of which exhibit a delayed onset of generalization well after training error reaches zero. We empirically show that the associated timescales align in two rather different settings. Mutual information between hidden layers and input data emerges as a natural progress measure, complementing circuit-based metrics such as local complexity and the linear mapping number. We argue that the second phase is not actively optimized by standard training algorithms and may be unnecessarily prolonged. Drawing on an analogy with the renormalization group, we suggest that this compression phase reflects a principled form of forgetting, critical for generalization.
Similar Papers
Explaining Grokking and Information Bottleneck through Neural Collapse Emergence
Machine Learning (CS)
Teaches computers to learn better and faster.
A dynamic view of some anomalous phenomena in SGD
Optimization and Control
Helps computers learn better by finding hidden patterns.
Phase Transitions between Accuracy Regimes in L2 regularized Deep Neural Networks
Machine Learning (CS)
Helps computers learn better by avoiding bad learning habits.