Deep Learning Through A Telescoping Lens: A Simple Model Provides Empirical Insights On Grokking, Gradient Boosting & Beyond
Not on the map. Why?
Abstract (arXiv)
Deep learning sometimes appears to work in unexpected ways. In pursuit of a deeper understanding of its surprising behaviors, we investigate the utility of a simple yet accurate model of a trained neural network consisting of a sequence of first-order approximations telescoping out into a single empirically operational tool for practical analysis. Across three case studies, we illustrate how it can be applied to derive new empirical insights on a diverse range of prominent phenomena in the literature -- including double descent, grokking, linear mode connectivity, and the challenges of applying deep learning on tabular data -- highlighting that this model allows us to construct and extract metrics that help predict and understand the a priori unexpected performance of neural networks. We also demonstrate that this model presents a pedagogical formalism allowing us to isolate components of the training process even in complex contemporary settings, providing a lens to reason about the effects of design choices such as architecture & optimization strategy, and reveals surprising parallels between neural network learning and gradient boosting.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 77 of its 103 references that are in the map. First posted on arXiv 31 Oct 2024.
Builds on
77 of 103 references.
References · 77
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,396
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 927
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 738
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2019Deep Double Descent: Where Bigger Models and More Data Hurtcited by 672
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 637
38 more
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 574
- 2019Benign Overfitting in Linear Regressioncited by 549
- 2014OpenML: networked science in machine learningcited by 526
- 2019Surprises in High-Dimensional Ridgeless Least Squares Interpolationcited by 513
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 471
- 2019Deep Ensembles: A Loss Landscape Perspectivecited by 443
- 2020What is being transferred in transfer learning?cited by 368
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 330
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 291
- 2017High-dimensional dynamics of generalization error in neural networkscited by 241
- 2022Towards Understanding Grokking: An Effective Theory of Representation Learningcited by 230
- 2023When Do Neural Nets Outperform Boosted Trees on Tabular Data?cited by 223
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 221
- 2019Model Fusion via Optimal Transportcited by 219
- 2019Two models of double descent for weak featurescited by 195
- 2022Patching open-vocabulary models by interpolating weightscited by 190
- 2018To understand deep learning we need to understand kernel learningcited by 187
- 2020Deep learning versus kernel learning: an empirical study of loss landscape geometry and the time evolution of the Neural Tangent Kernelcited by 187
- 2019On the Inductive Bias of Neural Tangent Kernelscited by 177
- 2020Finite Versus Infinite Neural Networks: an Empirical Studycited by 157
- 2017The Power of Interpolation: Understanding the Effectiveness of SGD in Modern Over-parametrized Learningcited by 150
- 2022Diverse Weight Averaging for Out-of-Distribution Generalizationcited by 148
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
- 2021Fit without fear: remarkable mathematical phenomena of deep learning through the prism of interpolationcited by 132
- 2020On the linearity of large non-linear models: when and why the tangent kernel is constantcited by 129
- 2023Explaining grokking through circuit efficiencycited by 111
- 2022Fusing finetuned models for better pretrainingcited by 110
- 2020Optimal Regularization Can Mitigate Double Descentcited by 109
- 2023Grokking as the Transition from Lazy to Rich Training Dynamicscited by 108
- 2016Topology and Geometry of Half-Rectified Network Optimizationcited by 102
- 2020Double Trouble in Double Descent : Bias and Variance(s) in the Lazy Regimecited by 95
- 2022The Slingshot Mechanism: An Empirical Study of Adaptive Optimizers and the Grokking Phenomenoncited by 94
- 2019Limitations of Lazy Training of Two-layers Neural Networkscited by 87
- 2020Finite-sample Analysis of Interpolating Linear Classifiers in the Overparameterized Regimecited by 81
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2020Understanding Double Descent Requires a Fine-Grained Bias-Variance Decompositioncited by 76
50 most cited of 77 All 77 (JSON)
Built on by
Citing papers per year: 2025: 6; 2026: 3.
9 papers.
- 2025Representation Learning for Tabular Data: A Comprehensive Surveycited by 36
- 2025Position: Solve Layerwise Linear Models First to Understand Neural Dynamical Phenomena (Neural Collapse, Emergence, Lazy/Rich Regime, and Grokking)cited by 9
- 2026To Grok Grokking: Provable Grokking in Ridge Regressioncited by 8
- 2025Mechanistic Insights into Grokking from the Embedding Layercited by 6
- 2025Not All Explanations for Deep Learning Phenomena Are Equally Valuablecited by 3
- 2025Is Supervised Learning Really That Different from Unsupervised?cited by 0
- 2025Random at First, Fast at Last: NTK-Guided Fourier Pre-Processing for Tabular DLcited by 0
- 2026A Rigorous, Tractable Measure of Model Complexitycited by 0
- 2026A Spectral Theory of Grokking: Weight Decay induces Feature Learningcited by 0
- 2026A Spectral Theory of Grokking: Weight Decay induces Feature Learning22 Sep 2026
- 2026A Rigorous, Tractable Measure of Model Complexity20 May 2026
- 2026To Grok Grokking: Provable Grokking in Ridge Regression27 Jan 2026
- 2025Not All Explanations for Deep Learning Phenomena Are Equally Valuable29 Jun 2025
- 2025Random at First, Fast at Last: NTK-Guided Fourier Pre-Processing for Tabular DL3 Jun 2025
- 2025Mechanistic Insights into Grokking from the Embedding Layer21 May 2025
- 2025Is Supervised Learning Really That Different from Unsupervised?16 May 2025
- 2025Representation Learning for Tabular Data: A Comprehensive Survey17 Apr 2025
- 2025Position: Solve Layerwise Linear Models First to Understand Neural Dynamical Phenomena (Neural Collapse, Emergence, Lazy/Rich Regime, and Grokking)28 Feb 2025