一种基于Transformer的城市降雨预测方法与流程

本发明属于降雨预测，尤其涉及一种基于transformer的城市降雨预测方法。

背景技术：

1、研建城市实时动态洪水风险分析或内涝预警系统对防汛预警和应急调度决策工作有非常重要的作用，但是受限于缺乏高时空分辨率的城市降雨数据等原因，导致此项工作还未能在全国各主要城市全面铺开和应用。城市降雨的时空分布特征会显著影响城市洪水的分布特征，例如不同区域的淹没水深、范围等，一般情况下，可以通过降雨设备监测或者模型模拟等方式来获取降雨数据。降雨监测一般包括地面雨量站、天气雷达和卫星遥感等，其中地面雨量站能够准确地监测降雨数据，但由于城市站点数量有限，因此降雨数据的空间分辨率较低；天气雷达可以利用电磁波的散射效应对降雨强度进行高时间分辨率的预测，但降雨垂直分布不均匀、电磁波异常传播、高层建筑等因素会影响天气雷达的进场方式，导致无法保证数据精度。此外，由于许多国家的地面雷达站数量很少，因此这种方法所能提供的空间范围有限。相比之下，卫星遥感监测可以提供大空间覆盖的降雨预测，但在城市尺度上的时空分辨率往往较低。

2、目前国内外研建了不少天气预报系统，例如美国的中尺度数值天气预报模式wrf、全球气候模型gcm、中国的全球预报系统grapes等，但是这些系统通常无法提供城市尺度的高时空分辨率的精确降雨估计。因此，亟需一种城市降雨预测方法，能够对实时监测数据进行有效利用和挖掘，实现对城市降雨的快速及准确预测。

技术实现思路

1、本发明的目的在于提供一种基于transformer的城市降雨预测方法，以解决上述技术问题。

2、为实现上述目的，本发明提供如下技术方案：

3、本发明提供了一种基于transformer的城市降雨预测方法，所述方法包括以下步骤：

4、步骤1、获取目标区域降雨数据：获取目标区域的真实降雨数据与合成降雨数据，所述真实降雨数据包括拍摄获取的真实降雨图像与雨量站获取的相应雨强值，所述合成降雨数据包括由网络搜集的原始图像与算法模拟不同雨强的降雨掩膜合成得到的合成降雨图像、相应的降雨掩膜以及雨强值；

5、步骤2、构建基于transformer的降雨预测模型并训练：构建基于transformer的降雨预测模型，所述模型的输入为降雨图像，输出分别为降雨掩膜与雨强值；

6、将获取的真实降雨数据划分为训练集数据与测试集数据，将合成降雨数据全部作为训练集数据，利用训练集数据对模型进行训练，利用测试集数据对模型进行验证，训练过程中调整模型参数反复训练，直至模型收敛且通过验证，完成训练；

7、步骤3、利用训练完成的降雨预测模型对目标区域进行降雨预测；采集目标区域当前真实降雨图像，将采集的真实降雨图像输入训练完成的降雨预测模型，得到相应的雨强值，实现对目标区域的降雨预测。

8、进一步的是，步骤1中所述的真实降雨图像由手机、监控设备拍摄获取。

9、进一步的是，步骤1中所述的网络搜集的原始图像是通过搜索引擎、电子地图以及社交app途径搜集获取，且网络搜集的原始图像不含有降雨。

10、进一步的是，步骤1中所述的合成降雨图像的具体合成过程为：

11、1)输入一张网络搜集的原始图像，分别进行光照信息估计与姿态信息估计；

12、2)采用粒子系统，输入设定的不同的雨强值，得到相应的雨滴的模拟效果图像，即降雨掩膜，所述输入的雨强值根据目标区域的历史降雨数据设定；

13、3)根据输入的原始图像的光照信息，对雨滴进行渲染得到雨水图像；

14、4)根据输入的原始图像的姿态信息，融合得到最终的合成降雨图像。

15、进一步的是，步骤2中所述的降雨预测模型包括编码器和解码器两部分，所述编码器由多个transformer模块组成，所述解码器包括融合模块、卷积层、池化层以及第二mlp层；

