The Power of Preconditioning in Overparameterized Low-Rank Matrix Sensing
Not on the map. Why?
Abstract (arXiv)
We propose textsfScaledGD(λ), a preconditioned gradient descent method to tackle the low-rank matrix sensing problem when the true rank is unknown, and when the matrix is possibly ill-conditioned. Using overparametrized factor representations, textsfScaledGD(λ) starts from a small random initialization, and proceeds by gradient descent with a specific form of damped preconditioning to combat bad curvatures induced by overparameterization and ill-conditioning. At the expense of light computational overhead incurred by preconditioners, textsfScaledGD(λ) is remarkably robust to ill-conditioning compared to vanilla gradient descent (textsfGD) even with overprameterization. Specifically, we show that, under the Gaussian design, textsfScaledGD(λ) converges to the true low-rank matrix at a constant linear rate after a small number of iterations that scales only logarithmically with respect to the condition number and the problem dimension. This significantly improves over the convergence rate of vanilla textsfGD which suffers from a polynomial dependency on the condition number. Our work provides evidence on the power of preconditioning in accelerating the convergence without hurting generalization in overparameterized learning.
Key facts. Cited by 32 papers in the map (data through 7 Oct 2026). Builds on 32 of its 62 references that are in the map. First posted on arXiv 2 Feb 2023.
Builds on
32 of 62 references.
References · 32
- 2017Implicit Regularization in Matrix Factorizationcited by 318
- 2018Nonconvex Optimization Meets Low-Rank Matrix Factorization: An Overviewcited by 159
- 2017Theoretical insights into the optimization landscape of over-parameterized shallow neural networkscited by 128
- 2020Spectral Methods for Data Science: A Statistical Perspectivecited by 111
- 2015Low-rank Solutions of Linear Matrix Equations via Procrustes Flowcited by 97
- 2018Overparameterized Nonlinear Learning: Gradient Descent Takes the Shortest Path?cited by 87
- 2021Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructioncited by 77
- 2018Gradient Descent with Random Initialization: Fast Global Convergence for Nonconvex Phase Retrievalcited by 71
- 2020Accelerating Ill-Conditioned Low-Rank Matrix Estimation via Scaled Gradient Descentcited by 70
- 2017Implicit Regularization in Nonconvex Statistical Estimation: Gradient Descent Converges Linearly for Phase Retrieval, Matrix Completion, and Blind Deconvolutioncited by 63
- 2021Global Convergence of Gradient Descent for Asymmetric Low-Rank Matrix Factorizationcited by 44
- 2015A Convergent Gradient Descent Algorithm for Rank Minimization and Semidefinite Programming from Random Linear Measurementscited by 34
20 more
- 2022Algorithmic Regularization in Model-free Overparametrized Asymmetric Matrix Factorizationcited by 34
- 2021On the computational and statistical complexity of over-parameterized matrix sensingcited by 31
- 2020When Does Preconditioning Help or Hurt Generalization?cited by 30
- 2021Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Estimation from Incomplete Measurementscited by 28
- 2020Low-Rank Matrix Recovery with Scaled Subgradient Methods: Fast and Robust Convergence Without the Condition Numbercited by 27
- 2018Harnessing Structures in Big Data via Guaranteed Low-Rank Matrix Estimationcited by 25
- 2021Rank Overspecified Robust Matrix Recovery: Subgradient Method and Exact Recoverycited by 21
- 2022Global Convergence of Sub-gradient Method for Robust Matrix Recovery: Small Initialization, Noisy Measurements, and Over-parameterizationcited by 21
- 2022Preconditioned Gradient Descent for Overparameterized Nonconvex Burer--Monteiro Factorization with Global Optimality Certificationcited by 21
- 2018Subgradient Descent Learns Orthogonal Dictionariescited by 17
- 2018Nonconvex Matrix Factorization from Rank-One Measurementscited by 13
- 2022Improved Global Guarantees for the Nonconvex Burer--Monteiro Factorization via Rank Overparameterizationcited by 13
- 2022Randomly Initialized Alternating Least Squares: Fast Convergence for Matrix Sensingcited by 11
- 2022Fast and Provable Tensor Robust Principal Component Analysis via Scaled Gradient Descentcited by 11
- 2018Efficient Dictionary Learning with Gradient Descentcited by 10
- 2022Alternating minimization for generalized rank one matrix sensing: Sharp predictions from a random initializationcited by 9
- 2019Manifold Gradient Descent Solves Multi-Channel Sparse Blind Deconvolution Provably and Efficientlycited by 6
- 2021Deep Networks Provably Classify Data on Curvescited by 6
- 2019A Nonconvex Approach for Exact and Efficient Multichannel Sparse Blind Deconvolutioncited by 5
- 2022Rank-1 Matrix Completion with Gradient Descent and Small Random Initializationcited by 4
Built on by
Citing papers per year: 2023: 8; 2024: 9; 2025: 8; 2026: 7.
32 papers.
Sort:
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muoncited by 28
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initializationcited by 18
- 2024On the Crucial Role of Initialization for Matrix Factorizationcited by 15
- 2023Canonical Factors for Hybrid Neural Fieldscited by 8
- 2024Guaranteed Nonconvex Factorization Approach for Tensor Train Recoverycited by 8
- 2025AltLoRA: Towards Better Gradient Approximation in Low-Rank Adaptation with Alternating Projectionscited by 8
- 2023Deflated HeteroPCA: Overcoming the curse of ill-conditioning in heteroskedastic PCAcited by 7
- 2024Low-Tubal-Rank Tensor Recovery via Factorized Gradient Descentcited by 7
- 2023Provably Accelerating Ill-Conditioned Low-rank Estimation via Scaled Gradient Descent, Even with Overparameterizationcited by 6
- 2023Fast and Accurate Estimation of Low-Rank Matrices from Noisy Measurements via Preconditioned Non-Convex Gradient Descentcited by 5
- 2024Guarantees of a Preconditioned Subgradient Algorithm for Overparameterized Asymmetric Low-rank Matrix Recoverycited by 4
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descentcited by 4
20 more
- 2024Non-convex matrix sensing: Breaking the quadratic rank barrier in the sample complexitycited by 3
- 2024Provable Acceleration of Nesterov's Accelerated Gradient for Rectangular Matrix Factorization and Linear Neural Networkscited by 3
- 2025On the Benefits of Weight Normalization for Overparameterized Matrix Sensingcited by 3
- 2023Can Learning Be Explained By Local Optimality In Robust Low-rank Matrix Recovery?cited by 2
- 2023Implicit Regularization Makes Overparameterized Asymmetric Matrix Sensing Robust to Perturbationscited by 2
- 2024Improving Implicit Regularization of SGD with Preconditioning for Least Square Problemscited by 2
- 2024Robust Low-rank Tensor Train Recoverycited by 2
- 2024BLAST: Block-Level Adaptive Structured Matrices for Efficient Deep Neural Network Inferencecited by 2
- 2026Low-Rank Adaptation Redux for Large Modelscited by 2
- 2025Faster Than SVD, Smarter Than SGD: The OPLoRA Alternating Updatecited by 1
- 2025Q3R: Quadratic Reweighted Rank Regularizer for Effective Low-Rank Trainingcited by 1
- 2023Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensingcited by 0
- 2025k-SVD with Gradient Descentcited by 0
- 2025RGNMR: A Gauss-Newton method for robust matrix completion with theoretical guaranteescited by 0
- 2025Sharper Convergence Rates for Nonconvex Optimisation via Reduction Mappingscited by 0
- 2026Beyond SGD, Without SVD: Proximal Subspace Iteration LoRA with Diagonal Fractional K-FACcited by 0
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Ratecited by 0
- 2026Scaled Gradient Descent for Ill-Conditioned Low-Rank Matrix Recovery with Optimal Sampling Complexitycited by 0
- 2026Direction-Magnitude Decomposition for Low-Rank Matrix Optimization: Faster Convergence and Saddle-to-saddle Dynamicscited by 0
- 2026Reassessing Muon for Matrix Factorizationcited by 0
- 2026Reassessing Muon for Matrix Factorization14 Jul 2026
- 2026Direction-Magnitude Decomposition for Low-Rank Matrix Optimization: Faster Convergence and Saddle-to-saddle Dynamics30 Jun 2026
- 2026Low-Rank Adaptation Redux for Large Models23 Apr 2026
- 2026Scaled Gradient Descent for Ill-Conditioned Low-Rank Matrix Recovery with Optimal Sampling Complexity31 Mar 2026
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate22 Mar 2026
- 2026Beyond SGD, Without SVD: Proximal Subspace Iteration LoRA with Diagonal Fractional K-FAC18 Feb 2026
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muon20 Jan 2026
- 2025Q3R: Quadratic Reweighted Rank Regularizer for Effective Low-Rank Training6 Nov 2025
- 2025On the Benefits of Weight Normalization for Overparameterized Matrix Sensing1 Oct 2025
- 2025Faster Than SVD, Smarter Than SGD: The OPLoRA Alternating Update24 Sep 2025
- 2025Sharper Convergence Rates for Nonconvex Optimisation via Reduction Mappings10 Jun 2025
- 2025RGNMR: A Gauss-Newton method for robust matrix completion with theoretical guarantees19 May 2025
20 more
- 2025AltLoRA: Towards Better Gradient Approximation in Low-Rank Adaptation with Alternating Projections18 May 2025
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descent1 Feb 2025
- 2025k-SVD with Gradient Descent1 Feb 2025
- 2024BLAST: Block-Level Adaptive Structured Matrices for Efficient Deep Neural Network Inference28 Oct 2024
- 2024On the Crucial Role of Initialization for Matrix Factorization24 Oct 2024
- 2024Guarantees of a Preconditioned Subgradient Algorithm for Overparameterized Asymmetric Low-rank Matrix Recovery22 Oct 2024
- 2024Robust Low-rank Tensor Train Recovery19 Oct 2024
- 2024Provable Acceleration of Nesterov's Accelerated Gradient for Rectangular Matrix Factorization and Linear Neural Networks12 Oct 2024
- 2024Non-convex matrix sensing: Breaking the quadratic rank barrier in the sample complexity20 Aug 2024
- 2024Improving Implicit Regularization of SGD with Preconditioning for Least Square Problems13 Mar 2024
- 2024Low-Tubal-Rank Tensor Recovery via Factorized Gradient Descent22 Jan 2024
- 2024Guaranteed Nonconvex Factorization Approach for Tensor Train Recovery5 Jan 2024
- 2023Provably Accelerating Ill-Conditioned Low-rank Estimation via Scaled Gradient Descent, Even with Overparameterization9 Oct 2023
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initialization3 Oct 2023
- 2023Implicit Regularization Makes Overparameterized Asymmetric Matrix Sensing Robust to Perturbations4 Sep 2023
- 2023Canonical Factors for Hybrid Neural Fields29 Aug 2023
- 2023Fast and Accurate Estimation of Low-Rank Matrices from Noisy Measurements via Preconditioned Non-Convex Gradient Descent26 May 2023
- 2023Implicit Balancing and Regularization: Generalization and Convergence Guarantees for Overparameterized Asymmetric Matrix Sensing24 Mar 2023
- 2023Deflated HeteroPCA: Overcoming the curse of ill-conditioning in heteroskedastic PCA10 Mar 2023
- 2023Can Learning Be Explained By Local Optimality In Robust Low-rank Matrix Recovery?21 Feb 2023
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muoncited by 28
- 2023How Over-Parameterization Slows Down Gradient Descent in Matrix Sensing: The Curses of Symmetry and Initializationcited by 18
- 2024On the Crucial Role of Initialization for Matrix Factorizationcited by 15
- 2023Canonical Factors for Hybrid Neural Fieldscited by 8
- 2024Guaranteed Nonconvex Factorization Approach for Tensor Train Recoverycited by 8
- 2025AltLoRA: Towards Better Gradient Approximation in Low-Rank Adaptation with Alternating Projectionscited by 8
- 2023Deflated HeteroPCA: Overcoming the curse of ill-conditioning in heteroskedastic PCAcited by 7
- 2024Low-Tubal-Rank Tensor Recovery via Factorized Gradient Descentcited by 7
- 2023Provably Accelerating Ill-Conditioned Low-rank Estimation via Scaled Gradient Descent, Even with Overparameterizationcited by 6
- 2023Fast and Accurate Estimation of Low-Rank Matrices from Noisy Measurements via Preconditioned Non-Convex Gradient Descentcited by 5
- 2024Guarantees of a Preconditioned Subgradient Algorithm for Overparameterized Asymmetric Low-rank Matrix Recoverycited by 4
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descentcited by 4
12 most cited of 32 All 32 (JSON)
- 2026Reassessing Muon for Matrix Factorization14 Jul 2026
- 2026Direction-Magnitude Decomposition for Low-Rank Matrix Optimization: Faster Convergence and Saddle-to-saddle Dynamics30 Jun 2026
- 2026Low-Rank Adaptation Redux for Large Models23 Apr 2026
- 2026Scaled Gradient Descent for Ill-Conditioned Low-Rank Matrix Recovery with Optimal Sampling Complexity31 Mar 2026
- 2026Gradient Descent with Projection Finds Over-Parameterized Neural Networks for Learning Low-Degree Polynomials with Nearly Minimax Optimal Rate22 Mar 2026
- 2026Beyond SGD, Without SVD: Proximal Subspace Iteration LoRA with Diagonal Fractional K-FAC18 Feb 2026
- 2026Preconditioning Benefits of Spectral Orthogonalization in Muon20 Jan 2026
- 2025Q3R: Quadratic Reweighted Rank Regularizer for Effective Low-Rank Training6 Nov 2025
- 2025On the Benefits of Weight Normalization for Overparameterized Matrix Sensing1 Oct 2025
- 2025Faster Than SVD, Smarter Than SGD: The OPLoRA Alternating Update24 Sep 2025
- 2025Sharper Convergence Rates for Nonconvex Optimisation via Reduction Mappings10 Jun 2025
- 2025RGNMR: A Gauss-Newton method for robust matrix completion with theoretical guarantees19 May 2025
12 newest of 32 All 32 (JSON)