基于自监督学习和子孔径分解的SAR目标识别方法技术

本发明专利技术为一种基于自监督学习和子孔径分解的SAR目标识别方法，将大量杂波数据加入原始SAR图像数据集中进行扩充；将数据集中图像进行包括子孔径分解在内的两种数据扩充后输入到以编码器和动量编码器为主体的两个分支；分别采用梯度回传和动量公式更新两个编码器的参数；以队列形式更新动量编码器的输入后重复以上步骤直至遍历训练集中所有数据，并保存模型参数；最后加载自监督的模型参数，重置分类器部分参数完成有监督的下游任务的训练和测试。本发明专利技术相比于有监督的SAR图像识别方法具有需要标记数量少，检测精度高的优点，适用于SAR图像在小样本情形下的目标识别，主要解决了现有的SAR图像目标识别技术中样本数量不足以满足大规模有监督训练的问题。足以满足大规模有监督训练的问题。足以满足大规模有监督训练的问题。

【技术实现步骤摘要】
基于自监督学习和子孔径分解的SAR目标识别方法


[0001]本专利技术属于雷达目标识别
，尤其涉及一种基于自监督学习和子孔径分解的SAR目标识别方法。

技术介绍

[0002]作为遥感技术的一种，合成孔径雷达(Synthetic Aperture Radar,SAR)是一种主动微波成像雷达，并由于其成像原理，相较于其他各种遥感成像技术具有更加不受天气及光照条件影响的优点，具有全天时、全天候的优良特性。但同时由于其成像机制的特殊性，图像中有大量杂波存在。同时应用于车辆、飞机等目标识别任务时的标签标记需要众多的相关经验。因此缺少大量有标记图像是当前SAR图像处理领域的一个重要问题。
[0003]当前随着机器学习相关理论和模型的发展，深度神经网络被应用于目标识别任务中，获得了越来越令人惊喜的结果。也有越来越多的人将机器学习应用于SAR图像处理中，并取得了一定成果。然而为达到较好的效果，在网络训练时需要使用大量有标记图像，然而SAR图像在目标识别任务中可用于训练的数据较少，容易产生网络过拟合的现象，无法达到最优的结果。
[0004]为减少在训练中对有标签数据的依赖，深度学习发展出了半监督学习、弱监督学习等一系列不同于全监督学习的方法，在网络的训练过程中加入无标签数据帮助网络学习相应特征，减少有标签数据的使用，同时也能够提高网络的泛化性能。然而基于以上方法的深度学习任务依旧对有标签数据的数量有一定的依赖，并且其本质原理依旧是在做监督学习，不能关注到数据本身。而自监督学习通过各种数据变换为无标记的数据创造“伪标签”，并以此进行网络的训练，在不依赖人工标记的同时依然能够达到与监督学习相似的效果。
[0005]与有监督学习依赖标签的特性不同，自监督学习能够降低注释大规模数据集的成本。它采用自定义的伪标记来监督，并将学习的表示形式用于多个下游任务，并在对下游任务的微调中体现自监督模型的效果。自监督学习首先在上游任务部分从大规模的无标签数据中挖掘数据自身的监督信息，即“伪标签”，通过使用“伪标签”对网络进行有监督的训练，使网络学习到对下游任务有用的表征。之后将自监督训练的模型应用于下游任务，在下游任务中采用极少量的有标签数据进行微调，同时，为体现自监督模型的效果，在训练时固定下游任务中网络模型的大部分参数，例如在分类任务中，只改变分类器部分的参数而其他部分参数固定。
[0006]多年来，SAR图像缺少大量有标记样本的问题未能被完全解决，而基于自监督学习的方法恰好能够解决相关困难。当前自监督学习在SAR图像领域主要被应用在变化检测及针对SAR图像的去噪方面，而面向SAR目标检测及识别的工作较少。同时直接将当前流行的自监督学习算法应用在SAR目标检测中虽然能起到一定效果，但并未与SAR图像的物理特性联系起来，不能发挥更好的作用。由于SAR本身独特的物理特性，使得其存在不计其数的杂波图像，不仅降低了图像的分辨率同时也影响标记标签工作的进行。而自监督学习中不需要标记数据，在进行自监督训练的过程中加入大量的杂波图像能在扩充数据集的同时对其
进行充分利用并使得网络能够学习到更好的表征。另一方面，作为合成孔径雷达，当前使用的图像也可以将其分解为相对应的子孔径图像，在自监督“伪标签”的生成中作为其中一种伪标签参与训练，由于其与原始图像的相似性，在自监督对比学习的过程中能够更好的促进网络对特征的学习。

