arrow
Return

Distributed Random Reshuffling Methods With Improved Convergence

delete
delete0
PRE
AI
K
Kun Huang
L
Linli Zhou
S
Shi Pu
DOI:10.1109/TAC.2025.3552743delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
This article proposes two distributed random reshuffling (RR) methods, namely gradient tracking with RR (GT-RR) and exact diffusion with RR (ED-RR), to solve the distributed optimization problem over a connected network, where a set of agents aim to minimize the average of their local cost functions. Both algorithms invoke RR update for each agent, inherit favorable characteristics of RR for minimizing smooth nonconvex objective functions, and improve the performance of previous distributed RR methods both theoretically and empirically. Specifically, both GT-RR and ED-RR achieve the convergence rate of $\mathcal {O}(1/[(1-\lambda)^{1/3}m^{1/3}T^{2/3}])$ in driving the (minimum) expected squared norm of the gradient to zero, where $T$ denotes the number of epochs, $m$ is the sample size for each agent, and $(1-\lambda)$ represents the spectral gap of the mixing matrix. When the objective functions further satisfy the Polyak–Łojasiewicz condition, we show GT-RR and ED-RR both achieve $\mathcal {O}(1/[(1-\lambda)mT^{2}])$ convergence rate in terms of the averaged expected differences between the agents' function values and the global minimum value. Notably, both results are comparable to the convergence rates of centralized RR methods (up to constant factors depending on the network topology) and outperform those of previous distributed RR algorithms.
Keywords:
Distributed optimization
nonconvex optimization
stochastic optimization

Journal

IEEE Transactions on Automatic Control cover
IEEE Transactions on Automatic Control
IF:
7
Papers:
1.3W
Citations:
6.7W

Organization

T
The Chinese University of Hong Kong
Scholars:
3.8K
Papers: 1.9K
Citations: 3
Cited Papers

Cited Papers

errShare
errSave
Fast incremental method for smooth nonconvex optimization
err2016-12-01
err0
PREAI
errSashank J. Reddi; Suvrit Sra; Barnabas Poczos; Alex Smola
errShare
errSave
Distributed Random Reshuffling Over Networks
err2023-01-01
err0
errOAAI
errKun Huang; Xiao Li; Andre Milzarek; Shi Pu; Junwen Qiu
errShare
errSave
On the Influence of Bias-Correction on Distributed Stochastic Optimization
err2020-01-01
err0
errOAAI
errKun Yuan; Sulaiman A. Alghunaim; Bicheng Ying; Ali H. Sayed
errShare
errSave
researcher View more