Federated Learning with Client Availability Budgets

Yunkai Bao,Steve Drew,Xin Wang,Jiayu Zhou,Xiaoguang Niu
DOI: https://doi.org/10.1109/GLOBECOM54140.2023.10437111
2023-01-01
Abstract:Federated learning (FL) sheds light on efficiently and privately learning from massive Internet of Things (IoT) devices. However, the iterative training and aggregation pose additional stress on the limited energy and availability budgets of clients. In this paper, we discuss two types of availability budgets of IoT clients, including the timing to start participating in FL and the communication budgets due to their constrained energy. We theoretically analyze the effect of availability budgets on FL, based on the availability constraints, by leveraging a decaying quadratic function to prioritize learning from statistically heterogeneous clients during the initial training rounds. We also consider the effects of client availability in terms of their participation to find a balance among clients with varying availability. We present FedCAB, an algorithm applying our theoretical model for the probabilistic rankings of the available clients to select in each round of FL model aggregation. Numerical results show the effectiveness of FedCAB under label distribution skew with a limited communication budget and clients that join the learning process in later rounds. We release the source code of FedCAB at https://github.com/denoslab/FedCAB.
What problem does this paper attempt to address?