基于Attention机制的短临降水预测方法

摘要

本发明公开了基于Attention机制的短临降水预测方法，属于天气预报技术领域，其包括以下步骤：首先输入前t时刻的雷达回波图序列(V1,V2,…,Vt)和模式数据Key(K)，对输入数据进行数据清洗与去噪，得到处理后的输入数据；然后将雷达回波图的灰度图像输入包含4层卷积和最大池化的神经网络，然后经过线性全连接层得到n维数组；接着将雷达回波图序列、模式数据和S2中的Query输入包括4个Attention单元和Norm层的编码器，将结果输入包括4个Attention单元、Norm层和全连接层的解码器，得到预测雷达回波图序列；最后对预测的雷达回波图像，通过Z‑R变换得到未来的区域降水预测，输出短临降水预测结果。本发明能够融合多种气象数据的特征、降低噪声干扰、提升降水预测的精度。