ML4WX

Sub-seasonal to seasonal

Medium- and Long-Term Precipitation Forecasting Method Based on Data Augmentation and Machine Learning Algorithms

Tiantian Tang, Donglai Jiao, Tao Chen, Guan Gui

IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing ·

PaperPDF

In one sentence. K-means plus SMOTE augmentation improves random forest and XGBoost monthly grid precipitation forecasts for the Danjiangkou basin.

Abstract

Accurate medium and long-term precipitation forecasting plays a vital role in disaster prevention and mitigation and rational allocation of water resources. In recent years, there are various methods for medium- and long-term precipitation forecasting based on machine learning algorithms. However, machine learning has a high demand for the size of sample data. Therefore, this article proposes a data augmentation algorithm based on the K-means clustering algorithm and synthetic minority oversampling technique (SMOTE), which can effectively enhance sample information. Besides, through constructing random forest (RF), extreme gradient boosting (XGB), recurrent neural network (RNN), and long short-term memory (LSTM) are, respectively, constructed as the models to forecast monthly grid precipitation of the Danjiangkou River Basin. This study aims to improve the accuracy of medium- and long-term precipitation forecasting. The main results are the following two aspects: 1) in most years, the anomaly correlation coefficient and Pg score of SMOTE-km-XGB and SMOTE-km-RF exceed that of XGB and RF; furthermore, compared with the other three methods, SMOTE-km-XGB method is more suitable for precipitation forecasting in the studied basin in this article; and 2) the forecasting results of two deep learning methods (RNN and LSTM) show that the sample data processed by the K-means clustering algorithm and SMOTE data augmentation algorithm have not achieved considerable results in deep learning. This study improves the accuracy of precipitation forecast by expanding and balancing the information of sample data, and provides a new research idea for improving the accuracy of medium- and long-term hydrological forecasting.

Labels

Topic
Sub-seasonal to seasonal
Type
New model or method
Method
Classical ML (trees, GP, linear, kernels), RNN / LSTM / ConvLSTM
Time scale
Seasonal to annual
Domain
Regional
Data
Other
Status
Research
Identifiers
arXiv:2022.31404 DOI:10.1109/jstars.2022.3140442 OpenAlex:W4206347244
Added
2026-10-22

Labels and summary are generated by Claude and reviewed weekly. Wrong? Edit this entry.