FedMEM: Adaptive Personalized Federated Learning Framework for Heterogeneous Mobile Edge Environments

Abstract

Abstract With the growth of the Internet of Things (IoT) and communication technologies, edge devices have become more diverse. This diversity has increased the computational load on these systems and led to differences between devices. In mobile edge computing, variations in communication and computing resources can prevent some devices from updating models quickly. This delay affects overall performance. In addition, in federated learning, data that is not independently and identically distributed (non-IID) makes it hard for clients to maintain personalized models.To address these issues, this paper introduces a personalized federated learning framework. This framework enhances the resource allocation optimization algorithm by dynamically adjusting the depth of model inference and the bandwidth allocation strategy, which assists devices with limited computational capabilities in completing inference tasks promptly. Furthermore, it divide the client models into global and personalized layers. Only the global layers are combined, which helps manage the diversity in data distributions. Simulation results show that the proposed FedMEM method is superior to other state-of-the-art methods, and can drastically reduce system latency.