[发明专利]基于自适应确定拒判准则的SAR目标开集识别方法

权利要求书

1.一种基于自适应确定拒判准则的SAR目标开集识别方法，其特征在于，构建一个用于输出SAR目标的预测类别概率的卷积自编码网络，通过自适应设置拒判阈值准则，检测SAR目标是否为未知类目标；该识别方法的步骤包括如下：

步骤1，生成训练集：

选取雷达工作俯仰角为17°的至少1000张SAR图像，对每张SAR图像进行裁剪和类别标记，将所有裁剪后的图像和对应类别标签组成训练集；

步骤2，构建卷积自编码模型开集识别模型：

步骤2.1，构建编码子网络：

搭建一个由五个具有相同结构的通道注意力模块串联组成的编码子网络；

每个通道注意力模块由两条并联支路组成：

第一支路的结构依次为：卷积层、Batch Norm批次归一化层、ReLU非线性层；将第一至第五模块中第一支路卷积层的卷积核数量依次设置为16、32、64、128、256，卷积核大小均设置为3×3，卷积核步长均设置为1，填充方式均设置为等大填充方式；

第二支路的结构依次为：全局平均池化层、第一卷积层、ReLU非线性层、第二卷积层、Sigmoid激活函数；将第一至第五模块中第二支路中第一卷积层的卷积核数量依次设置为4、8、16、32、64，卷积核大小均设置为1×1，卷积核步长均设置为1，填充方式均设置为等大填充方式；将第一至第五模块中第二支路第二卷积层的卷积核数量依次设置为16、32、64、128、256，卷积核大小均设置为1×1，卷积核步长均设置为1，填充方式均设置为等大填充方式；

步骤2.2，构建条件子网络：

搭建一个由三层全连接层组成的条件子网络；将全连接层的节点数依次设置为256、1024、4096；

步骤2.3，构建分类子网络：

搭建一个由两层卷积层组成的分类子网络；将卷积核数量依次设置为128、K；其中，K表示训练集类别；将卷积核大小均设置为1×1，卷积核步长均设置为1，填充方式均设置为等大填充方式；

步骤2.4，构建解码子网络：

搭建一个由四个反卷积层串联组成的解码子网络；将第一至第四反卷积层的卷积核数量依次设置为128、64、32和16，卷积核大小均设置为4×4，卷积核步长均设置为2，填充方式均设置为等大填充方式，偏差置均设置为0；

步骤2.5，将条件子网络和编码子网络并联得到子网络1，将分类子网络和解码子网络并联得到子网络2，再将子网络1和子网络2串联，构成基于卷积自编码的开集识别模型；

步骤3，训练开集识别模型：

步骤3.1，将训练集的每张图像输入到开集识别模型中，经编码子网络后输出训练集中每张图像的样本隐特征，经分类子网络后输出每张图像的预测类别概率；利用交叉熵损失函数，计算每张图像的预测类别概率与该图像对应的类别标签之间的交叉熵损失L_c：

步骤3.2，将训练集的每张图像类别标签输入到开集识别模型中，经条件子网络输出每张图像的条件隐特征；每张图像的样本隐特征与其对应的条件隐特征点乘后得到该张图像的匹配融合特征；匹配融合特征经解码子网络输出每张图像的匹配重构图像；每张图像的样本隐特征与其不对应的条件隐特征点乘后得到该张图像的非匹配融合特征；非匹配融合特征经解码子网络输出每张图像的非匹配重构图像；计算匹配重构图像与该张图像的重构误差，计算非匹配重构图像与随机选取的非该张图像类别的其他类别图像的重构误差，得到重构损失L_d：

步骤3.3，将步骤3.1计算得到的交叉熵损失和步骤3.2计算得到的重构损失求和，得到模型的总损失；使用Adam优化算法，迭代更新开集识别模型参数，直到总损失函数收敛为止，得到训练好的开集识别模型；

步骤4，利用极值理论自适应设置各类别的拒判阈值：

步骤4.1，将训练集中每张图像和类别标签输入到训练好的开集识别模型中，输出每张图像的匹配重构图像和非匹配重构图像；计算训练集各类别的匹配重构误差和非匹配重构误差；

步骤4.2，基于极值理论分别对训练集各类别匹配重构误差和非匹配重构误差进行极值分布拟合，输出训练集各类别的极值分布；

步骤4.3，遍历训练集中各类别的重构误差，得到该类别的极值分布极小值，将各类别的极值分布极小值对应的重构误差作为该类别的最优拒判阈值；

步骤5，获取待识别SAR图像的预测类别概率和重构图像：

采用与步骤1相同的方式，对待识别的SAR图像做裁剪，得到裁剪后的SAR图像，将裁剪后的SAR图像输入到训练好的卷积自编码网络中，输出SAR图像的预测类别概率和对应的重构图像，将预测类别概率中得分最高值对应的类别作为待识别SAR图像的预测类别；

步骤6，判断待识别SAR图像的重构误差是否小于预测类别的最优拒判阈值，若是，则执行步骤7，否则，执行步骤8；

步骤7，选择SAR图像预测类别概率得分最高的类别作为识别结果输出；

步骤8，将SAR图像判定为未知类目标输出。