水利水电技术(中英文)2026,Vol.57Issue(2):83-94,12.DOI:10.13928/j.cnki.wrahe.2026.02.006
基于SAO优化的LSTM模型在北流河的径流预报研究
Runoff forecasting in Beiliu River based on SAO-optimized LSTM model
摘要
Abstract
[Objective]Due to the frequent occurrence of disasters triggered by extreme runoff events under the influence of climate change and human activities,Long Short-Term Memory(LSTM)—a deep learning model—has been widely applied for runoff forecasting.However,it still requires improvements in both accuracy and interpretability.[Methods]A hybrid model for runoff forecasting was developed by combining the Snow Ablation Optimizer(SAO)with the LSTM model.Hydrometeorological characteristics of the watershed(runoff,precipitation,temperature)and six large-scale climate factors selected based on Pearson Correlation Coefficient(PCC)were used as model inputs.The model was compared with the Back Propagation(BP)neural network,and the Shapley value method was applied to analyze the importance and contribution of the input features.[Results]The SAO-LSTM model achieved a Mean Absolute Percentage Error(MAPE)of 0.26,a Coefficient of Determination(R2)of 0.80,and a Nash-Sutcliffe Efficiency(NSE)of 0.80,significantly outperforming both the LSTM and BP models and demonstrating excellent generalization ability.Shapley interpretation result indicated that precipitation was the key driving factor,while large-scale climate factors had no significant impact on small watersheds and failed to improve the model's forecasting performance.[Conclusion]The SAO-LSTM model significantly improves forecasting performance and exhibits excellent generalization ability and robustness.For runoff forecasting in small watersheds,precipitation is the key driving factor,with significantly higher importance than other feature variables,while large-scale climate factors contribute relatively little.The proposed SAO-LSTM model offers higher forecasting accuracy,provides insights into key factors influencing runoff,demonstrates excellent generalization ability,and shows promising application potential,thereby offering model support for flood control and drought decision-making.关键词
LSTM/SAO/BP/深度学习/径流预报/北流河/大尺度气候因子/影响因素Key words
LSTM/SAO/BP/deep learning/runoff forecasting/Beiliu River/large-scale climate factors/influencing factors分类
天文与地球科学引用本文复制引用
郑凯丰,马兴华,崔国韬,陶昌弟..基于SAO优化的LSTM模型在北流河的径流预报研究[J].水利水电技术(中英文),2026,57(2):83-94,12.基金项目
国家自然科学基金项目(42301012) (42301012)
广州市基础与应用基础研究项目(2024A04J3814) (2024A04J3814)
国家重点研发计划项目(2024YFD1700801-04) (2024YFD1700801-04)
中山大学高校基本科研业务(24qnpy020) (24qnpy020)