基于CNN和BiLSTM的临近降水预测方法与流程

本发明涉及计算机视觉及大气科学技术领域，更具体的说是涉及一种基于cnn和bilstm的临近降水预测方法。

背景技术：

临近降水预报在气象服务领域一直是非常重要的问题。临近降水预报是指针对某个区域，在相对较短的时间里(一般是0～2小时)预测降雨强度。目前，国内外最常采用的临近降水预报方法是基于雷达外推的方法，其中光流法的雷达回波外推方法最为常用。光流法是利用图像序列中像素在时间域上的变化以及相邻帧之间的相关性来找到上一帧与当前帧之间存在的对应关系，从而计算出相邻帧之间物体的运动信息的一种方法。

但是，光流法的基本假设条件包括亮度恒定不变和时间连续“小运动”，也就是说，光流法应用的前提是同一目标在不同图像帧之间的运动，其亮度不会发生改变，且相邻图像帧之间的位移要比较小。这种条件在实际应用中较难达到。此外，为了得到预测的降水，在得到预测的雷达回波图像之后，还要根据z-i关系反演降雨量，这样分步预测降水容易造成误差叠加，降低降水预测的整体精度。

近年来，图像检测、分类、回归等复杂问题随着机器学习的兴起，正逐渐简化成大数据集和训练模型的问题，上述问题也迎刃而解。随着计算机计算能力的大幅度提升，深度学习逐渐成为机器学习领域的一个新的研究方向，并在气象预报领域取得了初步成功的应用。但由于降水既与前一段时间的气象条件相关，又和周围地区的气象条件相关。实际上，降水量的预测是一个时空序列预测问题，它以过去的雷达图序列为输入，以未来一段时间的降雨量作为输出。长短时记忆网络(longshorttermmemorynetworks，简称lstm)特别是双向长短时记忆网络(bilstm)为处理具有时间维度以及类似时间维度信息的循环神经网络的一种改进模型，能弥补卷积神经网络(convolutionalneuralnetworks，简称cnn)只能处理空间维度信息的缺陷。

但是，双向长短时记忆网络本身的计算单元较多，其深度极易受到内存的限制，目前并未取得理想的应用效果。

因此，如何提供一种预测精度高、预测过程更加高效的临近降水预测方法是本领域技术人员亟需解决的问题。

技术实现要素：

有鉴于此，本发明提供了一种基于cnn和bilstm的临近降水预测方法，该方法针对复杂序列雷达回波进行降水预测的问题，首先通过卷积神经网络提取序列雷达回波图的上下文信息，再利用双向长短时记忆网络根据雷达回波序列的前后文关系进行分析预测，该方法解决了现有的降水预测方法预测精度低且效率难以保证的问题。

为了实现上述目的，本发明采用如下技术方案：

一种基于cnn和bilstm的临近降水预测方法，该方法包括：

获取序列雷达图像以及对应未来一小时的降雨数据集；

将数据集按照预设比例分为训练集和测试集；

构建tiny-rainnet网络模型；

根据预设的高度和时间跨度条件，将序列雷达图像分别缩放至对应的尺寸；

分别将缩放获得的雷达图像输入到tiny-rainnet网络模型中，训练得到降水预测模型；

输入测试集序列雷达回波图像至降水预测模型，预测对应区域的降雨量，并计算预测降雨量和实际降雨量之间的均方根误差，将最小的均方根误差对应的图像尺寸作为最佳图像缩放尺寸；

使用最佳图像缩放尺寸的序列雷达图像作为输入，对应降水量作为输出，训练tiny-rainnet降水预测模型；

输入任一组序列雷达图像到训练后的tiny-rainnet降水预测模型中，得到对应区域未来一小时降雨量的预测值。

进一步地，获取序列雷达图像的过程，具体包括：

获取t时刻，t-6min时刻，t-2×6min时刻…,t-15×6min时刻共15个时间跨度，每个时间跨度下包括0.5km，1.5km，2.5km，3.5km共4个高度的一共60张像素为101×101的序列雷达回波图像。

进一步地，根据预设的高度和时间跨度条件，将序列雷达图像分别缩放至对应的尺寸，具体为：

将像素为101×101的雷达图像分别在同一高度不同时间跨度和同一时间跨度不同高度下分别缩放至像素为5×5，10×10，…，90×90，95×95，101×101不同的尺寸。

进一步地，将数据集按照预设比例分为训练集和测试集，所述预设比例为8:2。

进一步地，所述tiny-rainnet模型包括：

卷积层1，参数为(30,(1,1))，表示卷积核的个数为30，像素大小为1×1；

卷积层2，参数为(60,(2,2))；

防止过拟合层dropout3，参数为0.5，表示按照0.5的概率屏蔽部分神经元；

最大池化层4，参数为(2,2)，表示按照2×2的像素大小选取最大值替代该单元；

双向长短时记忆网络层5，参数为256，表示隐藏神经元个数为256；

防止过拟合层dropout6，参数为0.5；

双向长短时记忆网络层7，参数为128；

防止过拟合层dropout8，参数为0.5；

双向长短时记忆网络层9，参数为64；

防止过拟合层dropout10，参数为0.5；

全连接层11，参数为100，表示当前层的输出维度为100；

全连接层12，参数为1，表示当前层的输出维度为1。

本发明中提出的tiny-rainnet模型是由卷积神经网络(cnn)和双向长短时记忆网络(bilstm)组合而成的模型，通过该模型结构的合理设计，可以更好的实现临近降水的精准预测。

