Federated Learning Under Heterogeneous and Correlated Client Availability

Angelo Rodio (1) , Francescomaria Faticanti (1) , Othmane Marfoq (1) , Giovanni Neglia (1) , Emilio Leonardi (2)

1 NEO - Network Engineering and Operations
2 DET - Department of Electronics and Telecommunications [Torino]

In Federated Learning (FL), devices-also referred to as clients-can exhibit heterogeneous availability patterns, often correlated over time and with other clients. This paper addresses the problem of heterogeneous and correlated client availability in FL. Our theoretical analysis is the first to demonstrate the negative impact of correlation on FL algorithms' convergence rate and highlights a trade-off between optimization error (related to convergence speed) and bias error (indicative of model quality). To optimize this trade-off, we propose Correlation-Aware FL (CA-Fed), a novel algorithm that dynamically balances the competing objectives of fast convergence and minimal model bias. CA-Fed achieves this by dynamically adjusting the aggregation weight assigned to each client and selectively excluding clients with high temporal correlation and low availability. Experimental evaluations on diverse datasets demonstrate the effectiveness of CA-Fed compared to state-of-the-art methods. Specifically, CA-Fed achieves the best trade-off between training time and test accuracy. By dynamically handling clients with high temporal correlation and low availability, CA-Fed emerges as a promising solution to mitigate the detrimental impact of correlated client availability in FL.

Keywords: Federated Learning; Correlated Client Availability; Markov Chains

Domains: Machine Learning [cs.LG]; Machine Learning [stat.ML]

Angelo Rodio, Francescomaria Faticanti, Othmane Marfoq, Giovanni Neglia, Emilio Leonardi. Federated Learning Under Heterogeneous and Correlated Client Availability. IEEE/ACM Transactions on Networking, 2023, pp.1-10. ⟨10.1109/TNET.2023.3324257⟩. ⟨hal-04364293⟩