Short-term load forecasting with bidirectional LSTM-attention based on the sparrow search optimisation algorithm

Aiming at the complexity and diversity of short-term power load data, a bidirectional long short-term memory (BILSTM) prediction model based on attention was proposed for the pretreatment collected data, and the different kinds of data were divided to obtain a training set and test set. The BILSTM layer was used for modelling to enable the extraction of the internal dynamic change rules of features and reduce the loss of historical information. An attention mechanism was used to give different weights to the implied BILSTM states, which enhanced the influence of important information. The sparrow search (SS) algorithm was used to optimise the hyperparameter selection process of the model. The test results showed that the performance of the proposed method was better than that of the traditional prediction model, and the root mean square errors (RMSEs) decreased by (1.18, 1.09, 0.60, 0.54) and (2.11, 0.45, 0.21, 0.11) on different datasets.