APP下载

基于跨域元学习的SAR自动目标识别研究

2023-06-25王可乔琪

现代信息科技 2023年4期
关键词:知识迁移

王可 乔琪

摘  要:近年来,基于数据驱动的SAR自动目标识别研究技术取得了很大进展。虽然这类方法识别性能较好,但是在实际应用场景中很难采集到足够的真实SAR数据用于训练。文章通过引入仿真SAR数据来扩充训练数据集,弥补真实SAR数据的不足。为了解决仿真和真实SAR数据之间差异导致的跨域和跨任务迁移问题,提出了一种基于跨域元学习机制的知识迁移算法。利用特征变换和数据增强方法来解决跨领域迁移,利用元学习机制来解决跨任务迁移。实验证明算法可以在有限的训练数据下取得良好的识别性能。

关键词:合成孔径雷达;知识迁移;元学习

中图分类号:TP18;TN957.52 文献标识码:A 文章编号:2096-4706(2023)04-0057-04

Research on SAR Automatic Target Recognition Based on Cross-Domain Meta-Learning

WANG Ke, QIAO Qi

(Jiangsu Vocational College of Electronics and Information, Huai'an  223003, China)

Abstract: In recent years, SAR automatic target recognition research technology based on data driven has made great progress. Although the recognition performance of such methods is better, it is difficult to collect enough real SAR data for training. This paper expands the training data set by introducing simulation SAR data to make up for the shortage of real SAR data. To solve cross-domain and cross-task transfer problems caused by the differences between simulation and real SAR data, a knowledge transfer algorithm based on cross-domain Meta-Learning mechanism is proposed. Feature transformation and data enhancement are used to solve cross-domain transfer, and Meta-Learning mechanism is used to solve cross-task transfer. Experiments demonstrate that the algorithm can achieve good recognition performance with limited training data.

Keywords: synthetic aperture radar; knowledge transfer; Meta-Learning

0  引  言

合成孔径雷达(Synthetic Aperture Radar, SAR)是利用合成孔径技术来对地物目标进行高分辨率成像的传感器系统。SAR工作在微波波段,在军用和民用领域都得到了广泛应用。与光学图像相比,SAR图像存在一些缺陷,如分辨率相对较低、图像为缺乏对比度的灰度图等。这些缺陷使得SAR图像难以通过人工方式进行解译和分类,导致了SAR图像目标自动识别(Synthetic Aperture Radar Automatic Target Recognition, SAR-ATR)技术的产生[1]。目前,SAR-ATR是SAR图像应用中最重要的任务之一。

随着人工智能和大数据技术的迅猛发展,如深度学习这样的数据驱动方法被广泛应用于SAR-ATR领域中,并取得了优异的分类效果。复旦大学的Chen等人[2]首次将卷积神经网络(Convolutional Neural Network, CNN)引入到SAR-ATR领域,并设计了一种去除全连接层的全卷积网络(A-ConvNet)。文献[3]利用多通道CNN来提取多视角图像的特征图,这些特征图通过傅里叶特征融合模块处理后再进行分类。Min等人[4]采用知识蒸馏技术将网络结构复杂的大型CNN压缩为微型 CNN(Micro CNN, MCNN)。MCNN 结构简单,网络参数很少,所需要的内存和CPU资源都很少,可以部署在嵌入式设备上。文献[5]基于目标散射机理来挖掘并优选SAR图像极化特征,这些特征包含旋转不变特征和旋转域零角特征。这些优选出来的特征有助于训练CNN网络,提高了CNN的泛化性能。数据增强[6]也是一种提高CNN分类精度和稳健性的有效手段,可以在SAR训练集中添加了平移、加噪处理后的人工合成图像。此外,类别依赖结构保持投影(Class-Dependent Structure Preserving Projection, CDSPP)[7]和核稳健局部判别投影(Kernel Robust Locality Discriminant Projection, KRLDP)[8]等多种非线性投影方法被用于在变换域提取 SAR 图像特征。

近年来,迁移学习、元学习等知识迁移技术在光学图像领域取得成功。SAR-ATR领域的研究者们开始引入这类技术来解决训练数据有限条件下的目标识别问题。迁移学习引入外部数据集(光学图像或仿真SAR图像)并设计相应的知识迁移框架。与采集真实SAR图像相比,光学图像和仿真SAR图像的采集工作要方便得多,这有助于建立更加完备的,含有大量样本的训练数据集。Malmgren-Hansen等人[9]构建各种车辆目标的CAD模型,将CAD模型送入电磁计算软件得到仿真SAR圖像。CNN首先采用这些仿真图像进行预训练,然后利用真实图像微调所有的网络参数。文献[10]首先生成测试目标的精确仿真图像,然后采用平移、加噪、MIX-UP[11]等多种方法进行数据增强,仅利用很少的训练样本就可以获得很高的识别率。文献[12]在图像特征提取器中额外插入了一个特征变换层,在训练阶段通过仿射变换增强图像特征来模拟不同领域中的特征分布,有利于解决跨域知识迁移问题。

知识迁移可以从源域中挖掘先验知识,然后将这些知识迁移到目标域的分类任务中,从而解决训练样本不足导致的SAR-ATR方法识别率下降,训练稳定性恶化等问题。但是源域和目标域数据之间的差异(图像的背景、成像角度、干扰程度不同)导致了跨领域迁移问题。源域和目标域图像的目标类别不完全一致产生了跨任务迁移问题。本文提出了一种基于跨域元学习的一种识别算法,利用特征变换和数据增强方法来解决跨领域迁移问题,利用元学习机制来解决跨任务迁移问题。

1  算法介绍

1.1  算法框架

本文提出的跨域元学习识别算法框架由特征提取器和分类器组成。特征提取器负责提取对应领域图像的特征,分类器用于确定图像特征的所属类别,如图1所示。算法基于元学习的训练机制来解决任务迁移问题,所以模型的输入数据被分为支撑集和查询集。模型的每一次训练迭代分为元训练和元测试两个阶段。在元训练阶段,模型利用支撑集的图像和标签对特征提取器和分类器的参数进行更新。在元测试时,模型首先用Mix-Up方法[11]对查询集的数据进行变换增强,然后利用多个特征变换层(Feature-Wise Transform, FWT)[12]来强化特征提取器,并利用强化后的特征提取器来提取查询集的特征。特征变换层通常部署在卷积层之下,负责对卷积层输出的特征图进行尺度和平移变换,以增强特征稳健性。模型的元训练和元测试都在源域中进行,当训练完成后,模型利用少量目标域的数据来微调特征提取器和分类器的参数,就可以完成目标域中的识别任务。

算法框架中的特征提取器和分类器都是基于CNN实现的,其中特征提取器由多层卷积模块组成,分类器则是单层全连接层。如图2所示,特征提取器由4个卷积模块串接而成,依次对图像的特征进行提取和精炼。Conv代表卷积层,16×5×5说明该层含有16个卷积核,卷积核的尺寸为4×4,BN代表批量归一化操作层,Relu是非线性激活层。分类器由单个全连接层(FC)实现,其中N为输出向量的维度。

1.2  领域泛化

源域和目标域数据分布之间的差异导致了跨领域迁移问题,该问题可以通过领域适配或者领域泛化方法来解决。领域适配在训练阶段就需要目标域的数据,所以采用领域泛化来解决跨领域迁移的问题。考虑到数据采集的困难,源域当中仅有一个数据集,本文利用Mix-Up方法[11]对该数据集进行变换混合,生成多种多样的伪域数据,利用这种伪域数据在元测试阶段更新模型的参数,提升模型对未知领域数据的泛化能力。此外,还使用特征变换层对卷积层提取的特征进行变换增强,期望提取出来的特征可以适用于未知领域的识别。

Mix-Up是一种生成数据集的近邻分布的方法,而且对类别标签进行近邻关系建模。Mix-Up在查询集中随机选择不同类别的数据和标签,进行凸组合,将凸组合作为数据增强后的查询集,用于训练特征提取器。其公式为:

(1)

(2)

其中xi和xj是从查询集中随机抽取的样本,yi和yj为对应标签, 和  是混合后的数据。 可以模拟来自未知领域的数据,提升特征提取器在未知领域上的泛化能力。

当训练样本较少,而CNN的参数较多的时候,CNN容易过拟合,网络参数得不到充分的训练。鉴于特征提取器是基于CNN实现的,通过向特征提取器中插入特征变换层,对CNN各卷积层的输出进行仿射变换,使得特征提取器提取的特征分布更加多样化,增强其领域泛化能力。给定某卷积层的输出z∈RC×H×W,其仿射变换的尺度系数γ和偏置系数β从高斯分布中采样得到:

(3)

其中θγ∈RC×1×1和θβ∈RC×1×1为γ和β对应标准差的参数,可以通过训练更新。仿射变换的输出  为:

(4)

可见仿射变换是逐个通道进行,每个特征通道上用的系数都是一样的。

1.3  元学习

本文利用元学习[12]解决跨任务的知识迁移问题,此时源域和目标域的图像类别不同。元学习通过一系列相似任务归纳这些任务的共同特性,借助这些信息进行新任务的快速学习。通过在源域中构建各种识别任务来训练算法参数,使得算法能够快速适应目标域数据上的新任务。给定特征提取器 ,分类器 ,特征变换层 。每一次训练迭代t在源域数据中随机抽样出支撑集τs={xs,ys}和查询集τq={xq,yq}。首先在支撑集τs上利用梯度下降法来更新 和 ,公式为:

(5)

其中α为梯度下降的步长,Lcls为交叉熵。然后对查询集

τq进行Mix-Up操作,得到混合后的新查询集 ,

接下来将变换层融入特征提取器并计算查询集上的分类误差Lq:

(6)

Lq体现了Mix-Up和特征变换层的应用效果,用来更新算法的所有参数,即:

(7)

在源域中的元学习过程(含有元训练和元测试)结束后,将特征变换层移除,在目标域中只要用几个样本微调参数  就可以适应未知新任务,刚好满足了小样本SAR-ATR任务的需求。

2  实验验证

2.1  实验数据

本文提出的识别算法在训练阶段利用仿真SAR数据进行元学习。在测试阶段,算法首先利用少量真实SAR数据微调算法参数,然后在剩余的真实SAR数据中进行测试。仿真SAR数据集由电磁仿真工具生成,共包含七类车辆目标,即汽车、卡车、摩托车、坦克、推土机、皮卡和公共汽车。每类仿真目标共含有72个方位角,6个俯角的不同图像。真实SAR数据来自移动和靜止目标获取和识别(MSTAR)数据集。该数据集被广泛用于SAR目标识别的基准性能评估。MSTAR数据集中包括不同方位角、不同俯角和不同地面目标的图像。地面目标分为10类,类型分别为T72、BMP2、BTR70、BTR60、2S1、BRDM2、D7、T62、ZIL131和ZSU234。俯角为17°的真实图像用来微调参数,15°和30°俯角的目标用于测试。

2.2  算法的识别性能评估

本实验测试了算法在有限真实数据下获得的识别率。对比方法包括CNN和微调了参数的CNN(CNN_TF)。图3显示了这些方法的识别率曲线。当利用10%的真实SAR数据(俯角为17°的MSTAR图像)进行微调时,本文算法、CNN和CNN_TF的识别精度为85.5%。76.9%,81.3%。当使用50%的真实SAR数据进行微调时,这三种方法的准确率分别提高到95.4%、93.2%和93.9%。如果使用全部真实SAR数据,三种方法的识别率分别达到96.3%、95.2%、和95.2%。CNN的识别率最低,因为它从头开始训练,没有从仿真SAR数据集中迁移知识。CNN_TF识别率稍高,因为它利用仿真SAR数据来初始化网络参数,并用真实数据来微调网络参数。本文算法的识别率最高,因为它考虑到了仿真SAR数据和真实SAR数据之间的差异,利用跨域元学习来解决了这一问题。本文提出算法仅用10%的真实数据就可以获取85.5%识别率,证明了当缺乏真实训练数据时,本文算法依然可以取得良好的识别性能。

2.3  和主流SAR识别算法的比较

本文算法还与近年来提出的多种类别的 SAR-ATR方法进行了比较。测试数据采用了俯角分别为15°和30°的真实SAR图像。如图4所示,本文提出的算法在两种测试俯角下分别取得了96.3和96.0的识别率。当测试俯角变大时,SAR目标的图像的形状轮廓会出现较大的变化,从而造成目标外形的改变并影响识别率。本文算法在30°测试俯角下依然表现良好,表现出一定的俯角稳健性。从图4中还可以看出,采用CNN来提取图像特征的算法(本文算法、A-ConvNet、MCNN)的识别率要明显高于采用投影变换来提取图像特征的算法(CDSPP和KRLDP)。CNN能自动提取图像的高层语义特征,优于传统的机器学习方法。

3  结  论

基于数据驱动的SAR-ATR算法往往缺乏足够的真实SAR数据用于训练。本文引入仿真SAR数据来弥补真实数据的不足。对于仿真和真实SAR数据之间差异导致的跨域和跨任务迁移问题,本文算法利用特征变换和数据增强方法解决跨领域迁移,利用元学习机制解决跨任务迁移。实验证明,本文算法在训练数据有限条件下的识别率要高于CNN和迁移学习方法。

参考文献:

[1] 保铮,刑孟道,王彤.雷达成像技术 [M].北京:电子工业出版社,2005.

[2] CHEN S Z,WANG H P,XU F,et al. Target classification using the deep convolutional networks for SAR images [J].IEEE Transactions on Geoscience and Remote Sensing,2016,54(8):4806-4817.

[3] ZHAO P,LIU K,ZOU H,et al. Multi-stream convolutional neural network for SAR automatic target recognition [J].Remote Sensing,2018,10(9):1473.

[4] MIN R,LAN H,CAO Z J,et al. A gradually distilled CNN for SAR target recognition [J].IEEE Access,2019,7:42190-42200.

[5] DING B Y,WEN G J. A region matching approach based on 3-D scattering center model with application to SAR target recognition [J].IEEE Sensors Journal,2018,18(11):4623-4632.

[6] DING J,CHEN B,LIU H W,et al. Convolutional neural network with data augmentation for SAR target recognition [J].IEEE Geoscience and Remote Sensing Letters,2016,13(3):364-368.

[7] LIU M,CHEN S C,WU J,et al. Configuration recognition via class-dependent structure preserving projections with application to targets in SAR images [J],IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing,2018,1(11):2134–2146.

[8] YU M T,ZHANG S Q,DONG G,et al. Target recognition in SAR image based on robust locality discriminant projection [J],IET Radar,Sonar & Navigation,2018,12(11):1285–1293.

[9] MALMGREN-HANSEN D,KUSK A,DALL J,et al. Improving SAR automatic target recognition models with transfer learning from simulated data [J].IEEE Geoscience and Remote Sensing Letters,2017,14(9):1484-1488.

[10] INKAWHICH N,INKAWHICH M J,DAVIS E K,et al. Bridging a gap in SAR-ATR: Training on fully synthetic and testing on measured data [J].IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing,2021,14:2942-2955.

[11] ZHANG H,CISSE M,DAUPHIN Y N,et al. mixup: Beyond empirical risk minimization [J/OL].arXiv:1710.09412 [cs.LG].[2022-09-13]. https://arxiv.org/abs/1710.09412.

[12] TSENG H Y,LEE H Y,HUANG J B,et al. Cross-domain few-shot classification via learned feature-wise transformation [J/OL].arXiv:2001.08735 [cs.CV].[2022-09-13].https://arxiv.org/abs/2001.08735v1.

作者簡介:王可(1981—),男,汉族,江苏淮安人,讲师,博士,研究方向:人工智能和大数据技术;乔琪(1982—),男,汉族,江苏淮安人,副教授,硕士,研究方向:通信技术和大数据技术。

收稿日期:2022-12-16

基金项目:江苏省电子信息职业教育研究常规课题(JSDX2021-51);江苏电子信息职业学院校级科研基金项目(JSEIYY2020003)

猜你喜欢

知识迁移
在素质教育中要重视培养学生独立获取知识的能力
初中化学图形分析题型解题思路研究
在初中英语教学中运用多媒体技术的探讨
在素质教育中要重视培养学生独立获取知识的能力
开展学习活动培养学生数学素养的研究
由一道物理题引发的对绳子张力的讨论
几则高中化学实验改进的探讨与反思
运用迁移理论 提高历史材料解析能力
高中数学教学中的迁移策略
把握考试真实内涵提高课堂教学效果