CN110826693A - 基于DenseNet卷积神经网络的三维大气温度廓线反演方法和系统 - Google Patents

Abstract

本发明公开了基于DenseNet卷积神经网络的三维大气温度廓线反演方法和系统，属于大气微波遥感领域。包括：根据氧气吸收频段的二维大气观测亮温图像和三维大气温度廓线，构建训练数据集；基于训练数据集，训练DenseNet卷积神经网络直至其收敛，得到训练好的网络；将待反演的亮温图像输入训练好的网络，输出反演得到的三维大气温度廓线。本发明训练所用数据集以二维亮温图像为单位，每个亮温图像覆盖了地球上的一定区域，整个数据集跨越长时间间隔，泛化能力将大大提高，反演误差也会降低。DenseNet卷积神经网络层数较深，因其密集连接的结构避免了训练过程中的梯度消失问题，适用于复杂的反演问题，可实现直接对晴空、云以及雨天三种场景的数据一起反演，减少耗时。

Description

基于DenseNet卷积神经网络的三维大气温度廓线反演方法和 系统

技术领域

本发明属于大气微波遥感及探测技术领域，更具体地，涉及基于DenseNet卷积神经网络的三维大气温度廓线反演方法和系统。

背景技术

台风、暴雨、强对流等天气因其突发性严重威胁着人们的生命安全，且对社会经济会产生严重的影响，大气温度廓线作为一种重要的大气气象要素，可用于数值天气预报初始场的建立，对于提升数值天气预报性能具有重要作用。因此，从卫星观测资料中及时、准确的反演大气温度廓线，具有十分重要的意义。

大气温度探测一般使用工作在微波被动遥感方式的星载微波辐射计实现，辐射计的直接观测资料为微波亮温数据，需要使用数学反演算法从观测亮温数据中反演得到大气温度廓线。目前普遍使用的微波被动遥感大气温度廓线反演算法主要有三种。统计类方法应用数学统计的思想处理亮温数据，算法简单稳定，但对于具有非线性关系的亮温与温度廓线，其效果不佳；一维变分方法利用最小二乘方法寻求温度廓线对应亮温的最优解；神经网络反演方法利用人工神经网络实现对温度廓线的反演。

BP神经网络是一种反向传播误差的前馈网络模型，在温度廓线反演领域应用较广。为降低反演复杂度和缩短反演时间，一般使用3层BP神经网络反演温度廓线，网络结构较为简单，但正因如此，网络泛化能力差，若训练与反演所用数据的分布差异较大，则反演所得温度的误差也会变大。在实际情况中，BP神经网络只能反演与训练所用亮温数据时间接近且区域相邻的场景的温度廓线。由于BP神经网络对反演数据分布。较高的敏感性，为提高反演精度，还需根据训练数据对不同天气情况进行筛选，并将晴空、云以及雨天三种场景的数据分开训练并反演，若要反演指定区域的温度廓线，则需用多个场景的模型分别反演再组合，耗时较长且流程复杂。

发明内容

针对现有技术基于BP神经网络的三维大气温度廓线反演方法泛化能力差，耗时较长流程复杂的问题，本发明提供了基于DenseNet卷积神经网络的三维大气温度廓线反演方法和系统，其目的在于简化温度廓线反演的流程、缩短反演时间，并提高反演网络的泛化能力，从而提高反演精度。

为实现上述目的，按照本发明的第一方面，提供了一种基于DenseNet卷积神经网络的三维大气温度廓线反演方法，该方法包括以下步骤：

S1.根据氧气吸收频段的二维大气观测亮温图像和三维大气温度廓线，构建训练数据集；

S2.基于训练数据集，训练DenseNet卷积神经网络直至其收敛，得到训练好的网络；

S3.将待反演的亮温图像输入训练好的网络，输出反演得到的三维大气温度廓线。

具体地，步骤S1包括以下子步骤：

S11.将频率不同、时间和区域相同的二维大气观测亮温图像进行组合，每组多频率亮温图像作为一个样本，不同样本之间选择的频率相同，时间和区域不同；

S12.根据区域的经纬度，将三维大气温度廓线与样本进行匹配，作为该样本的标签；

