Generalization Guarantees for Neural Networks via Harnessing the Low-rank Structure of the Jacobian
Not on the map. Why?
Abstract (arXiv)
Modern neural network architectures often generalize well despite containing many more parameters than the size of the training dataset. This paper explores the generalization capabilities of neural networks trained via gradient descent. We develop a data-dependent optimization and generalization theory which leverages the low-rank structure of the Jacobian matrix associated with the network. Our results help demystify why training and generalization is easier on clean and structured datasets and harder on noisy and unstructured datasets as well as how the network size affects the evolution of the train and test errors during training. Specifically, we use a control knob to split the Jacobian spectum into "information" and "nuisance" spaces associated with the large and small singular values. We show that over the information space learning is fast and one can quickly train a model with zero training loss that can also generalize well. Over the nuisance space training is slower and early stopping can help with generalization at the expense of some bias. We also show that the overall generalization capability of the network is controlled by how well the label vector is aligned with the information space. A key feature of our results is that even constant width neural nets can provably generalize for sufficiently nice datasets. We conduct various numerical experiments on deep networks that corroborate our theoretical findings and demonstrate that: (i) the Jacobian of typical neural networks exhibit low-rank structure with a few large singular values and many small ones leading to a low-dimensional information space, (ii) over the information space learning is fast and most of the label vector falls on this space, and (iii) label noise falls on the nuisance space and impedes optimization/generalization.
Key facts. Cited by 67 papers in the map (data through 7 Oct 2026). First posted on arXiv 12 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 15; 2022: 11; 2023: 7; 2024: 17; 2025: 8; 2026: 9.
67 papers.
Sort:
By area:
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2022Robust Training under Label Noise by Over-parameterizationcited by 93
- 2022Investigating Why Contrastive Learning Benefits Robustness Against Label Noisecited by 46
- 2025Parameter-Efficient Fine-Tuning for Foundation Modelscited by 43
- 2021Spectral Bias in Practice: The Role of Function Frequency in Generalizationcited by 36
- 2022Spectral Evolution and Invariance in Linear-width Neural Networkscited by 30
- 2022Parameter-Efficient Sparsity for Large Language Models Fine-Tuningcited by 27
- 2023Understanding and Mitigating the Label Noise in Pre-training on Downstream Taskscited by 27
- 2024Compressible Dynamics in Deep Overparameterized Low-Rank Learning & Adaptationcited by 26
- 2022Characterizing the Spectrum of the NTK via a Power Series Expansioncited by 22
- 2021Kernelized Heterogeneous Risk Minimizationcited by 21
38 more
- 2022Towards Sample-efficient Overparameterized Meta-learningcited by 21
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Widthcited by 21
- 2021Feature Learning and Signal Propagation in Deep Neural Networkscited by 18
- 2025LoR-VP: Low-Rank Visual Prompting for Efficient Vision Model Adaptationcited by 17
- 2024On Catastrophic Inheritance of Large Foundation Modelscited by 15
- 2024Exploring the Frontiers of Softmax: Provable Optimization, Applications in Diffusion Model, and Beyondcited by 15
- 2021Understanding Overparameterization in Generative Adversarial Networkscited by 14
- 2021Generalization Guarantees for Neural Architecture Search with Train-Validation Splitcited by 14
- 2024Differential Privacy Mechanisms in Neural Tangent Kernel Regressioncited by 14
- 2023Provable Guarantees for Neural Networks via Gradient Feature Learningcited by 13
- 2023Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimizationcited by 13
- 2021Relative stability toward diffeomorphisms indicates performance in deep netscited by 12
- 2022Spectral Bias Outside the Training Set for Deep Networks in the Kernel Regimecited by 11
- 2024Unifying Low Dimensional Spectra in Deep Learningcited by 11
- 2021Provable Super-Convergence with a Large Cyclical Learning Ratecited by 10
- 2021Edge of chaos as a guiding principle for modern neural network trainingcited by 10
- 2022Implicit Bias of MSE Gradient Optimization in Underparameterized Neural Networkscited by 9
- 2024SeTAR: Out-of-Distribution Detection with Selective Low-Rank Approximationcited by 9
- 2025On the Convergence Rate of LoRA Gradient Descentcited by 9
- 2021DSEE: Dually Sparsity-embedded Efficient Tuning of Pre-trained Language Modelscited by 8
- 2024Impact of Noisy Supervision in Foundation Model Learningcited by 8
- 2026On the Non-Identifiability of Steering Vectors in Large Language Modelscited by 8
- 2021Unsupervised Embedding Adaptation via Early-Stage Feature Reconstruction for Few-Shot Classificationcited by 7
- 2024Solving Inverse Problems via Diffusion Optimal Controlcited by 7
- 2021Deep Networks Provably Classify Data on Curvescited by 6
- 2021Representation Alignment in Neural Networkscited by 6
- 2024Weight Copy and Low-Rank Adaptation for Few-Shot Distillation of Vision Transformerscited by 6
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuningcited by 6
- 2021Towards Understanding Generalization via Decomposing Excess Risk Dynamicscited by 5
- 2024Structured Unrestricted-Rank Matrices for Parameter Efficient Fine-tuningcited by 5
- 2022Data-Efficient Augmentation for Training Neural Networkscited by 4
- 2024Revealing the Utilized Rank of Subspaces of Learning in Neural Networkscited by 4
- 2026Weight Decay Improves Language Model Plasticitycited by 4
- 2021On the Impact of Stable Ranks in Deep Netscited by 3
- 2023Few-shot Adaptation to Distribution Shifts By Mixing Source and Target Embeddingscited by 3
- 2024Train Faster, Perform Better: Modular Adaptive Training in Over-Parameterized Modelscited by 3
- 2024GRASP: Replace Redundant Layers with Adaptive Singular Parameters for Efficient Model Compressioncited by 3
- 2026CrispEdit: Low-Curvature Projections for Scalable Non-Destructive LLM Editingcited by 3
50 most cited of 67 All 67 (JSON)
- 2026Eigenvalues of the Hessian in Deep Learning: The Origin of Symmetry and Its Breaking7 Oct 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026Handling Feature Heterogeneity with Learnable Graph Patches16 Jun 2026
- 2026Curvature-Guided LoRA: Matching Full Fine-Tuning in Function Space31 Mar 2026
- 2026Deep Tabular Representation Corrector17 Mar 2026
- 2026CrispEdit: Low-Curvature Projections for Scalable Non-Destructive LLM Editing17 Feb 2026
- 2026Why Deep Jacobian Spectra Separate: Depth-Induced Scaling and Singular-Vector Alignment12 Feb 2026
- 2026Weight Decay Improves Language Model Plasticity11 Feb 2026
- 2026On the Non-Identifiability of Steering Vectors in Large Language Models6 Feb 2026
- 2025How Much Data Is Enough? Uniform Convergence Bounds for Generative & Vision-Language Models under Low-Dimensional Structure28 Dec 2025
- 2025On the Convergence Rate of LoRA Gradient Descent20 Dec 2025
- 2025Verifying Large Language Models' Reasoning Paths via Correlation Matrix Rank28 Oct 2025
38 more
- 2025Projective Kolmogorov Arnold Neural Networks (P-KANs): Entropy-Driven Functional Space Discovery for Interpretable Machine Learning24 Sep 2025
- 2025Quantized Rank Reduction: A Communications-Efficient Federated Learning Scheme for Network-Critical Applications15 Jul 2025
- 2025LoR-VP: Low-Rank Visual Prompting for Efficient Vision Model Adaptation2 Feb 2025
- 2025DeltaLLM: Compress LLMs with Low-Rank Deltas between Shared Weights30 Jan 2025
- 2025Parameter-Efficient Fine-Tuning for Foundation Models23 Jan 2025
- 2024GRASP: Replace Redundant Layers with Adaptive Singular Parameters for Efficient Model Compression31 Dec 2024
- 2024Towards a Statistical Understanding of Neural Networks: Beyond the Neural Tangent Kernel Theories25 Dec 2024
- 2024Solving Inverse Problems via Diffusion Optimal Control21 Dec 2024
- 2024Differential Privacy Mechanisms in Neural Tangent Kernel Regression18 Jul 2024
- 2024Revealing the Utilized Rank of Subspaces of Learning in Neural Networks5 Jul 2024
- 2024Increasing Model Capacity for Free: A Simple Strategy for Parameter Efficient Fine-tuning1 Jul 2024
- 2024Structured Unrestricted-Rank Matrices for Parameter Efficient Fine-tuning25 Jun 2024
- 2024SeTAR: Out-of-Distribution Detection with Selective Low-Rank Approximation18 Jun 2024
- 2024Compressible Dynamics in Deep Overparameterized Low-Rank Learning & Adaptation6 Jun 2024
- 2024Length independent generalization bounds for deep SSM architectures via Rademacher contraction and stability constraints30 May 2024
- 2024Train Faster, Perform Better: Modular Adaptive Training in Over-Parameterized Models13 May 2024
- 2024Exploring the Frontiers of Softmax: Provable Optimization, Applications in Diffusion Model, and Beyond6 May 2024
- 2024Weight Copy and Low-Rank Adaptation for Few-Shot Distillation of Vision Transformers14 Apr 2024
- 2024Unifying Low Dimensional Spectra in Deep Learning9 Apr 2024
- 2024AILS-NTUA at SemEval-2024 Task 9: Cracking Brain Teasers: Transformer Models for Lateral Thinking Puzzles1 Apr 2024
- 2024Impact of Noisy Supervision in Foundation Model Learning11 Mar 2024
- 2024On Catastrophic Inheritance of Large Foundation Models2 Feb 2024
- 2023Outliers with Opposing Signals Have an Outsized Effect on Neural Network Optimization7 Nov 2023
- 2023Optimization dependent generalization bound for ReLU networks based on sensitivity in the tangent bundle26 Oct 2023
- 2023Provable Guarantees for Neural Networks via Gradient Feature Learning19 Oct 2023
- 2023Learning A Disentangling Representation For PU Learning5 Oct 2023
- 2023Understanding and Mitigating the Label Noise in Pre-training on Downstream Tasks29 Sep 2023
- 2023Few-shot Adaptation to Distribution Shifts By Mixing Source and Target Embeddings23 May 2023
- 2023Generalization and Stability of Interpolating Neural Networks with Minimal Width18 Feb 2023
- 2022Label Alignment Regularization for Distribution Shift27 Nov 2022
- 2022Characterizing the Spectrum of the NTK via a Power Series Expansion15 Nov 2022
- 2022Spectral Evolution and Invariance in Linear-width Neural Networks11 Nov 2022
- 2022Data-Efficient Augmentation for Training Neural Networks15 Oct 2022
- 2022Neural Networks can Learn Representations with Gradient Descent30 Jun 2022
- 2022Spectral Bias Outside the Training Set for Deep Networks in the Kernel Regime6 Jun 2022
- 2022Parameter-Efficient Sparsity for Large Language Models Fine-Tuning23 May 2022
- 2022Robust Training under Label Noise by Over-parameterization28 Feb 2022
- 2022Investigating Why Contrastive Learning Benefits Robustness Against Label Noise29 Jan 2022
50 newest of 67 All 67 (JSON)
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
- 2022Neural Networks can Learn Representations with Gradient Descentcited by 161
- 2022Robust Training under Label Noise by Over-parameterizationcited by 93
- 2022Investigating Why Contrastive Learning Benefits Robustness Against Label Noisecited by 46
- 2025Parameter-Efficient Fine-Tuning for Foundation Modelscited by 43
- 2021Spectral Bias in Practice: The Role of Function Frequency in Generalizationcited by 36
- 2022Spectral Evolution and Invariance in Linear-width Neural Networkscited by 30
- 2022Parameter-Efficient Sparsity for Large Language Models Fine-Tuningcited by 27
- 2023Understanding and Mitigating the Label Noise in Pre-training on Downstream Taskscited by 27
- 2024Compressible Dynamics in Deep Overparameterized Low-Rank Learning & Adaptationcited by 26
- 2022Characterizing the Spectrum of the NTK via a Power Series Expansioncited by 22
- 2021Kernelized Heterogeneous Risk Minimizationcited by 21
12 most cited of 62 All 62 (JSON)
- 2026Eigenvalues of the Hessian in Deep Learning: The Origin of Symmetry and Its Breaking7 Oct 2026
- 2026A Data-dependent Early Stopping Rule using Rademacher Complexity with L1-norm25 Aug 2026
- 2026Handling Feature Heterogeneity with Learnable Graph Patches16 Jun 2026
- 2026Curvature-Guided LoRA: Matching Full Fine-Tuning in Function Space31 Mar 2026
- 2026Deep Tabular Representation Corrector17 Mar 2026
- 2026Why Deep Jacobian Spectra Separate: Depth-Induced Scaling and Singular-Vector Alignment12 Feb 2026
- 2026Weight Decay Improves Language Model Plasticity11 Feb 2026
- 2026On the Non-Identifiability of Steering Vectors in Large Language Models6 Feb 2026
- 2025On the Convergence Rate of LoRA Gradient Descent20 Dec 2025
- 2025Projective Kolmogorov Arnold Neural Networks (P-KANs): Entropy-Driven Functional Space Discovery for Interpretable Machine Learning24 Sep 2025
- 2025Quantized Rank Reduction: A Communications-Efficient Federated Learning Scheme for Network-Critical Applications15 Jul 2025
- 2025LoR-VP: Low-Rank Visual Prompting for Efficient Vision Model Adaptation2 Feb 2025
12 newest of 62 All 62 (JSON)