舰船科学技术  2023, Vol. 45 Issue (18): 123-128    DOI: 10.3404/j.issn.1672-7649.2023.18.021   PDF    
基于舰载雷达小样本的低慢小目标识别方法
李轲1, 张建强1, 李胜军2, 王沫然1,2     
1. 海军工程大学 电子工程学院,湖北 武汉 430033;
2. 四川九洲电器集团有限责任公司,四川 绵阳 621000
摘要: 本文针对舰载雷达低、慢、小目标探测面临的小样本识别问题,提出一种基于小样本学习的低慢小目标分类识别方法。该方法将低慢小目标雷达回波数据转换到小波变换域,利用多头注意力机制和双向长短记忆人工神经网络相结合的方式,解决了小样本目标分类识别的问题。在低慢小目标雷达回波仿真数据集上,开展模型训练和算法验证,分析任务差异性与识别准确率的关系,实验结果表明该方法对典型低慢小目标识别精度可达90%以上,在雷达目标识别领域具有较好的应用价值。
关键词: 目标识别     小波变换     多头注意力机制     长短期记忆网络     小样本学习    
Low-slow and small target recognition method with few shot learning based on shipborne radar
LI Ke1, ZHANG Jian-qiang1, LI Sheng-jun2, WANG Mo-ran1,2     
1. Electronic Engineering Institute, Naval University of Engineering, Wuhan 430033, China;
2. Sichuan Jiuzhou Electric Group Co., Ltd., Mianyang 621000, China
Abstract: Aiming at a small number of sample recognition problems faced by low-slow and small target detection of shipborne radar, a low-slow and small target recognition method with few shot learning based on shipborne radar is proposed. The method converts the radar echo data of low-slow and small targets into the wavelet transform domain, and solves the problem of target recognition by using a combination of multi-headed attention mechanism and bi-directional long short-term memory artificial neural network. Model training and algorithm validation are carried out on the low-slow and small target radar echo simulation dataset, and the relationship between task variability and recognition accuracy is analyzed. Experimental results show that the recognition accuracy of typical low-slow and small targets can reach more than 90%, which verifies the good performance of this method and shows promising application in the field of radar target recognition.
Key words: target recognition     wavelet transform     multi-head attention     long-short term memory     few shot learning    
0 引 言

低、慢、小目标具有雷达散射截面积小、飞行高度低、速度慢或多普勒频移不明显的特点,在复杂海洋环境中,低、慢、小目标的识别成为一个世界性难题[1-3]。目前低、慢、小目标主要通过雷达、光电、声学等手段探测[4-5]。雷达作为一种全天候探测手段,在低、慢、小目标识别中应用越来越广。舰载雷达接收的雷达回波主要由目标回波和海杂波组成,由于海杂波空时变化的复杂性,传统基于规则和目标特征库的目标识别方式难以满足复杂海况下低、慢、小目标识别的需求,深度学习技术作为智能识别的有效手段,为低、慢、小目标识别提供了技术基础。然而,传统深度学习对样本的过度依赖限制了模型在实际场景种的应用。如何挖掘小样本的优势已成为目标识别应用的迫切问题[6]

小样本学习的概念最早源自于计算机视觉领域,近几年在电磁目标识别方向受到广泛关注。在视觉方向,Oriol Vinyals等[7]利用余弦注意力机制学习类别间的度量来推断目标样本的类别,通过训练和测试一致性匹配原则,该方法在目标识别和自然语言处理上表现出优异的性能。Li等[8]提出了一个协方差度量网络,在基于小样本分类任务的分布一致性上,利用协方差表示和协方差矩阵对图片进行识别。在电磁方向,丁辰伟等[9]将图像增强技术和生成对抗网络应用于雷达辐射源个体识别中,实验证明该方法的有效性,解决了小样本条件下目标识别概率较低的问题。

综上,当前主流先进的小样本识别算法模型主要针对视觉图像等领域,对于电磁信号目标识别领域的研究相对较少,特别是低、慢、小目标识别领域。针对以上问题,本文借鉴匹配网络,整合多头注意力机制和残差模块去构造一个表征学习单元,将经过小波变换预处理后的信号输入学习单元,利用FocalLoss损失函数实现对低慢小目标雷达回波信号的特征提取和分类。

1 网络结构设计

本文通过多头注意力机制和双向长短期记忆人工神经网络相结合的方式,利用少量带标签的样本训练一个分类器解决舰载雷达小样本目标识别的问题,整体框架如图1所示。

图 1 基于舰载雷达小样本的低慢小目标识别分类框架 Fig. 1 Low-slow and small target recognition framework with few shot learning based on shipborne radar
1.1 基于小波变换的域转换模块

考虑到雷达在探测过程中受到海上多种环境因素干扰,降低了目标识别性能[10]。基于小波变换原理,对探测到的目标信号进行分解,尽可能消除干扰因素带来的影响。

