Fast low-rank estimation by projected gradient descent: General statistical and algorithmic guarantees
Not on the map. Why?
Abstract (arXiv)
Optimization problems with rank constraints arise in many applications, including matrix regression, structured PCA, matrix completion and matrix decomposition problems. An attractive heuristic for solving such problems is to factorize the low-rank matrix, and to run projected gradient descent on the nonconvex factorized optimization problem. The goal of this problem is to provide a general theoretical framework for understanding when such methods work well, and to characterize the nature of the resulting fixed point. We provide a simple set of conditions under which projected gradient descent, when given a suitable initialization, converges geometrically to a statistically useful solution. Our results are applicable even when the initial solution is outside any region of local convexity, and even when the problem is globally concave. Working in a non-asymptotic framework, we show that our conditions are satisfied for a wide range of concrete models, including matrix regression, structured PCA, matrix completion with real and quantized observations, matrix decomposition, and graph clustering problems. Simulation results show excellent agreement with the theoretical predictions.
Key facts. Cited by 68 papers in the map (data through 7 Oct 2026). First posted on arXiv 10 Sep 2015.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 17; 2022: 14; 2023: 12; 2024: 12; 2025: 11; 2026: 2.
68 papers.
Sort:
By area:
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2021Large Learning Rate Tames Homogeneity: Convergence and Balancing Effectcited by 48
- 2024Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?cited by 39
- 2023PDPP: Projected Diffusion for Procedure Planning in Instructional Videoscited by 36
- 2021On the computational and statistical complexity of over-parameterized matrix sensingcited by 31
- 2021Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Estimation from Incomplete Measurementscited by 28
- 2021Mind Mappings: Enabling Efficient Algorithm-Accelerator Mapping Space Searchcited by 24
- 2022Convergence of gradient descent for deep neural networkscited by 24
- 2021Inference for Heteroskedastic PCA with Missing Datacited by 21
- 2022Global Convergence of Sub-gradient Method for Robust Matrix Recovery: Small Initialization, Noisy Measurements, and Over-parameterizationcited by 21
- 2024VeLoRA: Memory Efficient Training using Rank-1 Sub-Token Projectionscited by 18
- 2021Noisy Gradient Descent Converges to Flat Minima for Nonconvex Matrix Factorizationcited by 15
38 more
- 2021Sparse Plus Low Rank Matrix Decomposition: A Discrete Optimization Approachcited by 14
- 2024Propulsion: Steering LLM with Tiny Fine-Tuningcited by 14
- 2023Subspace based Federated Unlearningcited by 13
- 2022Federated Learning and Meta Learning: Approaches, Applications, and Directionscited by 12
- 2025Preconditioned Gradient Descent for Over-Parameterized Nonconvex Matrix Factorizationcited by 12
- 2021Implicit Regularization in Matrix Sensing via Mirror Descentcited by 11
- 2024Gradient Weight-normalized Low-rank Projection for Efficient LLM Trainingcited by 11
- 2021Multi-source Learning via Completion of Block-wise Overlapping Noisy Matricescited by 10
- 2024Parameter-Efficient Fine-Tuning of Large Language Models using Semantic Knowledge Tuningcited by 10
- 2023Approximate message passing from random initialization with applications to mathbbZ2 synchronizationcited by 9
- 2022A Validation Approach to Over-parameterized Matrix and Image Recoverycited by 8
- 2024COAP: Memory-Efficient Training with Correlation-Aware Gradient Projectioncited by 8
- 2024Low-Tubal-Rank Tensor Recovery via Factorized Gradient Descentcited by 7
- 2022Accelerating SGD for Highly Ill-Conditioned Huge-Scale Online Matrix Completioncited by 6
- 2022Towards a Theoretical Foundation of Policy Optimization for Learning Control Policiescited by 6
- 2023Provably Accelerating Ill-Conditioned Low-rank Estimation via Scaled Gradient Descent, Even with Overparameterizationcited by 6
- 2024Convergence of Gradient Descent with Small Initialization for Unregularized Matrix Completioncited by 6
- 2021Fast quantum state reconstruction via accelerated non-convex programmingcited by 5
- 2022Model-free Neural Lyapunov Control for Safe Robot Navigationcited by 5
- 2023Fast and Accurate Estimation of Low-Rank Matrices from Noisy Measurements via Preconditioned Non-Convex Gradient Descentcited by 5
- 2025Riemannian Optimization on Relaxed Indicator Matrix Manifoldcited by 5
- 2022Robust Matrix Completion with Heavy-tailed Noisecited by 4
- 2022Rank-1 Matrix Completion with Gradient Descent and Small Random Initializationcited by 4
- 2023Computationally Efficient and Statistically Optimal Robust High-Dimensional Linear Regressioncited by 4
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descentcited by 4
- 2021Riemannian Perspective on Matrix Factorizationcited by 3
- 2021Exact Linear Convergence Rate Analysis for Low-Rank Symmetric Matrix Completion via Gradient Descentcited by 3
- 2021Nonconvex Factorization and Manifold Formulations are Almost Equivalent in Low-rank Matrix Optimizationcited by 3
- 2022Nonconvex Matrix Factorization is Geodesically Convex: Global Landscape Analysis for Fixed-rank Matrix Optimization From a Riemannian Perspectivecited by 3
- 2023Online Tensor Learning: Computational and Statistical Trade-offs, Adaptivity and Optimal Regretcited by 3
- 2023Gradient-Based Spectral Embeddings of Random Dot Product Graphscited by 3
- 2025One Rank at a Time: Cascading Error Dynamics in Sequential Learningcited by 3
- 2021Beyond Procrustes: Balancing-Free Gradient Descent for Asymmetric Low-Rank Matrix Sensingcited by 2
- 2021On Asymptotic Linear Convergence of Projected Gradient Descent for Constrained Least Squarescited by 2
- 2022Error Bound of Empirical ℓ2 Risk Minimization for Noisy Standard and Generalized Phase Retrieval Problemscited by 2
- 2026Low-Rank Adaptation Redux for Large Modelscited by 2
- 2021Square Root Principal Component Pursuit: Tuning-Free Noisy Robust Matrix Recoverycited by 1
- 2022An Exponentially Increasing Step-size for Parameter Estimation in Statistical Modelscited by 1
50 most cited of 68 All 68 (JSON)
- 2026Dynamic Topic Modeling with a Higher-Order Hypergraphical Representation27 May 2026
- 2026Low-Rank Adaptation Redux for Large Models23 Apr 2026
- 2025SOCIA-Nabla: Textual Gradient Meets Multi-Agent Orchestration for Automated Simulator Generation21 Oct 2025
- 2025Global Convergence Analysis of Vanilla Gradient Descent for Asymmetric Matrix Completion13 Aug 2025
- 2025One Rank at a Time: Cascading Error Dynamics in Sequential Learning28 May 2025
- 2025SOCIA-nabla: Textual Gradient Meets Multi-Agent Orchestration for Automated Simulator Generation17 May 2025
- 2025Euclidean Distance Matrix Completion via Asymmetric Projected Gradient Descent28 Apr 2025
- 2025Preconditioned Gradient Descent for Over-Parameterized Nonconvex Matrix Factorization13 Apr 2025
- 2025Riemannian Optimization on Relaxed Indicator Matrix Manifold26 Mar 2025
- 2025Recommendations with Sparse Comparison Data: Provably Fast Convergence for Nonconvex Matrix Factorization27 Feb 2025
- 2025Matrix Completion with Graph Information: A Provable Nonconvex Optimization Approach12 Feb 2025
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descent1 Feb 2025
38 more
- 2025GWT: Scalable Optimizer State Compression for Large Language Model Training13 Jan 2025
- 2024Gradient Weight-normalized Low-rank Projection for Efficient LLM Training27 Dec 2024
- 2024COAP: Memory-Efficient Training with Correlation-Aware Gradient Projection26 Nov 2024
- 2024Parameter-Efficient Fine-Tuning of Large Language Models using Semantic Knowledge Tuning11 Oct 2024
- 2024Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?2 Oct 2024
- 2024Propulsion: Steering LLM with Tiny Fine-Tuning17 Sep 2024
- 2024Predictive Low Rank Matrix Learning under Partial Observations: Mixed-Projection ADMM18 Jul 2024
- 2024VeLoRA: Memory Efficient Training using Rank-1 Sub-Token Projections28 May 2024
- 2024The radius of statistical efficiency15 May 2024
- 2024Near-Optimal differentially private low-rank trace regression with guaranteed private initialization24 Mar 2024
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection6 Mar 2024
- 2024Convergence of Gradient Descent with Small Initialization for Unregularized Matrix Completion9 Feb 2024
- 2024Low-Tubal-Rank Tensor Recovery via Factorized Gradient Descent22 Jan 2024
- 2023Provably Accelerating Ill-Conditioned Low-rank Estimation via Scaled Gradient Descent, Even with Overparameterization9 Oct 2023
- 2023Gradient-Based Spectral Embeddings of Random Dot Product Graphs25 Jul 2023
- 2023Non-Convex Optimizations for Machine Learning with Theoretical Guarantee: Robust Matrix Completion and Neural Network Learning28 Jun 2023
- 2023Exploiting Observation Bias to Improve Matrix Completion7 Jun 2023
- 2023Online Tensor Learning: Computational and Statistical Trade-offs, Adaptivity and Optimal Regret6 Jun 2023
- 2023Fast and Accurate Estimation of Low-Rank Matrices from Noisy Measurements via Preconditioned Non-Convex Gradient Descent26 May 2023
- 2023Computationally Efficient and Statistically Optimal Robust High-Dimensional Linear Regression10 May 2023
- 2023PDPP: Projected Diffusion for Procedure Planning in Instructional Videos26 Mar 2023
- 2023Greedy Pruning with Group Lasso Provably Generalizes for Matrix Sensing20 Mar 2023
- 2023Subspace based Federated Unlearning24 Feb 2023
- 2023Approximate message passing from random initialization with applications to mathbbZ2 synchronization7 Feb 2023
- 2023Matrix Estimation for Individual Fairness4 Feb 2023
- 2022Rank-1 Matrix Completion with Gradient Descent and Small Random Initialization19 Dec 2022
- 2022Federated Learning and Meta Learning: Approaches, Applications, and Directions24 Oct 2022
- 2022Towards a Theoretical Foundation of Policy Optimization for Learning Control Policies10 Oct 2022
- 2022Nonconvex Matrix Factorization is Geodesically Convex: Global Landscape Analysis for Fixed-rank Matrix Optimization From a Riemannian Perspective29 Sep 2022
- 2022A Validation Approach to Over-parameterized Matrix and Image Recovery21 Sep 2022
- 2022Accelerating SGD for Highly Ill-Conditioned Huge-Scale Online Matrix Completion24 Aug 2022
- 2022A Novel Plug-and-Play Approach for Adversarially Robust Generalization19 Aug 2022
- 2022Robust Matrix Completion with Heavy-tailed Noise9 Jun 2022
- 2022Error Bound of Empirical ℓ2 Risk Minimization for Noisy Standard and Generalized Phase Retrieval Problems27 May 2022
- 2022An Exponentially Increasing Step-size for Parameter Estimation in Statistical Models16 May 2022
- 2022Convergence of gradient descent for deep neural networks30 Mar 2022
- 2022Model-free Neural Lyapunov Control for Safe Robot Navigation2 Mar 2022
- 2022Global Convergence of Sub-gradient Method for Robust Matrix Recovery: Small Initialization, Noisy Measurements, and Over-parameterization17 Feb 2022
50 newest of 68 All 68 (JSON)
- 2023Subspace based Federated Unlearningcited by 13
- 2023Subspace based Federated Unlearning24 Feb 2023
- 2024GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projectioncited by 447
- 2021Large Learning Rate Tames Homogeneity: Convergence and Balancing Effectcited by 48
- 2024Fira: Can We Achieve Full-rank Training of LLMs Under Low-rank Constraint?cited by 39
- 2023PDPP: Projected Diffusion for Procedure Planning in Instructional Videoscited by 36
- 2021On the computational and statistical complexity of over-parameterized matrix sensingcited by 31
- 2021Scaling and Scalability: Provable Nonconvex Low-Rank Tensor Estimation from Incomplete Measurementscited by 28
- 2021Mind Mappings: Enabling Efficient Algorithm-Accelerator Mapping Space Searchcited by 24
- 2022Convergence of gradient descent for deep neural networkscited by 24
- 2021Inference for Heteroskedastic PCA with Missing Datacited by 21
- 2022Global Convergence of Sub-gradient Method for Robust Matrix Recovery: Small Initialization, Noisy Measurements, and Over-parameterizationcited by 21
- 2024VeLoRA: Memory Efficient Training using Rank-1 Sub-Token Projectionscited by 18
- 2021Noisy Gradient Descent Converges to Flat Minima for Nonconvex Matrix Factorizationcited by 15
12 most cited of 65 All 65 (JSON)
- 2026Dynamic Topic Modeling with a Higher-Order Hypergraphical Representation27 May 2026
- 2026Low-Rank Adaptation Redux for Large Models23 Apr 2026
- 2025Global Convergence Analysis of Vanilla Gradient Descent for Asymmetric Matrix Completion13 Aug 2025
- 2025One Rank at a Time: Cascading Error Dynamics in Sequential Learning28 May 2025
- 2025Euclidean Distance Matrix Completion via Asymmetric Projected Gradient Descent28 Apr 2025
- 2025Preconditioned Gradient Descent for Over-Parameterized Nonconvex Matrix Factorization13 Apr 2025
- 2025Riemannian Optimization on Relaxed Indicator Matrix Manifold26 Mar 2025
- 2025Recommendations with Sparse Comparison Data: Provably Fast Convergence for Nonconvex Matrix Factorization27 Feb 2025
- 2025Matrix Completion with Graph Information: A Provable Nonconvex Optimization Approach12 Feb 2025
- 2025Efficient Over-parameterized Matrix Sensing from Noisy Measurements via Alternating Preconditioned Gradient Descent1 Feb 2025
- 2025GWT: Scalable Optimizer State Compression for Large Language Model Training13 Jan 2025
- 2024Gradient Weight-normalized Low-rank Projection for Efficient LLM Training27 Dec 2024
12 newest of 65 All 65 (JSON)