16、所述transformer模块用于对输入的降雨图像进行处理得到输入图像的多尺度特征图；所述融合模块用于融合多尺度特征图，所述融合模块包括第一mlp层、上采样层以及特征拼接运算块，所述第一mlp层用于统一多尺度特征图的特征维数，所述上采样层用于统一多尺度特征图的空间尺寸，所述特征拼接运算块用于对统一特征维数及空间尺寸后的多尺度特征图在特征维度进行拼接得到融合特征图；所述卷积层用于对融合特征图进行卷积操作；所述池化层用于对融合特征图进行平均池化操作；所述第二mlp层用于对池化后的特征向量进行降维操作。

17、进一步的是，步骤2中所述的对模型的训练过程为：

18、1)输入一张降雨图像，经由编码器的多个transformer模块处理得到多个不同尺度的特征图；

19、2)设置模型参数，包括特征数、块嵌入层参数、多头自注意力层参数，所述块嵌入层参数包括卷积核大小、步长、填充；所述多头自注意力层参数包括自注意力头数，自注意力特征维度；

20、其中，自注意力特征维度(dhead)＝特征数/自注意力头数(h)，即特征数能够被自注意力头数整除；

21、通过设置不同的模型参数输出不同空间尺寸的特征图；

22、3)多个不同尺度的特征图经过解码器的融合模块的第一mlp层和上采样层处理后得到大小统一的特征图，并在特征维度进行拼接，进而得到融合特征图；

23、4)将得到的融合特征图执行掩膜预测分支任务：进行卷积操作，然后采用归一化指数函数，得到预测掩膜，通过掩膜损失函数评估预测掩膜与掩膜真值之间的误差，所述掩膜损失函数为交叉熵函数；

24、具体的，掩膜损失函数的计算公式为：

25、

26、式中，mgt[i]为输入的雨水图像对应的掩膜图像在像素i的值且mgt[i]∈[0,1]，mpred[i]为预测的掩膜图像在像素i处的值且mpred[i]∈[0,1]，n为掩膜图像的像素数目；

27、5)将得到的融合特征图执行雨强预测分支任务：通过平均池化操作进行空间降维，再经过第二mlp层对池化后的特征向量进行降维操作，得到预测雨强，通过雨强损失函数评估预测雨强与雨强真值之间的误差，所述雨强损失函数为l2范数；

28、具体的，雨强损失函数的计算公式为：

29、雨强损失＝(igt–ipred)2

30、式中，igt为输入的雨水图像对应的真实雨强值，ipred为预测的雨强值；

31、6)计算整体损失函数，由掩膜损失与雨强损失组成整体损失函数为：

32、整体损失＝掩码损失+λ*雨强损失

33、式中，λ为雨强损失的权重；

34、通过不断优化模型参数来降低整体损失，直至整体损失函数收敛至小于设定阈值，完成训练。

35、进一步的是，所述transformer模块的具体执行过程为：

36、1)以卷积的方式执行块嵌入层；

37、2)执行层归一化、多头自注意力层，并以残差形式进行加和运算；

38、3)执行层归一化、前馈神经网络层，并以残差形式进行加和运算。

39、本发明的有益效果是：本发明所述方法克服了现有技术在城市降雨预测方面的不足，实现了对城市降雨的快速、准确预测。

40、具体来说，本发明的优点在于：

41、(1)空间分辨率高：在城市地区，雷达站密度普遍较低，在空间覆盖方面存在限制，因此使用雷达数据准确预测降雨强度的空间范围受限，难以捕捉整个城市地区的降雨情况。本发明利用人工智能方法进行大量数据学习，可以提高降雨数据的空间分辨率。

42、(2)降雨预测实时性：通过直接采集降雨图像分析降雨雨强，可以预测实时降雨数据。这对于城市洪水预警系统是非常重要的，因为及时准确的降雨预测可以提前采取必要的措施来减轻洪水风险。

43、(3)预测准确度高：基于天气雷达回波图的雨量预测，极易受到电磁波传播异常、地面杂波干扰，高层建筑遮挡等因素影响，导致预测的准确度下降。本发明所述方法通过移动设备等终端直接采集降雨图像进行雨强预测，准确度高。

44、(4)可扩展性强：使用道路监控设备和移动手机采集数据的方式具有很大的可扩展性。随着技术的不断进步和城市设施的增加，可以随时添加更多的数据采集点，从而进一步提高降雨预测的准确性和覆盖范围。

45、下面结合附图及实施例对本发明作进一步详细说明。