S13.对样本集和标签集分别进行归一化处理，再将数据集分为训练数据集和测试数据集。

具体地，通过气象数据网下载或者基于大气数值模式和RT模型模拟，得到二维大气观测亮温图像；通过气象数据网下载或者WRF数值预报模型计算，得到三维大气温度廓线。

具体地，所述DenseNet卷积神经网络包括：

输入层，用于提取样本特征，所述输入层采用“Conv2d+BN+Relu”结构；

中间层，用于对样本特征提取更多维的特征，所述中间层包括：DenseBlock_1、DenseBlock_2和连接层，所述DenseBlock_1包括6个卷积层，所述连接层采用“BN+Conv2d+Relu”结构，所述DenseBlock_2包括12个卷积层；

输出层，用于将中间层输出的特征的通道数调整为与三维大气温度廓线的通道数，所述输出层采用“Conv2d”结构。

具体地，将数据集的时间顺序打乱，并选择前M个样本作为测试数据集，其余作为训练数据集。

为实现上述目的，按照本发明的第二方面，提供了一种基于DenseNet卷积神经网络的三维大气温度廓线反演系统，该系统包括：

预处理模块，用于根据氧气吸收频段的二维大气观测亮温图像和三维大气温度廓线，构建训练数据集；

训练模块，用于基于训练数据集，训练DenseNet卷积神经网络直至其收敛，得到训练好的网络；

反演模块，用于将待反演的亮温图像输入训练好的网络，输出反演得到的三维大气温度廓线。

具体地，所述预处理模块通过以下方式构建训练数据集：

(1)将频率不同、时间和区域相同的二维大气观测亮温图像进行组合，每组多频率亮温图像作为一个样本，不同样本之间选择的频率相同，时间和区域不同；

(2)根据区域的经纬度，将三维大气温度廓线与样本进行匹配，作为该样本的标签；

(3)对样本集和标签集分别进行归一化处理，再将数据集分为训练数据集和测试数据集。

具体地，所述DenseNet卷积神经网络包括：

输入层，用于提取样本特征，所述输入层采用“Conv2d+BN+Relu”结构；

中间层，用于对样本特征提取更多维的特征，所述中间层包括：DenseBlock_1、DenseBlock_2和连接层，所述DenseBlock_1包括6个卷积层，所述连接层采用“BN+Conv2d+Relu”结构，所述DenseBlock_2包括12个卷积层；

输出层，用于将中间层输出的特征的通道数调整为与三维大气温度廓线的通道数，所述输出层采用“Conv2d”结构。

总体而言，通过本发明所构思的以上技术方案，能够取得以下有益效果：

(1)相对于BP神经网络的数据集由单个亮温点构成，其全部数据集也仅为一幅亮温图像上的部分区域，本发明通过卷积神经网络反演温度廓线，网络训练所用数据集以二维亮温图像为单位，每个亮温图像覆盖了地球上的一定区域，整个数据集跨越了很长的时间间隔，因此，一旦卷积神经网络经训练达到收敛时，其泛化能力将大大提高，反演误差也会降低。

(2)相对于BP神经网络结构简单，仅可用作简单问题的回归模型，为提高反演精度，常将晴空、云以及雨天三种场景的数据分离，本发明中反演网络为DenseNet卷积神经网络，网络层数较深，同时因其密集连接的结构避免了训练过程中的梯度消失问题，适用于复杂的反演问题，因此本发明中反演网络的输入为原始的亮温数据，可实现直接对晴空、云以及雨天三种场景的数据一起反演，减少耗时。

附图说明

图1为本发明实施例提供的一种基于DenseNet卷积神经网络的三维大气温度廓线反演方法流程图；

图2为本发明实施例提供的DenseNet卷积神经网络结构示意图；

图3(a)为本发明实施例提供的晴空场景对应神经网络反演温度廓线的误差和DenseNet网络反演混合场景的误差；

图3(b)为本发明实施例提供的雨场景对应神经网络反演温度廓线的误差和DenseNet网络反演混合场景的误差；

图3(c)为本发明实施例提供的云场景对应神经网络反演温度廓线的误差和DenseNet网络反演混合场景的误差；

