Efficient Communication for Decentralized Federated Learning: an Energy Disaggregation Case Study

Yusen Zhang,Feng Gao,Kangjia Zhou
DOI: https://doi.org/10.1109/pedg61800.2024.10667419
2024-01-01
Abstract:The terminal devices in the energy system face limitations in communication resources, storage space, computational power, and data security, making it challenging to train and deploy computationally intensive artificial intelligence models. Therefore, employing communication-efficient federated learning to train lightweight neural network models is a suitable solution. In this paper, we propose decentralized federated learning with efficient communication achieved by cyclically broadcasting weight parameters from each device during the model training process. We explore the proposed decentralized federated learning framework using energy disaggregation as a case study. Furthermore, grouped convolution is introduced to establish lightweight models, reducing the computational and storage costs during both the training process and model deployment. We conducted extensive experiments on publicly available datasets to validate that the proposed decentralized federated learning approach exhibits significant communication efficiency and minimal impact on the performance of the global model.
What problem does this paper attempt to address?