Towards Stagewise and Energy-Accuracy-balanced Client Selection and Resource Allocation over Dynamic Federated Learning Networks

Minghong Wu,Minghui Liwang,Yuhan Su,Yuliang Tang,Zhenzhen Jiao,Xianbin Wang
DOI: https://doi.org/10.1109/iccc62479.2024.10681991
2024-01-01
Abstract:Federated Learning (FL) represents a popular distributed learning architecture that facilitates data privacy by enabling clients (e.g., mobile devices) to train a FL model collaboratively without data sharing. Existing efforts mainly focus on accuracy, delay and energy consumption over a rather stable network with certain clients, with less emphasis on the impact of frequent decision-making processes during model training. Inspired by this, this paper considers a dynamic FL network with uncertain participation of clients, while we jointly optimize client selection and communication resource allocation to achieve a balance between energy cost and FL model accuracy, as proved to be NP-hard. To facilitate timely model training, we propose a prediction-based two-stage asynchronous programming mechanism, which decouples the problem in two subproblems, corresponding to two stages. In particular, the former stage determines some long-term clients which are more stable to join in prior to practical model training process, by estimating the online probability of clients. Then, the latter stage can be implemented by involving some temporary clients as backups when long-term ones are not able to show up. Such a well-designed mechanism offers a unique veiw on FL, while enabling a responsive and cost-effective decision-making process. Comprehensive simulations regarding both IID and non-IID data distributions on MNIST and CIFAR-10 datasets can prove our commendable performance on time efficiency, energy cost and accuracy.
What problem does this paper attempt to address?