A Two Layer Model Enabled by Federated Learning in the IoT Based Smart Grid Networks with MEC

Jing Zou,Peizhe Xin,Zhizhuo Li,Heli Zhang,Xin Wang
DOI: https://doi.org/10.1109/icet61945.2024.10673409
2024-01-01
Abstract:With the rapid growth of intelligent surveillance and new appliances in the smart grid networks, the scale of the Internet of Things (IoT) has increased dramatically. The data generated at the edge has experienced explosive growth. To address the limitations of centralized machine learning frameworks in terms of communication and computation, federated learning is considered for distributed machine learning to accomplish various related tasks. However, the computational and communication capabilities of devices in federated learning are unevenly distributed, resulting in some device nodes being unable to submit updates in a timely manner, leading to delayed training and improper allocation of training rounds. To ensure effective aggregation of the federated learning model, this paper proposes a two-layer algorithm for the allocation of computation power in the smart grids networks. Finally some simulation results demonstrate that the proposed algorithm can effectively improve the convergence rate of model accuracy while satisfying accuracy constraints. It also achieves reasonable allocation of training rounds for device nodes and minimizes energy consumption.
What problem does this paper attempt to address?