Global Convergence of Stochastic Gradient Descent for Some Non-convex Matrix Problems
Not on the map. Why?
Abstract (arXiv)
Stochastic gradient descent (SGD) on a low-rank factorization is commonly employed to speed up matrix problems including matrix completion, subspace tracking, and SDP relaxation. In this paper, we exhibit a step size scheme for SGD on a low-rank least-squares problem, and we prove that, under broad sampling conditions, our method converges globally from a random starting point within O(ε-1 n log n) steps with constant probability for constant-rank problems. Our modification of SGD relates it to stochastic power iteration. We also show experiments to illustrate the runtime and convergence of the algorithm.
Key facts. Cited by 25 papers in the map (data through 14 Oct 2026). First posted on arXiv 5 Nov 2014.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 10; 2022: 3; 2023: 4; 2024: 2; 2025: 3; 2026: 3.
25 papers.
Sort:
By area:
- 2023QuIP: 2-Bit Quantization of Large Language Models With Guaranteescited by 414
- 2021A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groupscited by 187
- 2021Scatterbrain: Unifying Sparse and Low-rank Attention Approximationcited by 130
- 2022Preference Dynamics Under Personalized Recommendationscited by 25
- 2021Streaming k-PCA: Efficient guarantees for Oja's algorithm, beyond rank-one updatescited by 21
- 2021On the Optimality of the Oja's Algorithm for Online PCAcited by 11
- 2021Bootstrapping the error of Oja's algorithmcited by 11
- 2023Global Optimality in Bivariate Gradient-based DAG Learningcited by 10
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problemscited by 10
- 2022A Novel Stochastic Gradient Descent Algorithm for Learning Principal Subspacescited by 7
- 2024Black-Box k-to-1-PCA Reductions: Theory and Applicationscited by 7
- 2021Rank-one matrix estimation: analytic time evolution of gradient descent dynamicscited by 5
13 more
- 2022From Gradient Flow on Population Loss to Learning with Stochastic Gradient Descentcited by 5
- 2023Matrix Completion in Almost-Verification Timecited by 4
- 2021Riemannian Perspective on Matrix Factorizationcited by 3
- 2021Learning Stochastic Optimal Policies via Gradient Descentcited by 3
- 2025Low-Precision Streaming PCAcited by 3
- 2026Escaping Local Minima Provably in Non-convex Matrix Sensing: A Deterministic Framework via Simulated Liftingcited by 3
- 2023Demystifying the Myths and Legends of Nonconvex Convergence of SGDcited by 2
- 2026Gap-Free Streaming PCA Beyond Rank-One Updates: Near-Optimal Rates and Applications to Differential Privacycited by 1
- 2021Non-Gradient Manifold Neural Networkcited by 0
- 2021AgFlow: Fast Model Selection of Penalized PCA via Implicit Regularization Effects of Gradient Flowcited by 0
- 2025k-SVD with Gradient Descentcited by 0
- 2025Global Convergence of Adaptive Sensing for Principal Eigenvector Estimationcited by 0
- 2026PEIRA: Learning Predictive Encoders through Inter-View Regressor Alignmentcited by 0
- 2026Gap-Free Streaming PCA Beyond Rank-One Updates: Near-Optimal Rates and Applications to Differential Privacy22 Sep 2026
- 2026PEIRA: Learning Predictive Encoders through Inter-View Regressor Alignment17 May 2026
- 2026Escaping Local Minima Provably in Non-convex Matrix Sensing: A Deterministic Framework via Simulated Lifting5 Feb 2026
- 2025Low-Precision Streaming PCA25 Oct 2025
- 2025Global Convergence of Adaptive Sensing for Principal Eigenvector Estimation16 May 2025
- 2025k-SVD with Gradient Descent1 Feb 2025
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problems18 Oct 2024
- 2024Black-Box k-to-1-PCA Reductions: Theory and Applications6 Mar 2024
- 2023Demystifying the Myths and Legends of Nonconvex Convergence of SGD19 Oct 2023
- 2023Matrix Completion in Almost-Verification Time7 Aug 2023
- 2023QuIP: 2-Bit Quantization of Large Language Models With Guarantees25 Jul 2023
- 2023Global Optimality in Bivariate Gradient-based DAG Learning30 Jun 2023
13 more
- 2022A Novel Stochastic Gradient Descent Algorithm for Learning Principal Subspaces8 Dec 2022
- 2022From Gradient Flow on Population Loss to Learning with Stochastic Gradient Descent13 Oct 2022
- 2022Preference Dynamics Under Personalized Recommendations25 May 2022
- 2021Scatterbrain: Unifying Sparse and Low-rank Attention Approximation28 Oct 2021
- 2021AgFlow: Fast Model Selection of Penalized PCA via Implicit Regularization Effects of Gradient Flow7 Oct 2021
- 2021Bootstrapping the error of Oja's algorithm28 Jun 2021
- 2021Non-Gradient Manifold Neural Network15 Jun 2021
- 2021Learning Stochastic Optimal Policies via Gradient Descent7 Jun 2021
- 2021Rank-one matrix estimation: analytic time evolution of gradient descent dynamics25 May 2021
- 2021A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groups19 Apr 2021
- 2021On the Optimality of the Oja's Algorithm for Online PCA31 Mar 2021
- 2021Streaming k-PCA: Efficient guarantees for Oja's algorithm, beyond rank-one updates6 Feb 2021
- 2021Riemannian Perspective on Matrix Factorization1 Feb 2021
- 2023QuIP: 2-Bit Quantization of Large Language Models With Guaranteescited by 414
- 2021A Practical Method for Constructing Equivariant Multilayer Perceptrons for Arbitrary Matrix Groupscited by 187
- 2021Scatterbrain: Unifying Sparse and Low-rank Attention Approximationcited by 130
- 2022Preference Dynamics Under Personalized Recommendationscited by 25
- 2021Streaming k-PCA: Efficient guarantees for Oja's algorithm, beyond rank-one updatescited by 21
- 2021On the Optimality of the Oja's Algorithm for Online PCAcited by 11
- 2021Bootstrapping the error of Oja's algorithmcited by 11
- 2023Global Optimality in Bivariate Gradient-based DAG Learningcited by 10
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problemscited by 10
- 2022A Novel Stochastic Gradient Descent Algorithm for Learning Principal Subspacescited by 7
- 2024Black-Box k-to-1-PCA Reductions: Theory and Applicationscited by 7
- 2021Rank-one matrix estimation: analytic time evolution of gradient descent dynamicscited by 5
12 most cited of 24 All 24 (JSON)
- 2026Gap-Free Streaming PCA Beyond Rank-One Updates: Near-Optimal Rates and Applications to Differential Privacy22 Sep 2026
- 2026Escaping Local Minima Provably in Non-convex Matrix Sensing: A Deterministic Framework via Simulated Lifting5 Feb 2026
- 2025Low-Precision Streaming PCA25 Oct 2025
- 2025Global Convergence of Adaptive Sensing for Principal Eigenvector Estimation16 May 2025
- 2025k-SVD with Gradient Descent1 Feb 2025
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problems18 Oct 2024
- 2024Black-Box k-to-1-PCA Reductions: Theory and Applications6 Mar 2024
- 2023Demystifying the Myths and Legends of Nonconvex Convergence of SGD19 Oct 2023
- 2023Matrix Completion in Almost-Verification Time7 Aug 2023
- 2023QuIP: 2-Bit Quantization of Large Language Models With Guarantees25 Jul 2023
- 2023Global Optimality in Bivariate Gradient-based DAG Learning30 Jun 2023
- 2022A Novel Stochastic Gradient Descent Algorithm for Learning Principal Subspaces8 Dec 2022
12 newest of 24 All 24 (JSON)