摘要
本发明涉及一种基于多模态特征融合的水库水位预测方法及装置,属于水文预测技术领域。其包括以下步骤:获取水库所在区域的气象数据、水文数据和图像数据;构建特征提取与转换模块,气象数据经过特征提取与转换模块进行处理,得到蒸发量、水库降水量以及水流量;构建特征融合模块,所述特征融合模块包括映射单元和门控注意力模块;蒸发量、水流量以及水库降水量经过映射单元映射到相同维度,然后经过门控注意力模块,得到融合特征;融合特征经过水库水位预测模块进行水位预测,得到水位预测值。本发明通过改进的门控注意力单元动态融合蒸发量、水流量、降水量等多源特征,能够有效降低高维特征的信息冗余与丢失风险,提升预测精度和模型鲁棒性。