结合难分样本生成与学习的SAR图像目标开集识别方法技术

技术编号:37272545 阅读:11 留言:0更新日期:2023-04-20 23:41
本发明专利技术公开一种结合难分样本生成与学习的SAR图像目标开集识别方法,其实现步骤为:生成训练集和测试集;构建卷积神经网络并结合难分样本进行训练;根据训练集图像类别,设置开集识别卷积神经网络的拒判阈值;获取待识别SAR图像目标的预测概率分布;将预测概率分布中最大值大于拒判阈值的样本识别为最大值对应的类别,小于拒判阈值的样本拒判为未知目标类。本发明专利技术解决了现有技术数据适应性差和拒判阈值设置敏感的问题,具有对库内已知目标类的高精度识别和对库外未知目标类的快速准确发现的优点。现的优点。现的优点。

【技术实现步骤摘要】
结合难分样本生成与学习的SAR图像目标开集识别方法


[0001]本专利技术属于雷达通信
,更进一步涉及雷达自动目标识别
中的一种结合难分样本生成与学习的合成孔径雷达SAR(Synthetic Aperture Radar)目标开集识别方法。本专利技术可用于雷达干扰环境为开放场景下识别SAR图像目标的类型。

技术介绍

[0002]雷达自动目标识别技术RATR(Radar Automatic target recognition)是典型的开放场景识别任务,其面向的是合作目标与非合作目标,类别众多、型号繁杂,训练阶段很难建立完备的目标识别库,即测试阶段存在未知类目标。传统的模式识别技术是在闭集假设条件下设计的,即假定训练样本中出现的类别标记集合包含所有待识别样本的类别标记,不存在未知类目标。在真实的开放环境中,当库外未知类目标进入到模型时,闭集识别模型会强制将未知类目标错分为某一库内类别,这极大地限制了模型在开放环境中的应用。因此,期望一个能够识别/拒判未知类目标同时保持对已知类目标识别性能的分类器,在雷达自动目标识别中实现对输入的已知类样本,输出为具体的某个类别,对输入的未知类样本,拒判为未知类类别。在开放环境中感知未知类目标,正确且自适应地检测到未知类目标的存在并识别已知类目标是雷达开放环境下识别的前提和基础。
[0003]Zeng Zhiqiang等人在其发表的论文“Unknown SAR Target Identification Method Based on Feature Extraction Network and KLD

RPA Joint Discrimination”(Remote Sensing,2021,7)中提出了一种结合相对熵与相对位置角度量的SAR图像目标开集识别方法。该方法包含一个特征提取器和分类器。特征提取器输入训练集的图像,输出128维高维特征,分类器输入128维高维特征,输出特征对应的预测概率分布。该方法实现步骤是,第一步,将训练集的所有图像输入特征提取器得到高维特征,高维特征输入分类器得到预测概率分布,最小化预测概率分布与真实标签之间的交叉熵损失;第二步,计算训练集不同类别图像128维特征的均值以及测试集对应特征相较于训练集特征的相对角度范围。在测试阶段,将待识别图像目标输入到特征提取器,输出128维高维特征。当128维高维特征与训练集各类别特征均值的相对熵均大于预设相对熵阈值,或者特征相较于训练集特征的相对角度范围大于预设相对角度阈值时,模型将待识别图像目标拒判为未知目标类,否则送入分类器得到预测概率分布,识别为最大概率预测值对应的类别。该方法存在的不足之处是,训练阶段需要借助测试集图像信息预设两个合适的阈值,由于在工程实践中无法提前获取测试集图像类别,因此,该方法的拒判阈值设置策略敏感性高且数据适应性差,远不能满足真实开放场景下工程实践的要求。
[0004]西安电子科技大学在其申请的专利文献“一种雷达高分辨距离像开集识别方法及装置”(专利申请号:CN202111199838.4,申请公布号CN 114137518 A)中公开了一种雷达目标开集识别方法。该方法采用卷积神经网络技术构建各类别原型向量增强类内聚合性,同时组合各层的初级特征,从而得到更高层的特征进行目标识别,因此专利技术对已知目标类的识别率有显著性提高。在测试阶段,当待识别目标的最大预测概率值小于人为设定的阈值
时,模型将待识别目标拒判为未知类目标,否则,识别为最大预测概率值对应的类别。该方法存在的不足之处是,训练阶段仅关注已知目标类的识别性能,无法有效地降低模型地开放空间风险,难以建立良好的拒判界面,直接应用于真实开放场景时会产生严重的混淆,远不能满足工程实践过程中识别性能的要求。