小波变换的重要核心是小波基函数 $\psi (t)$ ,通过尺度因子 $ a $ 和平移因子 $ b $ 来控制小波函数的伸缩和平移,小波基函数为:

$ {\psi _{a,b}}(t) = \frac{1}{{\sqrt a }}\psi \left(\frac{{t - b}}{a}\right)。$ (1)

式中: $ \psi $ 为小波基函数; $ a $ 为尺度因子; $ b $ 为平移因子。

小波变换同时满足了时域和频域的高分辨,有良好的时频局部化特性。考虑目标回波信号受海杂波影响较大以及SWT(Synchrosqueezed Wavelet Transform)小波函数的消噪特性。选用SWT小波函数进行目标回波特征域转换,使用db1小波基做3层小波分解。

1.2 基于残差与多头注意力机制的特征表征模块

在小样本学习中对目标信号的特征提取通常采用4~6层的卷积进行特征提取,该网络易于训练,但学习能力有限[11-15]。因此,本文提出使用残差神经网络与多头注意力机制进行特征提取。

多头自注意力机制发挥了对输入序列中元素之前相关关系进行建模的强大能力,可进一步联合关注来自不同表示子空间的信息,提高模型性能。多头注意力模型通过对输入的序列分别进行3次线性映射得到QKV,利用QK的点积计算得到的权重对V进行加权求和。在计算时通过输入多次映射,每个映射使用不同参数进行相同计算,最后将各个输出合并在一起,相比缩放点积注意力更适合处理含有细微变化的数据,多头注意力模块如图2所示。

图 2 多头注意力模块 Fig. 2 Multi-head attention module

给定目标回波通过线性映射计算查询向量、键向量和值向量注意力矩阵。注意力矩阵计算如下:

$ Attention({\boldsymbol{Q}},{\boldsymbol{K}},{\boldsymbol{V}}) = soft\max \left(\frac{{{\boldsymbol{Q}}{{\boldsymbol{K}}^{\rm{T}}}}}{{\sqrt {{d_k}} }}\right){\boldsymbol{V}}。$ (2)

式中:Q为查询矩阵;K为键值矩阵;V为值矩阵; $ {d_k} $ 为尺度因子。

基于残差与多头注意力机制的特征表征模块利用卷积神经网络提取目标回波信号的局部信息,同时使用多头注意力机制对全局信息进行建模,捕捉各散射点的本质特征,在特征学习中增强目标回波显著信息,特征提取模块如图3所示。

图 3 基于残差与多头注意力机制的特征提取 Fig. 3 Based on residual and multi-head attention mechanism of feature extraction
1.3 基于全局上下文注意力感知记忆增强模块

源域和目标域的样本特征由共有特征和相关特征构成。共有特征为可学习的通用特征,这些特征与领域无关;相关特征与每个域独有的特征,用来保持每个域的可区分性[16]

每个样本对象包含了每个样本的多种特征表征类型,每种表征都在存在不同程度的信息损失,为提高目标识别精度,本文结合多步双向长短记忆网络和注意力机制学习样本之间的共有特征。在强表示性的特征提取模块基础上,增加全条件编码,增强模型的共有特征提取性能。

图4所示,为增强网络记忆容量,将学习样本的共有特征信息保存在辅助记忆中,在特征信息提取时再进行读取。通过多步双向长短记忆网络进行共有特征提取,将支持集样本特征生成序列输入多步双向长短记忆网络使支持集和查询集的样本之间相互作用,通过关注样本间的共性特征提取样本间的共有语义信息。多步双向长短记忆函数为:

