首页|期刊导航|华中科技大学学报(自然科学版)|耦合CNN-SE Attention-GRU机制的中长期径流智能预测研究

耦合CNN-SE Attention-GRU机制的中长期径流智能预测研究OA

Mid-long term runoff intelligent prediction research based on CNN-SE Attention-GRU coupling mechanism

中文摘要英文摘要

为提高中长期径流预测准确性,提出了一种结合卷积神经网络(CNN)、门控循环单元(GRU)和通道注意力机制(SE Attention)的深度学习模型.该模型通过CNN提取水文气象数据的空间特征、GRU学习时间序列的长期规律、SE Attention则强化关键信息的影响,构建CNN-SEAttention-GRU复合模型,完成中长期径流预测.以斧头湖流域为例,采用1971-2017年的水文气象数据,对金口站和十好桥站月径流数据采用多元预测进行分析.为验证模型性能,将构建的CNN-SE Attention-GRU模型分别与单一模型LSTM、GRU、双模型CNN-LSTM、CNN-GRU、三模型CNN-SE Attention-LSTM进行了对比.结果表明:CNN-SE Attention-GRU模型表现最优,均方分误差(RMSE)分别达到17.86和5.18,平均绝对误差(MAE)为10.50和3.10,平均绝对百分比误差(MAPE)为15.03%和15.15%,决定系数R2为0.765和0.874,纳什效率系数(NSE)为0.79和0.88.

In order to improve the effect of the mid-long term runoff forecasting,a deep learning model combining convolutional neural networks(CNN)model,gated recurrent unit(GRU)and squeeze-and-excitation attention(SE Attention)was proposed.The model extracts spatial features of hydrometeorological data through CNN,learns long-term temporal patterns via GRU,and enhances the influence of key information using SE Attention,thereby constructing a CNN-SE Attention-GRU hybrid model for medium-and long-term runoff forecasting.The study was conducted in the Futou Lake basin,employing hydrometeorological data from 1971 to 2017 to perform multivariate runoff predictions at the Jinkou and Shihaoqiao stations.The proposed CNN-SE Attention-GRU model was compared with single models(LSTM,GRU),hybrid models(CNN-LSTM,CNN-GRU),and a composite model with the same architecture(CNN-SE Attention-LSTM).The results demonstrate that the CNN-SE Attention-GRU model exhibiting optimal performance,the model attained root mean square error(RMSE)values of 17.86 and 5.18,mean absolute error(MAE)values of 10.50 and 3.10,mean absolute percentage error(MAPE)values of 15.03%and 15.15%,coefficient of determination R2 values of 0.765 and 0.874,and Nash efficiency coefficient(NSE)values of 0.79 and 0.88,respectively.

黎育红;郝舒哲;陆佳俊;张勇传

华中科技大学土木与水利工程学院,湖北武汉 430074华中科技大学土木与水利工程学院,湖北武汉 430074五凌电力有限公司,湖南长沙 410004华中科技大学土木与水利工程学院,湖北武汉 430074

建筑与水利

智慧水文中长期径流预测卷积神经网络门控循环单元注意力机制斧头湖流域

smart hydrlogymid-long term runoff forecastingconvolutional neural networksgated recurrent unitSE AttentionFutou Lake basin

《华中科技大学学报(自然科学版)》 2026 (7)

27-33,7

湖北省水利重点科研项目(HBSLKY202325).

10.13245/j.hust.250280

评论