Delayed Momentum Aggregation: Communication-efficient Byzantine-robust Federated Learning with Partial Participation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Otsuka, Kaoru, Takezawa, Yuki, Yamada, Makoto
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911730140971008
author Otsuka, Kaoru
Takezawa, Yuki
Yamada, Makoto
author_facet Otsuka, Kaoru
Takezawa, Yuki
Yamada, Makoto
contents Partial participation is essential for communication-efficient federated learning at scale, yet existing Byzantine-robust methods typically assume full client participation. In the partial participation setting, a majority of the sampled clients may be Byzantine, once Byzantine clients dominate, existing methods break down immediately. We introduce delayed momentum aggregation, a principle where the central server aggregates cached momentum from non-sampled clients along with fresh momentum from sampled clients. This principle ensures Byzantine clients remain a minority from the server's perspective even when they dominate the sampled set. We instantiate this principle in our optimizer DeMoA. We analyze the convergence rate of DeMoA, showing that DeMoA is Byzantine-robust under partial participation. Experiments show that, with 20% Byzantine ratio and only 10% partial participation rate, DeMoA achieves the best accuracy even when existing methods fail empirically.
format Preprint
id arxiv_https___arxiv_org_abs_2509_02970
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Delayed Momentum Aggregation: Communication-efficient Byzantine-robust Federated Learning with Partial Participation
Otsuka, Kaoru
Takezawa, Yuki
Yamada, Makoto
Machine Learning
Optimization and Control
Partial participation is essential for communication-efficient federated learning at scale, yet existing Byzantine-robust methods typically assume full client participation. In the partial participation setting, a majority of the sampled clients may be Byzantine, once Byzantine clients dominate, existing methods break down immediately. We introduce delayed momentum aggregation, a principle where the central server aggregates cached momentum from non-sampled clients along with fresh momentum from sampled clients. This principle ensures Byzantine clients remain a minority from the server's perspective even when they dominate the sampled set. We instantiate this principle in our optimizer DeMoA. We analyze the convergence rate of DeMoA, showing that DeMoA is Byzantine-robust under partial participation. Experiments show that, with 20% Byzantine ratio and only 10% partial participation rate, DeMoA achieves the best accuracy even when existing methods fail empirically.
title Delayed Momentum Aggregation: Communication-efficient Byzantine-robust Federated Learning with Partial Participation
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2509.02970