Research on optimal selection of runoff prediction models based on coupled machine learning methods

Abstract Runoff fluctuations under the influence of climate change and human activities present a significant challenge and valuable application in constructing high-accuracy runoff prediction models. This study aims to address this challenge by taking the Wanzhou station in the Three Gorges Reservo...

Full description

Saved in:
Bibliographic Details
Main Authors: Xing Wei, Mengen Chen, Yulin Zhou, Jianhua Zou, Libo Ran, Ruibo Shi
Format: Article
Language:English
Published: Nature Portfolio 2024-12-01
Series:Scientific Reports
Subjects:
Online Access:https://doi.org/10.1038/s41598-024-83695-8
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1841559530993876992
author Xing Wei
Mengen Chen
Yulin Zhou
Jianhua Zou
Libo Ran
Ruibo Shi
author_facet Xing Wei
Mengen Chen
Yulin Zhou
Jianhua Zou
Libo Ran
Ruibo Shi
author_sort Xing Wei
collection DOAJ
description Abstract Runoff fluctuations under the influence of climate change and human activities present a significant challenge and valuable application in constructing high-accuracy runoff prediction models. This study aims to address this challenge by taking the Wanzhou station in the Three Gorges Reservoir area as a case study to optimize various prediction models. The study first selects artificial neural network (ANN) and support vector machine (SVM) as the base models. Then, it evaluates and selects from three time-series decomposition methods. Time-Varying Filter-based Empirical Mode Decomposition (TVF-EMD), Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN), and Variational Mode Decomposition (VMD). Subsequently, these decomposition methods are coupled with optimization algorithms, including Whale Optimization Algorithm (WOA), Grasshopper Optimization Algorithm (GOA), and Sparrow Search Algorithm (SSA), to construct various hybrid prediction models. The results indicate that: (1) The single prediction model LSTM demonstrated higher prediction accuracy compared to BP and SVM; (2) The VMD-LSTM model outperformed the CEEMDAN-LSTM and TVF-EMD-LSTM models. Compared to the single LSTM model, the Nash-Sutcliffe Efficiency (NSE) and Pearson’s correlation coefficient (R) of the VMD-LSTM model were improved by 15.06% and 6.82%, respectively; (3) Among the machine learning prediction models coupled with various methods, the VMD-SSA-LSTM model achieved the highest accuracy. Compared to the VMD-LSTM model, the NSE and R values of the VMD-SSA-LSTM model were further increased by 13.09% and 4.26%, respectively. Employing a “decomposition-reconstruction” strategy combined with robust optimization algorithms enhances the performance of machine learning prediction models, thereby significantly improving the runoff prediction capabilities in watershed hydrological models.
format Article
id doaj-art-198684c4a8a746bd8634934cd1c2ae2c
institution Kabale University
issn 2045-2322
language English
publishDate 2024-12-01
publisher Nature Portfolio
record_format Article
series Scientific Reports
spelling doaj-art-198684c4a8a746bd8634934cd1c2ae2c2025-01-05T12:24:40ZengNature PortfolioScientific Reports2045-23222024-12-0114111510.1038/s41598-024-83695-8Research on optimal selection of runoff prediction models based on coupled machine learning methodsXing Wei0Mengen Chen1Yulin Zhou2Jianhua Zou3Libo Ran4Ruibo Shi5School of Civil Engineering, Chongqing Three Gorges UniversitySchool of Civil Engineering, Chongqing Three Gorges UniversitySchool of Civil Engineering, Chongqing Three Gorges UniversitySchool of Civil Engineering, Chongqing Three Gorges UniversitySchool of Civil Engineering, Chongqing Three Gorges UniversitySchool of Civil Engineering, Chongqing Three Gorges UniversityAbstract Runoff fluctuations under the influence of climate change and human activities present a significant challenge and valuable application in constructing high-accuracy runoff prediction models. This study aims to address this challenge by taking the Wanzhou station in the Three Gorges Reservoir area as a case study to optimize various prediction models. The study first selects artificial neural network (ANN) and support vector machine (SVM) as the base models. Then, it evaluates and selects from three time-series decomposition methods. Time-Varying Filter-based Empirical Mode Decomposition (TVF-EMD), Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN), and Variational Mode Decomposition (VMD). Subsequently, these decomposition methods are coupled with optimization algorithms, including Whale Optimization Algorithm (WOA), Grasshopper Optimization Algorithm (GOA), and Sparrow Search Algorithm (SSA), to construct various hybrid prediction models. The results indicate that: (1) The single prediction model LSTM demonstrated higher prediction accuracy compared to BP and SVM; (2) The VMD-LSTM model outperformed the CEEMDAN-LSTM and TVF-EMD-LSTM models. Compared to the single LSTM model, the Nash-Sutcliffe Efficiency (NSE) and Pearson’s correlation coefficient (R) of the VMD-LSTM model were improved by 15.06% and 6.82%, respectively; (3) Among the machine learning prediction models coupled with various methods, the VMD-SSA-LSTM model achieved the highest accuracy. Compared to the VMD-LSTM model, the NSE and R values of the VMD-SSA-LSTM model were further increased by 13.09% and 4.26%, respectively. Employing a “decomposition-reconstruction” strategy combined with robust optimization algorithms enhances the performance of machine learning prediction models, thereby significantly improving the runoff prediction capabilities in watershed hydrological models.https://doi.org/10.1038/s41598-024-83695-8Runoff predictionThree gorges reservoir areaVariational mode decompositionSparrow search algorithmMachine learning
spellingShingle Xing Wei
Mengen Chen
Yulin Zhou
Jianhua Zou
Libo Ran
Ruibo Shi
Research on optimal selection of runoff prediction models based on coupled machine learning methods
Scientific Reports
Runoff prediction
Three gorges reservoir area
Variational mode decomposition
Sparrow search algorithm
Machine learning
title Research on optimal selection of runoff prediction models based on coupled machine learning methods
title_full Research on optimal selection of runoff prediction models based on coupled machine learning methods
title_fullStr Research on optimal selection of runoff prediction models based on coupled machine learning methods
title_full_unstemmed Research on optimal selection of runoff prediction models based on coupled machine learning methods
title_short Research on optimal selection of runoff prediction models based on coupled machine learning methods
title_sort research on optimal selection of runoff prediction models based on coupled machine learning methods
topic Runoff prediction
Three gorges reservoir area
Variational mode decomposition
Sparrow search algorithm
Machine learning
url https://doi.org/10.1038/s41598-024-83695-8
work_keys_str_mv AT xingwei researchonoptimalselectionofrunoffpredictionmodelsbasedoncoupledmachinelearningmethods
AT mengenchen researchonoptimalselectionofrunoffpredictionmodelsbasedoncoupledmachinelearningmethods
AT yulinzhou researchonoptimalselectionofrunoffpredictionmodelsbasedoncoupledmachinelearningmethods
AT jianhuazou researchonoptimalselectionofrunoffpredictionmodelsbasedoncoupledmachinelearningmethods
AT liboran researchonoptimalselectionofrunoffpredictionmodelsbasedoncoupledmachinelearningmethods
AT ruiboshi researchonoptimalselectionofrunoffpredictionmodelsbasedoncoupledmachinelearningmethods