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...
Saved in:
Main Authors: | , , , , , |
---|---|
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 |