其中，横坐标为温度误差，纵坐标为不同高空对应的气压值，实线为DenseNet网络反演温度廓线的RMSE曲线。

具体实施方式

为了使本发明的目的、技术方案及优点更加清楚明白，以下结合附图及实施例，对本发明进行进一步详细说明。应当理解，此处所描述的具体实施例仅仅用以解释本发明，并不用于限定本发明。此外，下面所描述的本发明各个实施方式中所涉及到的技术特征只要彼此之间未构成冲突就可以相互组合。

如图1所示，本发明提供了一种基于DenseNet卷积神经网络的三维大气温度廓线反演方法，该方法包括以下步骤：

步骤S1.获取温度反演对应的氧气吸收频段的二维大气观测亮温图像T_A数据和对应三维大气温度廓线T数据。

氧气吸收通道频率对应亮温用于反演温度廓线，这些通道的氧气吸收系数和温度变化相关性很强。可通过NOAA数据网站、中国气象数据网等网络途径下载真实卫星历史数据，或者，基于大气数值模式和RT模型(DOTLRT、RTTOV等辐射传输模型)模拟得到观测亮温。温度廓线数据可由气象数据网站获取，亦可经WRF数值预报模型计算得到。

实施例中，由NOAA数据网站下载FNL历史再分析资料，选择太平洋上预设时段内无台风的海域，将之输入WRF数值预报模式计算得到大气状态参数。设置区域网格大小为200*200，网格分辨率为15000m*15000m，设置大气廓线垂直分层为59。将WRF输出的大气参数输入DOTLRT辐射传输模型中，设置仿真中心频率为50.3GHz、51.76GHz、52.8GHz、53.596GHz、54.4GHz、54.94GHz、55.5GHz和57.29GHz，得到模拟亮温T_B。利用卫星微波载荷对应不同频率的天线方向图对T_B进行卷积操作，并加入模拟的观测噪声，最终得到接近真实观测亮温的模拟亮温T_A。

步骤S2.根据获取的二维观测亮温图像T_A和三维大气温度廓线T，构建训练数据集。

选择氧气吸收频段的二维大气观测亮温图像T_A所有频率的组合作为网络的输入数据集，用于温度反演频段有两个，本实施例选择50-60GHz频段，所述亮温中心频率分别为50.3GHz、51.76GHz、52.8GHz、53.596GHz、54.4GHz、54.94GHz、55.5GHz和57.29GHz，将每一组T_A数据分别集合至一个mat格式的文件中，作为一个训练样本。

选择与亮温时间相同的三维温度廓线T，亮温图像中每一点都有经纬度标识，且每一点都对应一条垂直温度廓线，温度廓线按照标准气压进行分为59层，其接近地表的气压为1000hPa，最高空可达50hPa，依据经纬度将T与T_A进行匹配，使得二者具有相同的网格数，将T作为训练数据集的标签，并分别保存每个标签所有气压层的数据至mat格式文件。

对样本集和标签集分别进行归一化处理，将数据集打乱顺序，并选择前M个样本作为测试数据集，其余作为训练数据集。

实施例中，训练样本多，且单个样本与标签的数据量较大，为方便网络训练时数据集的读取，分别将多频率亮温和多层廓线组合得到训练集。分别读取8个频率的200*200大小亮温，并保存至同一mat格式的文件，使得单个样本的大小8*200*200。从WRF输出的大气状态文件中提取温度廓线，并按照气压由低到高的分层顺序保存温度廓线数据，使得单个标签的大小为59*200*200。根据亮温和温度数据的特征，分别对其除以150和50，做归一化处理。打乱数据集，选择前50组作为测试数据集，剩余作为训练所用数据。

步骤S3.基于训练数据集训练DenseNet卷积神经网络直至其收敛，得到训练好的网络。

DenseNet是一种应用于计算机视觉领域的卷积神经网络，其特点在于在网络中建立了当前的卷积层与之前卷积层之间的直接连接，实现了卷积层之间的特征重用，在减少参数量的同时缓解了梯度消失的问题。

