2025/05/21 by Jiayuan Wu, Zhanwang Deng, Wu, Jiayuan +7
Computer Science · #Distributed Control Multi-Agent Systems #FOS: Mathematics #Optimization and Control (math.OC) #Stochastic Gradient Optimization Techniques #Topological and Geometric Data Analysis
paper · pdf · doi:10.48550/arxiv.2505.15537
openalex publication_date 2025/05/21 · openalex created_date 2025/10/19 · openalex updated_date 2026/07/28
We consider decentralized optimization over a compact Riemannian submanifold in a network of n agents, where each agent holds a smooth, nonconvex local objective defined by its private data. The goal is to collaboratively minimize the sum of these local objective functions. In the presence of data heterogeneity across nodes, existing algorithms typically require communicating both local gradients and iterates to ensure exact convergence with constant step sizes. In this work, we propose REXTRA, a Riemannian extension of the EXTRA algorithm [Shi et al., SIOPT, 2015], to address this limitation. On the theoretical side, we leverage proximal smoothness to overcome the challenges of manifold nonconvexity and establish a global sublinear convergence rate of O(1/k), matching the best-known results. To our knowledge, REXTRA is the first algorithm to achieve a global sublinear convergence rate under a constant step size while requiring only a single round of local iterate communication per iteration. Numerical experiments show that REXTRA achieves superior performance compared to state-of-the-art methods, while supporting larger step sizes and reducing total communication by over 50%.