胡鹤轩，隋华超，胡强*，张晔，胡震云，马能武

（1.河海大学 计算机与信息学院，南京 211100； 2.水利部水利大数据重点实验室（河海大学），南京 211100；3.西藏农牧学院 电气工程学院，西藏 林芝 8 60000； 4.河海大学 商学院，南京 211100；5.长江勘测规划设计研究有限责任公司，武汉 430010； 6.长江空间信息技术工程有限公司，武汉 430010；7.湖北省水利信息感知与大数据工程技术研究中心，武汉 430010）（∗通信作者电子邮箱huqianghhu@163.com）

基于图注意力网络与双阶注意力机制的径流预报模型

胡鹤轩1，2，3，隋华超1，2，胡强1，2*，张晔1，2，胡震云4，马能武5，6，7

（1.河海大学 计算机与信息学院，南京 211100； 2.水利部水利大数据重点实验室（河海大学），南京 211100；3.西藏农牧学院 电气工程学院，西藏 林芝 8 60000； 4.河海大学 商学院，南京 211100；5.长江勘测规划设计研究有限责任公司，武汉 430010； 6.长江空间信息技术工程有限公司，武汉 430010；7.湖北省水利信息感知与大数据工程技术研究中心，武汉 430010）（∗通信作者电子邮箱huqianghhu@163.com）

为了提高流域径流量预报的准确率，考虑数据驱动水文模型缺乏模型透明度与物理可解释性的问题，提出了一种使用图注意力网络与基于长短期记忆网络（LSTM）的双阶注意力机制（GAT-DALSTM）模型来进行径流预报。首先，以流域站点的水文资料为基础，引入图神经网络提取流域站点的拓扑结构并生成特征向量；其次，针对水文时间序列数据的特点，建立了基于双阶注意力机制的径流预报模型对流域径流量进行预测，并通过基于注意力系数热点图的模型评估方法验证所提模型的可靠性与透明度。……