The Low-Rank Simplicity Bias in Deep Networks

Abstract

Modern deep neural networks are highly over-parameterized compared to the data on which they are trained, yet they often generalize remarkably well. A flurry of recent work has asked: why do deep networks not overfit to their training data? In this work, we make a series of empirical observations that investigate and extend the hypothesis that deeper networks are inductively biased to find solutions with lower effective rank embeddings. We conjecture that this bias exists because the volume of functions that maps to low effective rank embedding increases with depth. We show empirically that our claim holds true on finite width linear and non-linear models on practical learning paradigms and show that on natural data, these are often the solutions that generalize well. We then show that the simplicity bias exists at both initialization and after training and is resilient to hyper-parameters and learning methods. We further demonstrate how linear over-parameterization of deep non-linear models can be used to induce low-rank bias, improving generalization performance on CIFAR and ImageNet without changing the modeling capacity.

References in corpus (17)

• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Neural Architecture Search with Reinforcement Learning
• Scaling Laws for Neural Language Models
• Skip-Thought Vectors
• Deep Learning Scaling is Predictable, Empirically
• A Closer Look at Memorization in Deep Networks
• Spectral Norm Regularization for Improving the Generalizability of Deep Learning
• Qualitatively characterizing neural network optimization problems
• In Search of the Real Inductive Bias: On the Role of Implicit Regularization in Deep Learning
• The Pitfalls of Simplicity Bias in Neural Networks
• Implicit Regularization in Deep Learning May Not Be Explainable by Norms
• SGD on Neural Networks Learns Functions of Increasing Complexity
• Gradient Starvation: A Learning Proclivity in Neural Networks
• Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learning
• Increasing Depth Leads to U-Shaped Test Risk in Over-parameterized Convolutional Networks

Cited by in corpus (7)

• Machine Learning and Deep Learning -- A review for Ecologists
• Increasing Depth Leads to U-Shaped Test Risk in Over-parameterized Convolutional Networks
• Layer Folding: Neural Network Depth Reduction using Activation Linearization
• How Does Adversarial Fine-Tuning Benefit BERT?
• The Difficulty of Passive Learning in Deep Reinforcement Learning
• Implicit Greedy Rank Learning in Autoencoders via Overparameterized Linear Networks
• Generalization by design: Shortcuts to Generalization in Deep Learning