Timezone: »

 
Reducing Communication in Nonconvex Federated Learning with a Novel Single-Loop Variance Reduction Method
Kazusato Oko · Shunta Akiyama · Tomoya Murata · Taiji Suzuki

@
in OPT 2022: Optimization for Machine Learning »
Event URL: https://openreview.net/forum?id=pYBZZzbJtE »
In Federated Learning (FL), inter-client heterogeneity causes two types of errors: (i) \emph{client drift error} which is induced by multiple local updates, (ii) \emph{client sampling error} due to partial participation of clients at each communication. While several solutions have been offered to the former one, there is still much room of improvement on the latter one.We provide a fundamental solution to this client sampling error. The key is a novel single-loop variance reduction algorithm, SLEDGE (Single-Loop mEthoD for Gradient Estimator), which does not require periodic computation of full gradient but achieves optimal gradient complexity in the nonconvex finite-sum setting. While sampling a small number of clients at each communication round, the proposed FL algorithm, FLEDGE, requires provably fewer or at least equivalent communication rounds compared to any existing method, for finding first and even second-order stationary points in the general nonconvex setting, and under the PL condition. Moreover, under less Hessian-heterogeneity between clients, the required number of communication rounds approaches to $\tilde{\Theta}(1)$.

Author Information

Kazusato Oko (The University of Tokyo)
Shunta Akiyama (Tokyo University)
Tomoya Murata (NTT DATA Mathematical Systems Inc.)
Taiji Suzuki (The University of Tokyo/RIKEN-AIP)

More from the Same Authors