图 4 目标源数据表示 Fig. 4 Target source data represention
$ f(\hat x,S) = attLSTM(f'(\hat x),g({\boldsymbol{S}}),k)。$ (3)

式中: $f'(\hat x)$ 为标准卷积神经网络,用来提取目标输入特征; $g({\boldsymbol{S}})$ 为嵌入支持集函数,用来提取支持集特征;S为支持集;k为固定展开迭代步数; $ \hat x $ 表示目标数据。

2 基于迁移学习的分类识别

利用迁移学习的思想来解决舰载雷达低慢小目标识别问题,流程如图5所示。舰载雷达低慢小目标识别包括线下训练、线下识别2个部分,训练采用n-way k-shot的形式构建包含支持集样本和查询集样本的任务数据集。面对目标任务时,模型利用源任务学习的共有特征,通过少量的支持集样本,实现目标的识别。

图 5 小样本迁移学习框架 Fig. 5 Few-shot transfer learning framework

分类器以“正间隔”来区分目标,由于低、慢、小目标样本特征极为相似,在基础类上,“正间隔”可形成清晰的类簇结构并取得了较高的准确率;然而在目标类上,正的“间隔”破坏了特征的类簇结构。考虑特征的可迁移性和可分性,利用FocalLoss损失函数解决难易样本数量不平衡的问题,完成低、慢、小目标的分类识别,其表达式为:

$ FL({p_t}) = - {\alpha _t}{(1 - {p_t})^\gamma }\log ({p_t}) 。$ (4)

式中: $ {\alpha _t} $ 为权重参数; $ {p_t} $ 为预测值; $ \gamma $ 为可调节因子。

训练中,对于容易区分的样本,也就是 $ {p_t} $ 值趋近于1的样本,可降低它的损失函数值,而对于 $ {p_t} $ 值趋近于0的样本,对损失函数值影响并不大,这样可促使网络的学习权重向困难样本倾斜,提高网络区分困难样本的敏感度。同时为提高误分类样本的权重,利用 $ {\alpha _t} $ 增加误分类权重。

3 实验结果 3.1 数据集及实验设置

仿真工具为电磁计算软件FEKO和Matlab,将仿真场景设置为远场舰载单站雷达,海杂波采用复合K分布,SCNR(信号杂波噪声比)为3 dB。仿真参数设置如下:雷达天线高度为20 m;工作波段为L波段;信号载频为1 GHz;脉冲重复频率PRF为4 096 Hz;波束宽度3°;天线扫描速度为30°/s。信号带宽50 MHz,时宽10 μs。以雷达所在位置在地平面上的投影为O-XYZ坐标系原点。仿真的11种低、慢、小目标具体参数如表1所示。

表 1 低慢小目标数据集 Tab.1 Low-slow and small target datasets

以滑翔机为例,设滑翔机高度500 m,与雷达水平距离5 km,雷达天线高度为20 m,采用水平极化。以雷达所在坐标系为参考坐标系,计算可得滑翔机与雷达之间俯仰角为5.47°。激励设置与计算得到的目标RCS值如图6图7所示。

图 6 激励设置 Fig. 6 Incentive settings

图 7 滑翔机RCS估计值 Fig. 7 RCS estimation for gliders

基于舰载雷达小样本的低、慢、小目标分类识别使用赤龙4固定翼无人机、日本ASM-2、雄达UF26快艇3类目标作为训练集。S100无人直升机、ME300领航者无人船、动力三角翼3类目标作为验证集。大疆精灵4、滑翔机、美国捕鲸叉、AD100轻型飞机、华微6号无人船5类目标作为测试集。训练采用3way-5shot的训练方式,测试以60次随机测试精度的均值作为最终识别结果。训练分为2种情况:1)从训练集每类中挑选30个样本,共90个训练样本;2)从训练集每类中挑选100个样本,共300个训练样本。验证集每类300个样本,共900个验证样本,测试集每类300个样本,共1800个测试样本。

3.2 实验结果与分析

从训练样本量和目标回波姿态角2个角度来探索样本差异性对模型性能的影响。数据集空中目标的信噪比(SNR)为15 dB,水面目标的SNR为5 dB,目标回波姿态角包含0°和30°,测试集每类300个样本,共1800个测试样本,实验使用姿态角差异性表示样本差异性,设置9组对比实验。实验设置及结果如表2表3所示。

表 2 90个训练样本 Tab.2 Difference experiment results of 90 training samples

表 3 300个训练样本差异性实验结果 Tab.3 Difference experiment results of 300 training samples

针对训练任务和测试任务样本量的差异,绘制折线如图8所示。

图 8 目标识别率与任务差异性关系 Fig. 8 The relationship between target recognition rate and task

图8(a)表示目标回波姿态角为0°时,在300个训练样本的情况下,目标识别率由99.65%降低到99.5%,下降了0.15%;在90个训练样本的情况下,目标识别率由99.59%降低到97.45%,下降了2.14%。图8(b)表示目标回波姿态角为30°时,当训练样本和测试样本存在姿态角差异时,在300个训练样本的情况下,0°的测试精度提高了0.5%,但混合的测试精度降低了0.09%;在90个训练样本的情况下,目标识别率由98.39%降低到97.82%,下降了0.57%。图8(c)展示了源任务姿态角为0°和30°时,在0°和30°测试时,识别精度都有相应的提高;在300个训练样本的情况下提高0.61%,在90个训练样本的情况下提高0.8%。

在针对训练任务和测试任务的内类差异性,为避免任务样本量的影响,根据表2表3的实验结果分别对比绘制类内差异性对比图,如图9所示。

图 9 目标识别率与类内差异性关系 Fig. 9 The relationship between target recognition rate and intraclass difference

