
科学技术与工程

中国技术经济学会
Short-term Runoff Prediction in the Yangtze River Basin Based on Bayesian Optimized CNN-LSTM-Att Model
Heng LIU , Zhi LI , Yang FAN
Science Technology and Engineering ›› 2026, Vol. 26 ›› Issue (16) : 6978 -6984.
Short-term river flow forecasting provides a critical time window and decision-making support for disaster prevention, mitigation, and sustainable utilization of water resources. To address the limitations of the single long short-term memory (LSTM) model, such as difficulties in capturing local features, sensitivity to hyperparameters, and poor prediction performance, a novel river discharge prediction model integrating convolutional neural network (CNN), LSTM, and a self-attention mechanism was proposed. The Bayesian optimization (BO) algorithm was employed for hyperparameter tuning, resulting in the construction of a coupled BO-CNN-LSTM-Att model. Taking the Yangtze River Basin in Jingzhou, Hubei Province as a case study, the prediction accuracy of various models was compared and evaluated. The results demonstrate that the proposed integrated model reduces the root mean square error (RMSE) and mean absolute error (MAE) by 51.55% and 53.56%, respectively, compared with the traditional LSTM model, and achieves the highest coefficient of determination (R2) of 0.972 8. The prediction performance is significantly superior to that of incomplete integrated models and other algorithmic models.
short-term water flow prediction / deep learning models / CNN-LSTM-Att / Bayesian optimization
/
| 〈 |
|
〉 |