mlsanityData to 14 Oct 2026

Generalization by design: Shortcuts to Generalization in Deep Learning

arXiv 2107.02253 ↗First posted 5 Jul 2021cs.LG math.DG math.PRcited by 0

Not on the map. Why?

Abstract (arXiv)

We take a geometrical viewpoint and present a unifying view on supervised deep learning with the Bregman divergence loss function - this entails frequent classification and prediction tasks. Motivated by simulations we suggest that there is principally no implicit bias of vanilla stochastic gradient descent training of deep models towards "simpler" functions. Instead, we show that good generalization may be instigated by bounded spectral products over layers leading to a novel geometric regularizer. It is revealed that in deep enough models such a regularizer enables both, extreme accuracy and generalization, to be reached. We associate popular regularization techniques like weight decay, drop out, batch normalization, and early stopping with this perspective. Backed up by theory we further demonstrate that "generalization by design" is practically possible and that good generalization may be encoded into the structure of the network. We design two such easy-to-use structural regularizers that insert an additional textitgeneralization layer into a model architecture, one with a skip connection and another one with drop-out. We verify our theoretical results in experiments on various feedforward and convolutional architectures, including ResNets, and datasets (MNIST, CIFAR10, synthetic data). We believe this work opens up new avenues of research towards better generalizing architectures.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 46 references that are in the map. First posted on arXiv 5 Jul 2021.

Builds on

20 of 46 references.

References · 20

  1. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  3. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  4. 2016Densely Connected Convolutional Networkscited by 6,720
  5. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
  6. 2016Understanding deep learning requires rethinking generalizationcited by 1,674
  7. 2015Cyclical Learning Rates for Training Neural Networkscited by 629
  8. 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 268
  9. 2018A disciplined approach to neural network hyper-parameters: Part 1 -- learning rate, batch size, momentum, and weight decaycited by 226
  10. 2019SGD on Neural Networks Learns Functions of Increasing Complexitycited by 204
  11. 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 184
  12. 2019Gradient Descent with Early Stopping is Provably Robust to Label Noise for Overparameterized Neural Networkscited by 159
8 more
  1. 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
  2. 2017Generalization in Deep Learningcited by 146
  3. 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 142
  4. 2020The Implicit Regularization of Stochastic Gradient Flow for Least Squarescited by 59
  5. 2019Why ResNet Works? Residuals Generalizecited by 39
  6. 2019Towards Robust ResNet: A Small Step but A Giant Leapcited by 15
  7. 2021SGD Implicitly Regularizes Generalization Errorcited by 10
  8. 2018Residual Networks as Geodesic Flows of Diffeomorphismscited by 2

Built on by

Not cited yet.