图9(a)可以看出,当训练样本为90个,0°、30°和混合姿态角的识别率分别为99.59%、98.82%和97.45%,姿态角混合后的识别率比0°和30°的识别率均值低1.76%;当训练样本为300个,0°、30°和混合姿态角的识别率分别为99.65%、99.56%和99.55%,姿态角混合后的识别率略低于单个姿态角识别率均值;图9(b)和图9(c)在目标识别率的变化趋势上整体与图9(a)一致。因此可知,目标回波姿态角增大,模型性能均呈现下降趋势,训练样本量大时,影响较小,但总体说明类内差异对模型推理性能有一定影响。

4 结 语

针对舰载雷达目标识别中存在的小样本问题,本文提出一种基于舰载雷达小样本的低、慢、小目标识别方法,使用具有强表示学习能力的基于残差与多头注意力机制的特征表征模块。基于全局上下文注意力感知记忆增强模块提取样本的共有特征,结合FocalLoss损失函数实现困难样本的分类。在FEKO和Matlab仿真的十一类低、慢、小目标雷达回波数据集上进行多组完备实验验证,充分挖掘了小样本学习中不同样本间共有特征和相关特征。实验结果表明,本文提出方法在小样本识别任务上具有优异的性能,进一步证明了本文方法在雷达目标识别领域的可行性和适用性。同时,由于场景的特殊性,仿真环境无法完全模拟真实环境,海杂波与低慢小目标回波建模具有一定的局限性,如何从仿真环境迁移到真实场景,这是下一步工作重点研究的问题。

参考文献
[1]
卞伟伟, 邱旭阳, 刘方.“低慢小”目标协同探测综合识别关键技术研究[C]//北京: 第五届中国指挥控制大会, 2017: 264–269.
[2]
李明明, 卞伟伟, 甄亚欣. 国外“低慢小”航空器防控装备发展现状分析[J]. 飞航导弹, 2017, 1(9): 62-70.
[3]
向凡夫, 郝冬青, 吴鹏. 针对低慢小目标的雷达信号处理算法[J]. 指挥控制与仿真, 2019, 41(4): 40-46.
[4]
苑文楠, 贾彦翔, 吕鑫, 等. “低慢小”目标多体制武器分配优化模型[J]. 电光与控制, 2022, 29(3): 1-5. DOI:10.3969/j.issn.1671-637X.2022.03.001
[5]
许述文, 白晓惠, 郭子薰, 等. 海杂波背景下雷达目标特征检测方法的现状与展望[J]. 雷达学报, 2020, 9(4): 684-714.
[6]
晏媛, 孙俊, 孙晶明, 等. 雷达小样本目标识别方法及应用分析[J]. 系统工程与电子技术, 2021, 43(3): 684-692. DOI:10.12305/j.issn.1001-506X.2021.03.11
[7]
VINYALS O, BLUNDELL C, LILLICRAO T P, et al. Mathcing networks for one shot learning[C]//Neural Information Processing Systems. Barcelona, 2016: 3630–3638.
[8]
LI W, XU J, HUO J, et al. Distribution consistency based covariance metric networks for few-shot learning[C]//Proceedings of the AAAI Conference on Artificial Intelligence. Honolulu, 2019: 8642–8649.
[9]
丁辰伟, 孙闽红, 官友廉. 基于小样本条件的雷达辐射源个体识别算法[J]. 弹箭与制导学报, 2021, 41(3): 25-30.
[10]
陈宏. 小波变换下舰船噪声信号识别方法[J]. 舰船科学技术, 2021, 43(14): 10-42.
CHEN H. Research on recognition method of ship noise signal based on wavelet transform[J]. Ship Science and Technology, 2021, 43(14): 10-42. DOI:10.3404/j.issn.16727649.2021.7A.004
[11]
LI R, WU Z, JIA J, et al. Towards discriminative representation learning for speech emotion recognition[C]//Proceedings of the 28th International Joint Conference on Artificial Intelligence. AAAI Press. AAAI Press, 2019: 5060–5066.
[12]
SNELL J, SWERSKY K, ZEMEL R. Prototypical networks for few-shot learning[J]. Advances in neural information processing systems, 2017: 4077–4087.
[13]
SUNG F, YANG Y, ZHANG L, et al. Learning to compare: Relation network for few-shot learning[C]//IEEE conference on computer vision and pattern recognition. Salt Lake City, 2018: 1199–1208.
[14]
SANTORO A, BARTUNOV S, BOTVINICK M, et al. Meta-learning with memory-augmented neural networks[C]//International conference on machine learning. New York City, 2016: 1842–1850.
[15]
HOU R, CHANG H, MA B, et al. Cross attention network for few-shot classification[C]//The 33rd International Conference on Neural Information Processing Systems. Vancouver, 2019: 4003–4014.
[16]
LI Y, TIAN X, GONG M, LIUY, et al. Deep domain generalization via conditional invariant adversarial networks[C]// European Conference on Computer Vision (ECCV). Germany, 2018: 624–639.