经由上述的技术方案可知，与现有技术相比，本发明公开提供了一种基于cnn和bilstm的临近降水预测方法，该方法针对复杂序列雷达回波进行降水预测的问题，首先通过卷积神经网络提取序列雷达回波图的上下文信息，再利用双向长短时记忆网络根据雷达回波序列的前后文关系进行分析预测，同时，在每一个卷积层和双向长短时记忆网络层后增加池化层有效地防止因训练集过快拟合而导致测试效果差的问题，该方法得到的临近降水预测结果准确可靠，且预测效率也大大提高。

附图说明

为了更清楚地说明本发明实施例或现有技术中的技术方案，下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本发明的实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据提供的附图获得其他的附图。

图1附图为本发明提供的基于cnn和bilstm的临近降水预测方法的流程示意图；

图2附图为本发明实施例中tiny-rainnet网络模型的结构示意图。

具体实施方式

下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本发明保护的范围。

参见附图1，本发明实施例公开了一种基于cnn和bilstm的临近降水预测方法，该方法包括：

s1：获取序列雷达图像以及对应未来一小时的降雨数据集；

s2：将数据集按照预设比例分为训练集和测试集；

s3：构建tiny-rainnet网络模型；

s4：根据预设的高度和时间跨度条件，将序列雷达图像分别缩放至对应的尺寸；

s5：分别将缩放获得的雷达图像输入到tiny-rainnet网络模型中，训练得到降水预测模型；

s6：输入测试集序列雷达回波图像至降水预测模型，预测对应区域的降雨量，并计算预测降雨量和实际降雨量之间的均方根误差，将最小的均方根误差对应的图像尺寸作为最佳图像缩放尺寸；

s7：使用最佳图像缩放尺寸的序列雷达图像作为输入，对应降水量作为输出，训练tiny-rainnet降水预测模型；

s8：输入任一组序列雷达图像到训练后的tiny-rainnet降水预测模型中，得到对应区域未来一小时降雨量的预测值。

下面通过具体的实例对上述方法做详细的介绍：

第一步：获取深圳市t时刻，t-6min时刻,t-2×6min时刻…,t-15×6min时刻这15个时间跨度，每个时间跨度下包括0.5km,1.5km,2.5km,3.5km4个高度的一共60张像素为101×101的序列雷达回波图像(15×4×101×101)及其对应区域的未来1小时的降雨量值的数据集10000组；

第二步：按照8:2的比例将10000组序列雷达回波图像和未来1小时降水量组成的数据集分为8000组训练集和2000组测试集；

第三步：根据图2构建tiny-rainnet网络模型；

第四步：将像素为101×101的雷达图像分别在同一高度不同时间跨度和同一时间跨度不同高度下分别缩放至像素为5×5，10×10，…，90×90，95×95，101×101不同的尺寸；

第五步：分别将缩放后的雷达图像输入到tiny-rainnet网络中，训练降水预测模型；

第六步：输入测试集序列雷达回波图像至降水预测模型，预测对应区域的降雨量大小，并计算预测降雨量和实际降雨量之间的均方根误差，最小的均方根误差对应的图像尺寸10×10为该网络对应的最佳图像缩放尺寸；

第七步：将序列雷达图像缩放至10×10作为输入，对应降水量作为输出，训练tiny-rainnet降水预测模型；

第八步：输入任一组序列雷达图像到训练后的tiny-rainnet降水预测模型中，即可得到对应区域未来1小时降雨量的预测值。

在一个具体的实施例中，tiny-rainnet模型包括：

卷积层1，参数为(30,(1,1))，表示卷积核的个数为30，像素大小为1×1；

卷积层2，参数为(60,(2,2))；

防止过拟合层dropout3，参数为0.5，表示按照0.5的概率屏蔽部分神经元；

最大池化层4，参数为(2,2)，表示按照2×2的像素大小选取最大值替代该单元；

双向长短时记忆网络层5，参数为256，表示隐藏神经元个数为256；

防止过拟合层dropout6，参数为0.5；

双向长短时记忆网络层7，参数为128；

防止过拟合层dropout8，参数为0.5；

双向长短时记忆网络层9，参数为64；

防止过拟合层dropout10，参数为0.5；

全连接层11，参数为100，表示当前层的输出维度为100；

全连接层12，参数为1，表示当前层的输出维度为1。

综上所述，本发明公开的基于cnn和bilstm的临近降水预测方法，与现有技术相比，具有如下优点：

1、该方法针对复杂序列雷达回波进行降水预测的问题，首先通过卷积神经网络提取序列雷达回波图的上下文信息，再利用双向长短时记忆网络根据雷达回波序列的前后文关系进行分析预测；

2、在每一个卷积层和双向长短时记忆网络层后增加池化层有效地防止因训练集过快拟合而导致测试效果差的问题，该方法得到的临近降水预测结果准确可靠，且预测效率也大大提高。

本说明书中各个实施例采用递进的方式描述，每个实施例重点说明的都是与其他实施例的不同之处，各个实施例之间相同相似部分互相参见即可。对于实施例公开的装置而言，由于其与实施例公开的方法相对应，所以描述的比较简单，相关之处参见方法部分说明即可。

对所公开的实施例的上述说明，使本领域专业技术人员能够实现或使用本发明。对这些实施例的多种修改对本领域的专业技术人员来说将是显而易见的，本文中所定义的一般原理可以在不脱离本发明的精神或范围的情况下，在其它实施例中实现。因此，本发明将不会被限制于本文所示的这些实施例，而是要符合与本文所公开的原理和新颖特点相一致的最宽的范围。