Reliable Federated Learning for Age Sensitive Mobile Edge Computing Systems

The conventional approach for Federated Learning (FL) is to train a global model by averaging local models trained on local data sets. However, given the limited computing resources at the mobile-edge nodes, unreliable models may be received from the Edge Nodes (ENs), which can lead to a significant performance degradation in the FL. Thus, this paper proposes a reliable and age sensitive FL framework that captures the dynamic nature of the local data and computing resources at each participating EN. Specifically, we formulate two optimization problems to select the optimal subset of ENs that can upload their local models in each round of the global model training, given a limited learning cost budget. The first problem aims at selecting the most reliable ENs that should cooperate to complete the FL process, while considering stationary data distributions at different ENs. The second problem aims at minimizing the average age of information experienced by each EN while selecting the most reliable ENs, given fast changing data distributions. Efficient solutions are proposed for the two problems with a worst-case linear complexity. Our Results, leveraging a real-world dataset, depict the efficiency of our solutions in obtaining a better performance compared to conventional FL approach.