[发明专利]一种基于变分自编码矿物信息定量分析方法

权利要求书

1.一种基于变分自编码矿物信息定量分析方法，其特征在于：包括以下步骤：

步骤一：采集研究区高光谱数据，读取多种矿物混合的光谱作为训练样本光谱；

步骤二：根据光谱波段数、矿物种类构建五层全连接神经网络；所述五层全连接神经网络，包括：全连接神经网络输入层、全连接神经网络第一隐含层、全连接神经网络第二隐含层、全连接神经网络第三隐含层和全连接神经网络输出层；

步骤2.1：计算每条训练样本光谱对应的全连接神经网络第一隐含层特征；

步骤2.2：计算全连接神经网络第二隐含层特征；

步骤2.3：计算每个训练样本光谱对应的全连接神经网络重构样本光谱；

步骤2.4：构建全连接神经网络的损失函数；

所述步骤二中全连接神经网络输入层的神经元个数为光谱波段数，每个波段的反射率为全连接神经网络输入层的神经元；设置全连接神经网络第三隐含层神经元数为矿物种类数，全连接神经网络输出层为重构光谱；

所述步骤2.1中，计算每条训练样本光谱对应的全连接神经网络第一隐含层特征公式如下：

其中，为第i个训练样本光谱对应的第一隐含层特征，Relu表示修正线性单元RectifiedlinearUnits，W¹为输入层映射到第一隐含层的权重系数矩阵，b¹为输入层映射到第一隐含层的偏置向量；x_i为第i个训练样本光谱；

所述的步骤2.2中，计算全连接神经网络第二隐含层特征的公式如下式(2)：

其中，为第i个训练样本光谱对应的第二层隐含层特征，W²为第一隐含层映射到第二隐含层的权重系数矩阵，b²为第一隐含层映射到第二隐含层的偏置向量；为第i个训练样本光谱对应的第一隐含层特征；

所述的步骤2.3中，计算每个训练样本光谱对应的全连接神经网络重构样本光谱，计算公式如下式(3)：

其中，为第i个训练样本光谱对应的重构样本，sigmoid为输出层激活函数，W³为第二隐含层映射到第三隐含层的权重系数矩阵，b³为第二隐含层映射到第三隐含层的偏置向量；为第i个训练样本光谱对应的第二隐含层特征，W⁴为第三隐含层映射到重构样本输出层的权重系数矩阵，b⁴为第三隐含层映射到重构样本输出层的偏置向量；

所述的步骤2.4中，构建全连接神经网络的损失函数如下式(4)：

其中，L为全连接神经网络的损失函数，|| ||表示取模值操作，arccos表示的是反三角函数中的反余弦，为第i个训练样本光谱对应的重构样本；x_i为第i个训练样本光谱；

步骤三：将每一条样本光谱转化为一个列向量，并进行光谱归一化处理；输入到全连接神经网络进行训练，获得矿物端元光谱的初始值；

步骤四：构建变分自编码神经网络；变分自编码神经网络包括：变分自编码神经网络输入层、变分自编码神经网络第一隐含层、变分自编码神经网络特征层、变分自编码神经网络第二隐含层和变分自编码神经网络输出层；

步骤4.1：计算步骤一中每条训练样本光谱对应特征的均值；

步骤4.2：计算步骤一中每条训练样本光谱对应特征的标准差；

步骤4.3：计算步骤一中每条训练样本光谱对应的特征值；

步骤4.4：计算步骤一中每条训练样本光谱对应的重构样本光谱；

步骤4.5：构建变分自编码神经网络的损失函数；

所述步骤四还包括：变分自编码神经网络输入层神经元个数为光谱的波段数，每个波段的反射率为变分自编码神经网络输入层的神经元；变分自编码神经网络特征层计算出每个样本的特征的均值和标准差，并进行重参数处理，变分自编码神经网络输出层为重构光谱；

步骤五：训练变分自编码神经网络，将变分自编码神经网络第二隐含层映射到变分自编码神经网络重构样本输出层的权重系数矩阵参数设置为步骤三中矿物端元光谱的初始值；

步骤5.1：对变分自编码神经网络的参数初始化赋值，隐含层映射到重构样本输出层的权重系数矩阵参数设置为步骤三获得的矿物端元光谱的初始值；

步骤5.2：将步骤一中每一条训练样本光谱转化为一个列向量输入到对变分自编码神经网络进行训练，获得更新后的对变分自编码神经网络参数；

步骤5.3：判断循环次数是否为50次，如果循环次数为是，则执行步骤5.4，如果循环次数为否，则训练次数加1次后执行步骤5.2；

步骤5.4：获得变分自编码神经网络隐含层映射到重构样本输出层的权重系数矩阵为矿物端元光谱，获得变分自编码神经网络特征层为矿物含量值。