Joint Scheduling of Participants, Local Iterations, and Radio Resources for Fair Federated Learning over Mobile Edge Networks

Federated learning (FL) provides a promising way to train a machine learning model among mobile devices without collecting their raw data to a central node. During training, proper devices are selected to participate in the training process to avoid model unfairness. In a mobile edge network, participant selection must be considered together with three factors: non-iid datasets possessed by devices, tunable local iterations on devices, and radio resource allocation to counter the impact of time-varying channel conditions on parameter transmissions. Since datasets of devices are given, to ensure model fairness and achieve fast convergence in the FL training process, participants, local iterations, and radio resources must be scheduled jointly in each iteration of FL training. In this paper, the joint scheduling problem is analyzed and formulated. Since it is NP-hard, a heuristic scheduling method called PALORA is designed to conduct joint scheduling of participants, local iterations, and radio resources. PALORA consists of three sequentially interactive function blocks: 1) a pointer network embedded deep reinforcement learning method to select participants, 2) an estimation algorithm to determine the numbers of local iterations, and 3) a breadth-first search method to allocate radio resources to the selected participants. PALORA is evaluated via extensive simulations based on real-world datasets. Results show that it significantly outperforms benchmark approaches.