On the Limits of Momentum in Decentralized and Federated Optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zaccone, Riccardo, Karimireddy, Sai Praneeth, Masone, Carlo
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914170337755136
author Zaccone, Riccardo
Karimireddy, Sai Praneeth
Masone, Carlo
author_facet Zaccone, Riccardo
Karimireddy, Sai Praneeth
Masone, Carlo
contents Recent works have explored the use of momentum in local methods to enhance distributed SGD. This is particularly appealing in Federated Learning (FL), where momentum intuitively appears as a solution to mitigate the effects of statistical heterogeneity. Despite recent progress in this direction, it is still unclear if momentum can guarantee convergence under unbounded heterogeneity in decentralized scenarios, where only some workers participate at each round. In this work we analyze momentum under cyclic client participation, and theoretically prove that it remains inevitably affected by statistical heterogeneity. Similarly to SGD, we prove that decreasing step-sizes do not help either: in fact, any schedule decreasing faster than $Θ\left(1/t\right)$ leads to convergence to a constant value that depends on the initialization and the heterogeneity bound. Numerical results corroborate the theory, and deep learning experiments confirm its relevance for realistic settings.
format Preprint
id arxiv_https___arxiv_org_abs_2511_20168
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle On the Limits of Momentum in Decentralized and Federated Optimization
Zaccone, Riccardo
Karimireddy, Sai Praneeth
Masone, Carlo
Machine Learning
Artificial Intelligence
Recent works have explored the use of momentum in local methods to enhance distributed SGD. This is particularly appealing in Federated Learning (FL), where momentum intuitively appears as a solution to mitigate the effects of statistical heterogeneity. Despite recent progress in this direction, it is still unclear if momentum can guarantee convergence under unbounded heterogeneity in decentralized scenarios, where only some workers participate at each round. In this work we analyze momentum under cyclic client participation, and theoretically prove that it remains inevitably affected by statistical heterogeneity. Similarly to SGD, we prove that decreasing step-sizes do not help either: in fact, any schedule decreasing faster than $Θ\left(1/t\right)$ leads to convergence to a constant value that depends on the initialization and the heterogeneity bound. Numerical results corroborate the theory, and deep learning experiments confirm its relevance for realistic settings.
title On the Limits of Momentum in Decentralized and Federated Optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2511.20168