Moreau Envelopes-Based Personalized Asynchronous Federated Learning: Improving Practicality in Network Edge Intelligence
Anwar Asad, Mostafa M. Fouda, Zubair Md. Fadlullah, Mohamed I. Ibrahem, Nidal Nasser
Abstract
Federated learning is a promising approach for training models on distributed data, driven by increasing demand in various industries. However, federated learning framework faces several key challenges, including communication bottlenecks and client data heterogeneity. Personalized asynchronous federated learning addresses these challenges by customizing the model for individual users based on their local data while trading model updates asynchronously. In this paper, we propose the Personal-ized Moreau Envelopes-based Asynchronous Federated Learning (APFedMe). Our approach combines the strengths of Moreau En-velopes to handle optimization problems and asynchronous weight updates to improve communication efficiency while mitigating heterogeneity data challenges through a personalized learning environment. We evaluate our approach on several datasets and compare it with the baseline PFedMe method. Our experiments demonstrate that the proposed APFedMe outperforms other meth-ods in terms of convergence speed and communication efficiency. Overall, our work contributes to developing more effective and efficient federated learning methods that can be applied in various real-world scenarios.