A STAM-LSTM model for wind power prediction with feature selection

Abstract

In an effort to enhance the precision of wind power prediction, this study proposes a wind power prediction model with a secondary-weighted attention mechanism, which is based on feature selection. During the pre-processing stage, wavelet denoising is employed on the original dataset to eliminate noise and enhance the convergence rate of the model. As for model improvement, a secondary-weighted time attention mechanism-LSTM (STAM-LSTM) model is proposed. Additionally, the random forest algorithm is employed to analyse the feature correlation, leading to the best feature combination for the construction of the final input vector. In the comparative experiments, the STAM-LSTM model shows good performance and stability compared to the other nine models and three reference methods. In addition, to validate the effectiveness of feature selection, different combinations of features are entered for prediction. The results show that the model metrics reach a better level after feature selection. Finally, the effects of the STAM mechanism and the Random Forest feature screening assistance strategy, on the model are analysed through ablation experiments.