技术实现思路

[0007]针对现有技术的不足，本专利技术拟解决的技术问题是，提出了一种基于自监督学习和子孔径分解的SAR目标识别方法。
[0008]本专利技术解决所述技术问题采用的技术方案如下：
[0009]1.一种基于自监督学习和子孔径分解的SAR目标识别方法，其特征在于，该方法包括如下步骤：
[0010](1)对原始SAR图像进行扩充；
[0011](2)对待训练数据x进行数据增强，通过两种不同的数据增强分别得到样本对x
query
和x
key
；
[0012](3)样本x
query
输入到编码器f
q
，然后经过多层感知器得到特征图q；样本x
key
输入到动量编码器f
k
中，得到特征图k；
[0013](4)计算特征图q和k的相似度，得到模型损失；
[0014](5)使用梯度回传更新编码器f
q
的参数，动量编码器f
k
的参数采用动量公式更新；
[0015](6)队列更新动量编码器的输入，将新的样本x
key
加入队列，同时将最老的样本x
key
移出，保证队列的长度不变；
[0016](7)重复步骤(2)
‑
步骤(7)，直到遍历训练集中所有数据，保存自监督模型参数。
[0017](8)加载自监督模型参数，重置分类器部分参数并固定其他层参数不变，输入训练集数据进行对下游任务的训练。
[0018](9)训练结束，保存参数并完成SAR目标识别测试。
[0019]与现有技术相比，本专利技术具有以下优点：
[0020]1.需要标记样本数据量少
[0021]现有的针对SAR图像识别任务的深度学习方法大多为有监督方法，而SAR图像数据集作为典型的小样本数据集，有标记图像难以达到普通光学图像的数据量，使得在有监督的训练过程中不仅容易产生过拟合的问题，而且也不能更好的完成SAR图像识别的任务。而本专利技术采用的基于对比式的自监督学习方法能够在上游任务阶段采用自身产生的“伪标签”而非来自外界的标签进行训练。此外，由于SAR自身独特从成像特性，其结果常含有大量杂波，使人无法对其中目标进行标记。为丰富自监督训练的训练数据，将大量有杂波的SAR图像加入其中，使网络拥有更好的泛化性能的同时充分利用到了这些杂波图像。
[0022]2.与SAR的物理特性相结合
[0023]SAR图像来自于合成孔径雷达，在深度学习的训练过程中大多被直接作为网络的输入，只采用了振幅信息，忽视了各个子孔径的信息，采用子孔径分解的方法能够得到相应的SAR图像内部的复杂信息。本专利技术将子孔径图像分解后合成伪彩色图像，并将其作为自监督阶段数据扩充的一种。由于伪彩色子孔径图像由原始SAR图像数据产生，与相应的SAR图像非常相似，而在自监督学习的过程中，将来自同一图像的数据增强认为是正样本对，来自
不同图像的数据增强是负样本对，因此采用子孔径分解后合成的伪彩色图像作为数据增强能够更好的使网络学习到不同图像之间的特征表达，使网络最终的识别精度有进一步的提升。
附图说明
[0024]图1是本专利技术的整体流程图；
[0025]图2是本专利技术实验中扩充自监督数据集采用的杂波图像示例；
[0026]图3是本专利技术实验中使用MSTAR的数据集中的原图及其对应的子孔径合成的伪彩色图像示例。
具体实施方式
[0027]下面结合附图给出具体实施例，具体实施例仅用于本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于自监督学习和子孔径分解的SAR目标识别方法，其特征在于，该方法包括如下步骤：(1)对原始SAR图像进行扩充；(2)对待训练数据x进行数据增强，通过两种不同的数据增强分别得到样本对x
query
和x
key
；(3)样本x
query
输入到编码器f
q
，然后经过多层感知器得到特征图q；样本x
key
输入到动量编码器f
k
中，得到特征图k；(4)计算特征图q和k的相似度，得到模型损失；(5)使用梯度回传更新编码器f
q
的参数，动量编码器f
k

【专利技术属性】
技术研发人员：王兆成，刘商艺，王若楠，刘璐，邵学彦，
申请(专利权)人：天津先进技术研究院，
类型：发明
国别省市：