DGD^2: A Linearly Convergent Distributed Algorithm For High-dimensional Statistical Recovery
Marie Maros, Gesualdo Scutari
Abstract
We study linear regression from data distributed over a network of agents (with no master node) under high-dimensional scaling, which allows the ambient dimension to grow faster than the sample size. We propose a novel decentralization of the projected gradient algorithm whereby agents iteratively update their local estimates by a “double-mixing” mechanism, which suitably combines averages of iterates and gradients of neighbouring nodes. Under standard assumptions on the statistical model and network connectivity, the proposed method enjoys global linear convergence up to the statistical precision of the model. This improves on guarantees of (plain) DGD algorithms, whose iteration complexity grows undesirably with the ambient dimension. Our technical contribution is a novel convergence analysis that resembles (albeit different) algorithmic stability arguments extended to high-dimensions and distributed setting, which is of independent interest.
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.
Cited by top-tier papers2
- Decentralized Matrix Sensing: Statistical Guarantees and Fast ConvergenceMarie Maros, Gesualdo ScutariNeurIPS 2023 · 3 citations
- Implicit Regularization of Decentralized Gradient Descent for Sparse RegressionTongle Wu, Ying SunNeurIPS 2024 · 2 citations
Related papers
- Acceleration in Distributed Sparse RegressionMarie Maros, Gesualdo ScutariNeurIPS 2022
- Distributed High-Dimensional Quantile Regression: Estimation Efficiency and Support RecoveryCaixing Wang, Ziliang ShenICML 2024 · 1 citation
- An Improved Analysis of Gradient Tracking for Decentralized Machine LearningAnastasia Koloskova, Tao Lin, Sebastian U. StichNeurIPS 2021 · 148 citations
- Decentralised Learning with Random Features and Distributed Gradient DescentDominic Richards, Patrick Rebeschini, Lorenzo RosascoICML 2020 · 20 citations
- Stability and Generalization of the Decentralized Stochastic Gradient Descent Ascent AlgorithmMiaoxi Zhu, Li Shen, Bo Du, Dacheng TaoNeurIPS 2023 · 12 citations
