Decentralized Matrix Sensing: Statistical Guarantees and Fast Convergence
Marie Maros, Gesualdo Scutari
Abstract
We explore the matrix sensing problem from near-isotropic linear measurements, distributed across a network of agents modeled as an undirected graph, with no server. We provide the first study of statistical, computational/communication guarantees for a decentralized gradient algorithm that solves the (nonconvex) Burer-Monteiro type decomposition associated to the low-rank matrix estimation. With small random initialization, the algorithm displays an approximate two-phase convergence: (i) a spectral phase that aligns the iterates' column space with the underlying low-rank matrix, mimicking centralized spectral initialization (not directly implementable over networks); and (ii) a local refinement phase that diverts the iterates from certain degenerate saddle points, while ensuring swift convergence to the underlying low-rank matrix. Central to our analysis is a novel "in-network" Restricted Isometry Property which accommodates for the decentralized nature of the optimization, revealing an intriguing interplay between sample complexity, network connectivity & topology, and communication complexity.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 94d8a2f6-2e00-40f1-8d0e-a7d18e3286b4Cited by top-tier papers2
- HyperPrism: An Adaptive Non-linear Aggregation Framework for Distributed Machine Learning over Non-IID Data and Time-varying Communication LinksHaizhou Du, Yijian Chen, Ryan Yang, Yuchen Li et al.NeurIPS 2024 · 4 citations
- Implicit Regularization of Decentralized Gradient Descent for Sparse RegressionTongle Wu, Ying SunNeurIPS 2024 · 2 citations
Builds on6
- Small random initialization is akin to spectral learning: Optimization and generalization guarantees for overparameterized low-rank matrix reconstructionDominik Stöger, Mahdi SoltanolkotabiNeurIPS 2021 · 101 citations
- DecentLaM: Decentralized Momentum SGD for Large-batch Deep TrainingKun Yuan, Yiming Chen, Xinmeng Huang, Yingya Zhang et al.ICCV 2021 · 73 citations
- Implicit Regularization in Tensor FactorizationNoam Razin, Asaf Maman, Nadav CohenICML 2021 · 60 citations
- Understanding Incremental Learning of Gradient Descent: A Fine-grained Analysis of Matrix SensingJikai Jin, Zhiyuan Li, Kaifeng Lyu, Simon Shaolei Du et al.ICML 2023 · 46 citations
- Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural NetworksNoam Razin, Asaf Maman, Nadav CohenICML 2022 · 34 citations
Related papers
- Semidefinite Programming versus Burer-Monteiro Factorization for Matrix SensingBaturalp Yalçin, Ziye Ma, Javad Lavaei, Somayeh SojoudiAAAI 2023 · 8 citations
- In-depth Analysis of Low-rank Matrix Factorisation in a Federated SettingConstantin Philippenko, Kevin Scaman, Laurent MassouliéAAAI 2025 · 3 citations
- Robust Matrix Sensing in the Semi-Random ModelXing Gao, Yu ChengNeurIPS 2023 · 6 citations
- Local and Global Convergence of General Burer-Monteiro Tensor OptimizationsShuang Li, Qiuwei LiAAAI 2022 · 3 citations
- Over-parametrization via Lifting for Low-rank Matrix Sensing: Conversion of Spurious Solutions to Strict Saddle PointsZiye Ma, Igor Molybog, Javad Lavaei, Somayeh SojoudiICML 2023 · 5 citations