本发明以DenseNet网络的密集连接块为基础搭建网络，并将原有的多个DenseBlock压缩至两个，减少网络层数，避免网络过拟合情况的出现，同时为应用于本发明所述的反演问题，还对输入和输出层做了修改，具体如下：

利用深度学习框架，搭建DenseNet网络基础结构DenseBlock，DenseBlock将网络当前层之前所有层的输出作为当前层的输入，设网络第i层的输入与输出变换关系为H_i，输出为y_i，则

y_i＝H_i(y₀+y₁+…+y_i-1)

网络的输入层为具有8个通道的亮温图像数据，输出层为具有59个通道(对应廓线分层)的温度廓线数据，且每个通道的亮温与廓线网格尺寸相等。本发明根据亮温多谱特征和温度廓线的分层特征修改了DenseNet输入和输出卷积层，并在减少中间卷积层数的同时保持网络特征重用的基本特性不变。

如图2所示，DenseNet网络结构中，“Conv2d”代表卷积层，“BN”代表归一化层，即将上层数据做归一化处理，可加快网络训练，“Relu”代表深度学习中的Relu激活函数。

在整个网络的计算过程中，网络的特征图大小保持不变；输入层中卷积层对应输入的8个亮温通道，并提取32维特征；整个网络主要由两个DenseBlock密集连接块构成，其中DenseBlock_1中含6个基础的卷积层，而DenseBlock_2中含12个卷积层，两个DenseBlock间采用“BN+Conv2d+Relu”的结构进行连接，网络经DenseBlock模块后共含有56维特征，此时需输出层的卷积操作提取出59维特征，对应廓线的59层垂直分层。

利用DenseNet网络反演温度廓线时，因网络的复杂程度可满足多种天气场景同时反演的需求，且网络本身的输入就是二维亮温图像，故无需再用背景场亮温数据或温湿度等大气参数筛选不同场景的亮温数据并作数据分类处理，符合实际反演时的情况，速度快、效率高，且其反演精度相比较利用神经网络反演的误差在可接受范围内，具有实用价值。

利用python语言编写训练所用程序，设置网络批处理量、最大迭代次数、训练平台(CPU或GPU)以及网络权重存储位置，训练网络，待网络输出的损失函数值不再下降，而于某一相比初始损失函数很小的值附近波动时，则判定针对当前数据集，网络已达到收敛，停止训练。

步骤S4.将待反演的亮温图像T_A输入训练好的网络，输出反演得到的三维大气温度廓线T。

对网络输出的温度廓线数据需要进行反归一化处理，方能同真实温度廓线作对比。

利用python语言编写测试程序，选择一测试样本输入训练好的网络，将网络输出的温度廓线同真实温度廓线对比，计算二者在每一层的RMSE值，利用该方法反演时未作数据的筛选，反演区域内包括晴空、云和雨三种不同的场景。将利用卷积神经网络反演得到的温度廓线误差同传统的人工神经网络训练所得误差对比，在本实例中卷积神经网络的测试样本为测试集中随机挑选的一个，且经测试其它测试样本(本实例中测试样本最大时间差为一个月)的误差曲线与此样本的计算结果较为接近，而下文所述神经网络的测试样本只与其训练样本相差6个小时，因为神经网络的测试集和训练集时间相差较大时，误差将会增大，在本实例中若选择相差15天的训练和测试样本，则神经网络的反演误差最大时可超过3.5K，且反演误差曲线没有分布规律，无法应用于反演问题。

图3(a)中虚线为利用神经网络反演晴空场景时温度廓线的误差，通过比较可知在低空区DenseNet反演混合场景的平均误差和神经网络反演晴空场景的误差精度相当，而在200hPa至800hPa的部分区间，DenseNet反演精度更高。

图3(b)中虚线为利用神经网络反演雨天场景时温度廓线的误差，在气压于800hPa以上的低空区，神经网络反演的误差更小，在其他区域二者误差相当。

图3(c)中虚线为利用神经网络反演有云场景时温度廓线的误差，由图可知神经网络反演时的整体误差小于DenseNet，但二者之间误差相差不大。

