English

On Second-order Optimization Methods for Federated Learning

Machine Learning 2021-09-07 v1

Abstract

We consider federated learning (FL), where the training data is distributed across a large number of clients. The standard optimization method in this setting is Federated Averaging (FedAvg), which performs multiple local first-order optimization steps between communication rounds. In this work, we evaluate the performance of several second-order distributed methods with local steps in the FL setting which promise to have favorable convergence properties. We (i) show that FedAvg performs surprisingly well against its second-order competitors when evaluated under fair metrics (equal amount of local computations)-in contrast to the results of previous work. Based on our numerical study, we propose (ii) a novel variant that uses second-order local information for updates and a global line search to counteract the resulting local specificity.

Keywords

Cite

@article{arxiv.2109.02388,
  title  = {On Second-order Optimization Methods for Federated Learning},
  author = {Sebastian Bischoff and Stephan Günnemann and Martin Jaggi and Sebastian U. Stich},
  journal= {arXiv preprint arXiv:2109.02388},
  year   = {2021}
}

Comments

ICML 2021 Workshop "Beyond first-order methods in ML systems"

R2 v1 2026-06-24T05:42:44.547Z