一种基于Transformer的城市降雨预测方法技术

本发明专利技术公开了一种基于Transformer的城市降雨预测方法，所述方法包括以下步骤：步骤1、获取目标区域降雨数据；步骤2、构建基于Transformer的降雨预测模型并训练；步骤3、利用训练完成的降雨预测模型对目标区域进行降雨预测。本发明专利技术所述方法克服了现有技术在城市降雨预测方面的不足，实现了对城市降雨的快速、准确预测。本发明专利技术利用人工智能方法进行大量数据学习，相对于雷达预测方式来说，本发明专利技术所述方法的空间分辨率高，能够预测实时降雨数据，准确度高，可扩展性强，为城市洪水预警系统提供数据支撑。提供数据支撑。提供数据支撑。

【技术实现步骤摘要】
一种基于Transformer的城市降雨预测方法


[0001]本专利技术属于降雨预测
，尤其涉及一种基于Transformer的城市降雨预测方法。

技术介绍

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

技术实现思路

[0004]本专利技术的目的在于提供一种基于Transformer的城市降雨预测方法，以解决上述技术问题。
[0005]为实现上述目的，本专利技术提供如下技术方案：
[0006]本专利技术提供了一种基于Transformer的城市降雨预测方法，所述方法包括以下步骤：
[0007]步骤1、获取目标区域降雨数据：获取目标区域的真实降雨数据与合成降雨数据，所述真实降雨数据包括拍摄获取的真实降雨图像与雨量站获取的相应雨强值，所述合成降雨数据包括由网络搜集的原始图像与算法模拟不同雨强的降雨掩膜合成得到的合成降雨图像、相应的降雨掩膜以及雨强值；
[0008]步骤2、构建基于Transformer的降雨预测模型并训练：构建基于Transformer的降雨预测模型，所述模型的输入为降雨图像，输出分别为降雨掩膜与雨强值；
[0009]将获取的真实降雨数据划分为训练集数据与测试集数据，将合成降雨数据全部作为训练集数据，利用训练集数据对模型进行训练，利用测试集数据对模型进行验证，训练过程中调整模型参数反复训练，直至模型收敛且通过验证，完成训练；
[0010]步骤3、利用训练完成的降雨预测模型对目标区域进行降雨预测；采集目标区域当前真实降雨图像，将采集的真实降雨图像输入训练完成的降雨预测模型，得到相应的雨强值，实现对目标区域的降雨预测。
[0011]进一步的是，步骤1中所述的真实降雨图像由手机、监控设备拍摄获取。
[0012]进一步的是，步骤1中所述的网络搜集的原始图像是通过搜索引擎、电子地图以及社交APP途径搜集获取，且网络搜集的原始图像不含有降雨。
[0013]进一步的是，步骤1中所述的合成降雨图像的具体合成过程为：
[0014]1)输入一张网络搜集的原始图像，分别进行光照信息估计与姿态信息估计；
[0015]2)采用粒子系统，输入设定的不同的雨强值，得到相应的雨滴的模拟效果图像，即降雨掩膜，所述输入的雨强值根据目标区域的历史降雨数据设定；
[0016]3)根据输入的原始图像的光照信息，对雨滴进行渲染得到雨水图像；
[0017]4)根据输入的原始图像的姿态信息，融合得到最终的合成降雨图像。
[0018]进一步的是，步骤2中所述的降雨预测模型包括编码器和解码器两部分，所述编码器由多个Transformer模块组成，所述解码器包括融合模块、卷积层、池化层以及第二MLP层；
[0019]所述Transformer模块用于对输入的降雨图像进行处理得到输入图像的多尺度特征图；所述融合模块用于融合多尺度特征图，所述融合模块包括第一MLP层、上采样层以及特征拼接运算块，所述第一MLP层用于统一多尺度特征图的特征维数，所述上采样层用于统一多尺度特征图的空间尺寸，所述特征拼接运算块用于对统一特征维数及空间尺寸后的多尺度特征图在特征维度进行拼接得到融合特征图；所述卷积层用于对融合特征图进行卷积操作；所述池化层用于对融合特征图进行平均池化操作；所述第二MLP层用于对池化后的特征向量进行降维操作。
[0020]进一步的是，步骤2中所述的对模型的训练过程为：
[0021]1)输入一张降雨图像，经由编码器的多个Transformer模块处理得到多个不同尺度的特征图；
[0022]2)设置模型参数，包括特征数、块嵌入层参数、多头自注意力层参数，所述块嵌入层参数包括卷积核大小、步长、填充；所述多头自注意力层参数包括自注意力头数，自注意力特征维度；
[0023]其中，自注意力特征维度(d
head
)＝特征数/自注意力头数(H)，即特征数能够被自注意力头数整除；
[0024]通过设置不同的模型参数输出不同空间尺寸的特征图；
[0025]3)多个不同尺度的特征图经过解码器的融合模块的第一MLP层和上采样层处理后得到大小统一的特征图，并在特征维度进行拼接，进而得到融合特征图；
[0026]4)将得到的融合特征图执行掩膜预测分支任务：进行卷积操作，然后采用归一化指数函数，得到预测掩膜，通过掩膜损失函数评估预测掩膜与掩膜真值之间的误差，所述掩膜损失函数为交叉熵函数；
[0027]具体的，掩膜损失函数的计算公式为：
[0028][0029]式中，M
GT
[i]为输入的雨水图像对应的掩膜图像在像素i的值且M
GT
[i]∈[0,1]，M
Pred
[i]为预测的掩膜图像在像素i处的值且M
Pred
[i]∈[0,1]，N为掩膜图像的像素数目；
[0030]5)将得到的融合特征图执行雨强预测分支任务：通过平均池化操作进行空间降维，再经过第二MLP层对池化后的特征向量进行降维操作，得到预测雨强，通过雨强损失函数评估预测雨强与雨强真值之间的误差，所述雨强损失函数为L2范数；
[0031]具体的，雨强损失函数的计算公式为：
[0032]雨强损失＝(I
GT
–
I
Pred
)2[0033]式中，I
GT
为输入的雨水图像对应的真实雨强值，I
Pred
为预测的雨强值；
[0034]6)计算整体损失函数，由掩膜损失与雨强损失组成整体损失函数为：
[0035]整体损失＝掩码损失+λ*雨强损失
[0036]式中，λ为雨强损失的权重；
[003本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于Transformer的城市降雨预测方法，其特征在于，所述方法包括以下步骤：步骤1、获取目标区域降雨数据：获取目标区域的真实降雨数据与合成降雨数据，所述真实降雨数据包括拍摄获取的真实降雨图像与雨量站获取的相应雨强值，所述合成降雨数据包括由网络搜集的原始图像与算法模拟不同雨强的降雨掩膜合成得到的合成降雨图像、相应的降雨掩膜以及雨强值；步骤2、构建基于Transformer的降雨预测模型并训练：构建基于Transformer的降雨预测模型，所述模型的输入为降雨图像，输出分别为降雨掩膜与雨强值；将获取的真实降雨数据划分为训练集数据与测试集数据，将合成降雨数据全部作为训练集数据，利用训练集数据对模型进行训练，利用测试集数据对模型进行验证，训练过程中调整模型参数反复训练，直至模型收敛且通过验证，完成训练；步骤3、利用训练完成的降雨预测模型对目标区域进行降雨预测；采集目标区域当前真实降雨图像，将采集的真实降雨图像输入训练完成的降雨预测模型，得到相应的雨强值，实现对目标区域的降雨预测。2.根据权利要求1所述的一种基于Transformer的城市降雨预测方法，其特征在于，步骤1中所述的真实降雨图像由手机、监控设备拍摄获取。3.根据权利要求1所述的一种基于Transformer的城市降雨预测方法，其特征在于，步骤1中所述的网络搜集的原始图像是通过搜索引擎、电子地图以及社交APP途径搜集获取，且网络搜集的原始图像不含有降雨。4.根据权利要求3所述的一种基于Transformer的城市降雨预测方法，其特征在于，步骤1中所述的合成降雨图像的具体合成过程为：1)输入一张网络搜集的原始图像，分别进行光照信息估计与姿态信息估计；2)采用粒子系统，输入设定的不同的雨强值，得到相应的雨滴的模拟效果图像，即降雨掩膜，所述输入的雨强值根据目标区域的历史降雨数据设定；3)根据输入的原始图像的光照信息，对雨滴进行渲染得到雨水图像；4)根据输入的原始图像的姿态信息，融合得到最终的合成降雨图像。5.根据权利要求1所述的一种基于Transformer的城市降雨预测方法，其特征在于，步骤2中所述的降雨预测模型包括编码器和解码器两部分，所述编码器由多个Transformer模块组成，所述解码器包括融合模块、卷积层、池化层以及第二MLP层；所述Transformer模块用于对输入的降雨图像进行处理得到输入图像的多尺度特征图；所述融合模块用于融合多尺度特征图，所述融合模块包括第一MLP层、上采样层以及特征拼接运算块，所述第一MLP层用于统一多尺度特征图的特征维数，所述上采样层用于统一多尺度特征图的空间尺寸，所述特征拼接运算块用于对统一特征维数及空间尺寸后的多尺度特征图在特征维度进行拼接得到融合特征图；所述卷积层用于对融...

【专利技术属性】
技术研发人员：俞茜，李娜，王艳艳，王静，王杉，韩松，张念强，丁志雄，王明阳，
申请(专利权)人：中国水利水电科学研究院，
类型：发明
国别省市：