Electrical Load Forecasting in Smart Grid: A Personalized Federated Learning Approach

Ratun Rahman,Neeraj Kumar,Dinh C. Nguyen
2024-11-16
Abstract:Electric load forecasting is essential for power management and stability in smart grids. This is mainly achieved via advanced metering infrastructure, where smart meters (SMs) are used to record household energy consumption. Traditional machine learning (ML) methods are often employed for load forecasting but require data sharing which raises data privacy concerns. Federated learning (FL) can address this issue by running distributed ML models at local SMs without data exchange. However, current FL-based approaches struggle to achieve efficient load forecasting due to imbalanced data distribution across heterogeneous SMs. This paper presents a novel personalized federated learning (PFL) method to load prediction under non-independent and identically distributed (non-IID) metering data settings. Specifically, we introduce meta-learning, where the learning rates are manipulated using the meta-learning idea to maximize the gradient for each client in each global round. Clients with varying processing capacities, data sizes, and batch sizes can participate in global model aggregation and improve their local load forecasting via personalized learning. Simulation results show that our approach outperforms state-of-the-art ML and FL methods in terms of better load forecasting accuracy.
Machine Learning,Signal Processing
What problem does this paper attempt to address?
This paper aims to solve the problem of power load forecasting in the smart grid. Specifically, it focuses on how to improve the accuracy of power load forecasting through the Personalized Federated Learning (PFL) method without sharing data. Although traditional machine - learning methods can be used for load forecasting, they require data sharing, which raises the issue of data privacy. While existing federated learning methods have solved the problem of data sharing, they perform poorly when dealing with non - independently and identically distributed (non - IID) data, especially among heterogeneous Smart Meters (SMs) with unbalanced data distributions. Therefore, this study proposes a personalized federated learning method based on Meta - learning to overcome the limitations of existing methods and achieve more efficient and accurate load forecasting. The key contributions of the paper include: - Proposing a new personalized federated learning method, called the Personalized Meta - LSTM algorithm, which allows complex and diverse data to be structured, assembled, and processed quickly while avoiding data sharing and protecting the privacy and security of household electricity consumption records. - Introducing an adaptive model aggregation mechanism. Before training the local model, a temporary evaluation is carried out with different learning rates, and the learning rate that produces the lowest loss value is selected as the optimal learning rate, and then this learning rate is used for local model training. - Conducting extensive experiments to prove that the proposed model outperforms existing machine - learning and federated learning methods in both IID and non - IID settings, especially in terms of the accuracy of load forecasting. Through these contributions, this study not only improves the accuracy of power load forecasting, but also solves the problems of data privacy and unbalanced data distribution, providing strong support for the management and stability of the smart grid.