Research on Numerical Weather Prediction Wind Speed Correction Based on Stacking Ensemble Learning Algorithm

Abstract

The transportation of wind fields plays a crucial role in the dispersion and distribution of air pollutants, and accurate wind speed prediction is essential for assessing pollutant concentrations. In this study, we constructed a Stacking ensemble learning model using three models, namely Random Forest, LightGBM, and XGBoost, as base learners, and the LASSO regression model as the meta-learner to optimize wind speed data forecasting for Zhengzhou City’s urban area using WRF. Firstly, based on Pearson correlation coefficients, we selected meteorological variables that have a significant impact on near-surface wind speed and derived historical lagged features. Bayesian TPE was utilized for hyperparameter tuning and model building. Finally, the performance of the trained model was evaluated by comparing it with ground observation data. The results showed that compared to the WRF model, Random Forest, LightGBM, and XGBoost effectively reduced forecast errors and significantly improved wind speed predictions. Both LightGBM and XGBoost demonstrated similar performance in the correction models across the 11 stations and outperformed Random Forest. The Stacking method integrated the advantages of the base learners and exhibited improvement capabilities over individual models, highlighting the potential of machine learning in enhancing localized and accurate weather forecasting.