本发明与传统BP神经网络温度廓线反演方法的区别主要在于：其一，针对亮温与温度的非线性关系，本发明应用卷积神经网络进行反演，且网络深度远比传统人工神经网络大，可适用于更为复杂的关系模型，网络的泛化能力更强；其二，本发明中网络训练数据皆以某一区域的二维观测亮温图像和三维大气温度廓线为单元，数据集可包含多种天气状况(晴空、云、雨)，更加适用于真实应用场景。经过试验和分析，本发明能较好地满足反演温度廓线的实际需要。

本领域的技术人员容易理解，以上所述仅为本发明的较佳实施例而已，并不用以限制本发明，凡在本发明的精神和原则之内所作的任何修改、等同替换和改进等，均应包含在本发明的保护范围之内。

Claims (8)

1.一种基于DenseNet卷积神经网络的三维大气温度廓线反演方法，其特征在于，该方法包括以下步骤：

S1.根据氧气吸收频段的二维大气观测亮温图像和三维大气温度廓线，构建训练数据集；

S2.基于训练数据集，训练DenseNet卷积神经网络直至其收敛，得到训练好的网络；

S3.将待反演的亮温图像输入训练好的网络，输出反演得到的三维大气温度廓线。

2.如权利要求1所述的方法，其特征在于，步骤S1包括以下子步骤：

S11.将频率不同、时间和区域相同的二维大气观测亮温图像进行组合，每组多频率亮温图像作为一个样本，不同样本之间选择的频率相同，时间和区域不同；

S12.根据区域的经纬度，将三维大气温度廓线与样本进行匹配，作为该样本的标签；

S13.对样本集和标签集分别进行归一化处理，再将数据集分为训练数据集和测试数据集。

3.如权利要求1或2所述的方法，其特征在于，通过气象数据网下载或者基于大气数值模式和RT模型模拟，得到二维大气观测亮温图像；通过气象数据网下载或者WRF数值预报模型计算，得到三维大气温度廓线。

4.如权利要求2所述的方法，其特征在于，所述DenseNet卷积神经网络包括：

输入层，用于提取样本特征，所述输入层采用“Conv2d+BN+Relu”结构；

中间层，用于对样本特征提取更多维的特征，所述中间层包括：DenseBlock_1、DenseBlock_2和连接层，所述DenseBlock_1包括6个卷积层，所述连接层采用“BN+Conv2d+Relu”结构，所述DenseBlock_2包括12个卷积层；

输出层，用于将中间层输出的特征的通道数调整为与三维大气温度廓线的通道数，所述输出层采用“Conv2d”结构。

5.如权利要求2所述的方法，其特征在于，将数据集的时间顺序打乱，并选择前M个样本作为测试数据集，其余作为训练数据集。

6.一种基于DenseNet卷积神经网络的三维大气温度廓线反演系统，其特征在于，该系统包括：

预处理模块，用于根据氧气吸收频段的二维大气观测亮温图像和三维大气温度廓线，构建训练数据集；

训练模块，用于基于训练数据集，训练DenseNet卷积神经网络直至其收敛，得到训练好的网络；

反演模块，用于将待反演的亮温图像输入训练好的网络，输出反演得到的三维大气温度廓线。

7.如权利要求6所述的系统，其特征在于，所述预处理模块通过以下方式构建训练数据集：

(1)将频率不同、时间和区域相同的二维大气观测亮温图像进行组合，每组多频率亮温图像作为一个样本，不同样本之间选择的频率相同，时间和区域不同；

(2)根据区域的经纬度，将三维大气温度廓线与样本进行匹配，作为该样本的标签；

(3)对样本集和标签集分别进行归一化处理，再将数据集分为训练数据集和测试数据集。

8.如权利要求6所述的系统，其特征在于，所述DenseNet卷积神经网络包括：

输入层，用于提取样本特征，所述输入层采用“Conv2d+BN+Relu”结构；

中间层，用于对样本特征提取更多维的特征，所述中间层包括：DenseBlock_1、DenseBlock_2和连接层，所述DenseBlock_1包括6个卷积层，所述连接层采用“BN+Conv2d+Relu”结构，所述DenseBlock_2包括12个卷积层；

输出层，用于将中间层输出的特征的通道数调整为与三维大气温度廓线的通道数，所述输出层采用“Conv2d”结构。

Images