NeurIPS2023
Convergence Analysis of Sequential Federated Learning on Heterogeneous Data
Yipeng Li, Xinchen Lyu
43 citations
Abstract
There are two categories of methods in Federated Learning (FL) for joint training across multiple clients: i) parallel FL (PFL), where clients train models in a parallel manner; and ii) sequential FL (SFL), where clients train models in a sequential manner. In contrast to that of PFL, the convergence theory of SFL on heterogeneous data is still lacking. In this paper, we establish the convergence guarantees of SFL for strongly/general/non-convex objectives on heterogeneous data. The convergence guarantees of SFL are better than that of PFL on heterogeneous data with both full and partial client participation. Experimental results validate the counterintuitive analysis result that SFL outperforms PFL on extremely heterogeneous data in cross-device settings. * Xinchen Lyu is the corresponding author. This version fixes some mistakes in the proof of the non-convex case in Thoerems 1 and 2. The original version was at 37th Conference on Neural Information Processing Systems (NeurIPS 2023).