FedBranched: Leveraging federated learning for anomaly-aware load forecasting in energy networks

Manzoor, H. U., Khan, A. R., Flynn, D. , Alam, M. M., Akram, M., Imran, M. A. and Zoha, A. (2023) FedBranched: Leveraging federated learning for anomaly-aware load forecasting in energy networks. Sensors, 23(7), 3570. (doi: 10.3390/s23073570)

Abstract

Increased demand for fast edge computation and privacy concerns have shifted researchers’ focus towards a type of distributed learning known as federated learning (FL). Recently, much research has been carried out on FL; however, a major challenge is the need to tackle the high diversity in different clients. Our research shows that using highly diverse data sets in FL can lead to low accuracy of some local models, which can be categorised as anomalous behaviour. In this paper, we present FedBranched, a clustering-based framework that uses probabilistic methods to create branches of clients and assigns their respective global models. Branching is performed using hidden Markov model clustering (HMM), and a round of branching depends on the diversity of the data. Clustering is performed on Euclidean distances of mean absolute percentage errors (MAPE) obtained from each client at the end of pre-defined communication rounds. The proposed framework was implemented on substation-level energy data with nine clients for short-term load forecasting using an artificial neural network (ANN). FedBranched took two clustering rounds and resulted in two different branches having individual global models. The results show a substantial increase in the average MAPE of all clients; the biggest improvement of 11.36% was observed in one client.

• Item Type: Articles
• Additional Information: Funding: This project has received funding from the European Union’s Horizon 2020 Research programme under grant agreement No. 101058505.
• Status: Published
• Refereed: Yes
• Glasgow Author(s) Enlighten ID: Zoha, Dr Ahmed and Khan, Ahsan Raza and Manzoor, Habib Ullah and Imran, Professor Muhammad and Flynn, Professor David

Creator Roles

Manzoor, H. U.: Conceptualization, Data curation, Formal analysis, Methodology, Software, Writing – original draft, Writing – review and editing
Khan, A. R.: Conceptualization, Data curation, Methodology, Software, Writing – original draft, Writing – review and editing
Zoha, A.: Conceptualization, Methodology, Writing – review and editing, Supervision
Flynn, D.: Formal analysis, Resources, Writing – review and editing
Imran, M.: Methodology, Writing – review and editing

• Authors: Manzoor, H. U., Khan, A. R., Flynn, D., Alam, M. M., Akram, M., Imran, M. A., and Zoha, A.
• College/School: College of Science and Engineering > School of Engineering > Autonomous Systems and Connectivity; College of Science and Engineering > School of Engineering > Systems Power and Energy
• Journal Name: Sensors
• Publisher: MDPI
• ISSN: 1424-8220
• ISSN (Online): 1424-8220
• Copyright Holders: Copyright © 2023 by The Authors
• First Published: First published in Sensors 23(7):3570
• Publisher Policy: Reproduced under a Creative Commons licence