Large Stepsizes Federated Learning on Logistic Regression with Linearly Separable Data: The Case of Heterogeneous Devices
Abstract
This paper revisits the distributed learning problem for training a multinomial logistic regression model with the Federated Averaging (FedAvg) algorithm. We concentrate on a scenario with arbitrarily large stepsizes and heterogeneous update rules where the devices may perform a different number of local updates in each round. We show that, with linearly separable data, FedAvg is stable with any stepsizes and the objective values converge to zero at the rate of cal O(1/R), where R is the number of communication rounds. Our result also demonstrates that the effects of device heterogeneity vanish asymptotically. For sufficiently large R, the objective values decrease monotonically and is bounded by cal O( 1 / (R Tavg)), where Tavg is the average number of local update steps per communication round across devices. Numerical experiments support our findings.
Related
- Federated Nonparametric Regression with Heterogeneous Clients
- Over-the-Air Federated Learning in Heterogeneous Mobile Wireless Networks
- Mathematical Analysis of Backdoor-Induced Melanoma Risk in Modality-Heterogeneous Federated Learning
- PA-CFL: Privacy-Preserving Adaptive Clustered Federated Learning Under Heterogeneous Data
- A Bayesian Vertical Federated Learning Framework for Multivariate Reduced-Rank High-Dimensional Regression
BibTeX
@misc{wong2026large,
title = {{Large Stepsizes Federated Learning on Logistic Regression with Linearly Separable Data: The Case of Heterogeneous Devices}},
author = {Hok Fong Wong and Hoi-To Wai and Chung-Yiu Yau},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.05915},
eprint = {2610.05915},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.05915},
note = {CC BY-NC-ND 4.0},
}
RIS
TY - UNPB
TI - Large Stepsizes Federated Learning on Logistic Regression with Linearly Separable Data: The Case of Heterogeneous Devices
AU - Hok Fong Wong
AU - Hoi-To Wai
AU - Chung-Yiu Yau
PY - 2026
DA - 2026/10/05
PB - arXiv
DO - 10.48550/arxiv.2610.05915
UR - https://arxiv.org/abs/2610.05915
AB - This paper revisits the distributed learning problem for training a multinomial logistic regression model with the Federated Averaging (FedAvg) algorithm. We concentrate on a scenario with arbitrarily large stepsizes and heterogeneous update rules where the devices may perform a different number of local updates in each round. We show that, with linearly separable data, FedAvg is stable with any stepsizes and the objective values converge to zero at the rate of cal O(1/R), where R is the number of communication rounds. Our result also demonstrates that the effects of device heterogeneity vanish asymptotically. For sufficiently large R, the objective values decrease monotonically and is bounded by cal O( 1 / (R T_avg)), where T_avg is the average number of local update steps per communication round across devices. Numerical experiments support our findings.
ER -