Journal of Hydroinformatics (Jun 2024)

A hybrid annual runoff prediction model using echo state network and gated recurrent unit based on sand cat swarm optimization with Markov chain error correction method

  • Jun Wang,
  • Wenchuan Wang,
  • Xiao-xue Hu,
  • Miao Gu,
  • Yang-hao Hong,
  • Hong-fei Zang

DOI
https://doi.org/10.2166/hydro.2024.038
Journal volume & issue
Vol. 26, no. 6
pp. 1425 – 1453

Abstract

Read online

This study proposes a hybrid model based on the combination of Sand Cat Swarm Optimization (SCSO), Echo State Network (ESN), Gated Recurrent Unit (GRU), Least Squares Method (LSM), and Markov Chain (MC) to improve the accuracy of annual runoff prediction. Firstly, conduct correlation analysis on multi-factor data related to runoff to determine the input of the model. Secondly, the SCSO algorithm is used to optimize the parameters of the ESN and GRU models, and the SCSO-ESN and SCSO-GRU models are established. Next, the prediction results of these two models are coupled using LSM to obtain the preliminary prediction results of the SCSO-ESN-GRU model. Finally, the initial prediction results are corrected for errors using MC to get the final prediction results. Choose Changshui Station and Lanxi Station for experiments, and evaluate the predictive performance of the model through five evaluation indicators. The results show that the combined prediction model corrected by the MC achieved the optimal prediction performance at both experimental stations. This study emphasizes that using a combination prediction model based on Markov chain correction can significantly improve the accuracy of annual runoff prediction, providing a reliable basis for predicting annual runoff in complex watersheds. HIGHLIGHTS A new hybrid sand cat swarm optimization-echo state network-gated recurrent unit-Markov chain (SCSO-ESN-GRU-MC) model is proposed for annual runoff prediction.; A new SCSO algorithm is used to optimize the parameters of the prediction model.; Least squares method is used to couple the ESN and GRU models for annual runoff prediction.; MC is used to correct errors in annual runoff prediction results.; The proposed model achieves the best prediction results at both experimental stations.;

Keywords