技术实现思路

[0005]本专利技术的目的在于针对上述现有技术存在的不足,提出了一种结合难分样本生成与学习的SAR图像目标开集识别方法,旨在解决现有技术数据适应性差和拒判阈值设置敏感的问题。
[0006]实现本专利技术目的的思路是,本专利技术使用训练集图像和噪声序列,计算特征空间易混淆的难分样本特征,生成数据空间高质量的难分图像,组成能够表征实际开放空间中未知目标类统计特性的难分样本。网络对难分样本的学习增强了未知目标类的感知能力,缓解了现有技术无法降低开放空间风险而引起的数据适应性差的问题。本专利技术根据训练集图像的统计特性自动确立拒判阈值,缓解了现有技术人为设置拒判阈值而出现的敏感性高的问题。本专利技术设计的难分样本生成与学习方法克服了现有技术数据适应性差和拒判阈值设置敏感的问题,实现了库内已知目标类的准确识别和库外未知目标类的精准拒判。
[0007]本专利技术采取的技术方案包括如下步骤:
[0008]步骤1,生成训练集:
[0009]将样本集中雷达工作俯仰角为17
°
共2746张图像中的每张图像均裁剪为64
×
64个像素的图像,所选取的所有图像包含10个类别,在类别标签[1,10]的范围内,对每张图像的目标类别进行标注,再将所有裁剪后的图像和对应类别标签组成训练集;
[0010]步骤2,构建开集识别卷积神经网络:
[0011]步骤2.1,构建生成器子网络:
[0012]搭建一个由五个反卷积层串联组成的生成器子网络。将第一至第四反卷积层的卷积核数量依次设置为1024、512、256、128和3,卷积核大小均设置为4
×
4,卷积核步长依次设置为1、2、2、2和2,填充方式均设置为等大填充方式,偏差置均设置为0;
[0013]步骤2.2,构建判别器子网络:
[0014]搭建一个由四个卷积层串联组成的生成器子网络。将第一至第四卷积层的卷积核数量依次设置为128、256、512和1024,卷积核大小均设置为4
×
4,卷积核步长依次设置为1、1、1和0,填充方式均设置为等大填充,偏差置均设置为0;
[0015]步骤2.3,构建分类器子网络:
[0016]构建一个由一组卷积层组和一个全连接层串联组成的分类器子网络,其中,一组卷积层组由九个卷积层串联组成;将第一至第九卷积层的卷积核数量依次设置为64、64、128、128、128、128、128、128和128,卷积核大小均设置为3
×
3,卷积核步长依次设置为1、1、2、1、1、2、1、1和2,填充方式均设置为等大填充,偏差置均设置为0;将全连接层参数设置为128
×
K,K表示训练样本的总类别数,偏差置均设置为0;
[0017]步骤2.4,将生成器子网络和判别器子网络并联后再与分类器子网络串联,构成开集识别卷积神经网络;
[0018]步骤3,结合难分样本训练开集识别卷积神经网络:
[0019]步骤3.1,将从标准正态分布中随机采样的噪声序列,输入到开集识别卷积神经网络中,经生成器子网络后输出的噪声生成图像,经判别器子网络后输出每张噪声生成图像的判别置信得分;将训练集的每张图像输入本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.一种结合难分样本生成与学习的SAR图像目标开集识别方法,其特征在于,通过难分样本的生成与学习,训练构建的用于输出SAR图像目标预测概率分布的卷积神经网络;该识别方法的步骤包括如下:步骤1,生成训练集:将样本集中雷达工作俯仰角为17
°
共2746张图像中的每张图像均裁剪为64
×
64个像素的图像,所选取的所有图像包含10个类别,在类别标签[1,10]的范围内,对每张图像的目标类别进行标注,再将所有裁剪后的图像和对应类别标签组成训练集;步骤2,构建开集识别卷积神经网络:步骤2.1,构建生成器子网络:搭建一个由五个反卷积层串联组成的生成器子网络;将第一至第四反卷积层的卷积核数量依次设置为1024、512、256、128和3,卷积核大小均设置为4
×
4,卷积核步长依次设置为1、2、2、2和2,填充方式均设置为等大填充方式,偏差置均设置为0;步骤2.2,构建判别器子网络:搭建一个由四个卷积层串联组成的生成器子网络;将第一至第四卷积层的卷积核数量依次设置为128、256、512和1024,卷积核大小均设置为4
×
4,卷积核步长依次设置为1、1、1和0,填充方式均设置为等大填充,偏差置均设置为0;步骤2.3,构建分类器子网络:构建一个由一组卷积层组和一个全连接层串联组成的分类器子网络,其中,一组卷积层组由九个卷积层串联组成;将第一至第九卷积层的卷积核数量依次设置为64、64、128、128、128、128、128、128和128,卷积核大小均设置为3
×
3,卷积核步长依次设置为1、1、2、1、1、2、1、1和2,填充方式均设置为等大填充,偏差置均设置为0;将全连接层参数设置为128
×
K,K表示训练样本的总类别数,偏差置均设置为0;步骤2.4,将生成器子网络和判别器子网络并联后再与分类器子网络串联,构成开集识别卷积神经网络;步骤3,结合难分样本训练开集识别卷积神经网络:步骤3.1,将从标准正态分布中随机采样的噪声序列,输入到开集识别卷积神经网络中,经生成器子网络后输出的噪声生成图像,经判别器子网络后输出每张噪声生成图像的判别置信得分;将训练集的每张图像输入到开集识别卷积神经网络中,经判别器子网络后输出每张图像的判别置信得分;利用交叉熵损失函数,计算每张图像的判别置信得分与该图像对应的域标签之间的损失值,得到判别器子网络的总损失函数;使用Adam优化算法,更新当前判别器子网络参数;步骤3.2,将重新从标准正态分布中随机采样的噪声序列,输入到开集识别卷积神经网络中,经生成器子网络后输出的噪声生成图像,经判别器子网络后输出每张噪声生成图像的判别置信得分,经分类器子网络后输出每张噪声生成图像的预测概率分布;利用交叉熵损失函数,计算每张噪声生成图像的预测概率分布与该图像对应的类别标签之间的损失值,计算每张噪声生成图像的判别置信得分与域标签之间的损失值,得到生成器子网络的总损失函数;使用Adam优化算法,更新当前生成器子网络参数;步骤3.3,将训练集的每张图像输入到开集识别卷积神经网络中,经分类器子网络后输出每张图像的已知目标类特征,经梯度下降的难分样本特征生成算法计算每张图像对应的
难分样本特征;步骤3.4,将重新从标准正态分布中随机采样的噪声序列,输入到开集识别卷积神经网络中,经生成器子网络后输出噪声生成图像;噪声生成图像经过判别器子网络后,根据判别置信得分生成判别置信得分较高的难分图像;难分图像经过分类子网络后输出每张难分图像的难分图像特征;难分图像特征和难分样本特征共同组成难分特征集,已知目标类特征组成训练特征集;步骤3.5,将训练特征集和难分特征集输入到开集识别卷积神经网络中,经分类器子网络后输出每个特征对应的预测概率分布;利用交叉熵损失函数,分别计算三个损...

【专利技术属性】
技术研发人员:杜兰李逸明陈健宋佳伦雍奇锋周宇
申请(专利权)人:西安电子科技大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1