基于迁移学习的被动雷达外辐射源信号识别方法与流程

文档序号:15381470发布日期:2018-09-08 00:04阅读:1108来源:国知局

本发明属于雷达和通信信号识别领域,迁移学习领域。基于训练过的深度学习模型,设计了一种基于迁移学习的被动雷达外辐射源信号识别方法。



背景技术:

复杂异构的电磁环境,为信号处理带来巨大的挑战,迫切需要发展军民融合特殊应用、公众系统与专用系统相结合的技术,以有效提高频谱利用效率、改善环境、协同共存。因此,未来的雷达系统设计必须从提高频谱资源利用率的角度出发,被动雷达、波形分集、仿生设计和认知方法是解决频谱拥堵的有效方法。

被动雷达(又称无源雷达、外辐射源雷达)自身不发射电磁波,而是利用外辐射源进行目标探测和跟踪,具有较好的隐蔽性、“四抗”(抗干扰、抗低空突防、抗反辐射导弹、抗隐身)能力、无电磁污染、无探测盲区、便携和低成本等优势。但被动雷达波形不受控制、信道传播环境复杂等问题给研究带来挑战。认知或知识辅助能带来雷达系统性能的提升。simonhykin首次将认知无线电的概念引入雷达,所提出的认知雷达具有连续感知环境、快速调整系统参数和快速数字信号处理的能力。

被动雷达中对外辐射源信号的准确认知和识别是非常关键的问题,识别出空间中的电磁信号从而选择合适的波形将极大提升被动雷达的性能。传统的信号识别任务往往是通过专家特征工程,如累积量,循环平稳特征和分布距离来实现的。上面提到的基于特征工程的方法能够很好地完成专门的任务,但是缺乏灵活性。这些方法的关键在于先特征提取,然后根据其特点进行分类,缺乏对原始数据的充分利用。原始数据中包含了信号特性很重要的信息,深度学习可以从原始数据中自动学习特征,是一种新型的有效的信号识别方法。

通常,训练一个鲁棒的深度学习模型会耗费很大的成本,而且以不同采样频率采集的数据训练的模型之间的泛化能力并不理想。一般在一个采样频率的数据集上训练得到的模型在另一个采样率的数据集上性能会下降。所以,模型的迁移问题显得很重要。迁移学习(transferlearning)目标是将从一个环境中学到的知识用来帮助新环境中的学习任务。具体地,在迁移学习中,我们已有的知识叫做源域(sourcedomain),要学习的新知识叫目标域(targetdomain)。迁移学习研究如何把源域的知识迁移到目标域上。特别地,在机器学习领域中,迁移学习研究如何将已有模型应用到新的不同的、但是有一定关联的领域中。传统机器学习在应对数据的分布、维度,以及模型的输出变化等任务时,模型不够灵活、结果不够好,而迁移学习放松了这些假设。在数据分布、特征维度以及模型输出变化条件下,有机地利用源域中的知识来对目标域更好地建模。另外,在有标定数据缺乏的情况下,迁移学习可以很好地利用相关领域有标定的数据完成数据的标定。

迁移学习按照学习方式可以分为基于样本的迁移,基于特征的迁移,基于模型的迁移,以及基于关系的迁移。基于样本的迁移通过对源域中有标定样本的加权利用完成知识迁移;基于特征的迁移通过将源域和目标域映射到相同的空间(或者将其中之一映射到另一个的空间中)并最小化源域和目标域的距离来完成知识迁移;基于模型的迁移将源域和目标域的模型与样本结合起来调整模型的参数;基于关系的迁移则通过在源域中学习概念之间的关系,然后将其类比到目标域中,完成知识的迁移。



技术实现要素:

为克服现有技术的不足,本发明旨在提出一种外辐射源信号识别的迁移学习方法,可解决不同采样率信号训练得到的模型的迁移问题。该方法与源域和目标域采样频率的大小关系无关。为此,本发明采用的技术方案是,基于迁移学习的被动雷达外辐射源信号识别方法,首先在一个采样频率数据集上训练得到一个模型作为基础网络,然后将基础网络的模型和参数直接迁移到不同采样频率的数据集目标域,做微调训练,基础网络使用含3层卷积和两层全连接层的神经网络模型。

具体地:

数据集产生,使用矢量信号产生仪产生包括调频信号fm,移动通信系统信号gsm,长期演进信号lte,宽带码分多址wcdma,蓝牙bluetooth和包括五种不同的无线局域网信号wlan-ac、wlan-a/g/j/p、wlan-g、wlan-b/g、wlan-n的10种协议信号作为数据集;

参数迁移步骤如下:

步骤1:选择识别模型为包含三层卷积和两层全连接层的神经网络架构a,训练网络并保存模型和参数p作为基础网络;

步骤2:将基础网络的前n层作为目标网络的前n层,用参数p初始化前n层;

步骤3:随机初始化目标网络剩余的层即初始化全连接层;

步骤4:将目标网络在目标域不同采样率数据集重新训练。

本发明的特点及有益效果是:

本发明的迁移学习模型可以很好地解决不同采样率数据集识别模型的泛化问题,把一个预先训练好的模型的部分迁移到一个类似的任务上使用。一方面这节省了训练的时间;另一方面对于某些任务我们可能有很大的数据集而对于另一些相似的任务并没有那么大的数据集,就可以直接迁移训练好的模型和参数,然后做微调即可。

为了将在一种采样率的数据集上训练的模型应用到不同采样率的数据集上,比较了同一种网络在同一类数据集上从零开始训练(trainfromscratch)和迁移学习再训练(transfertraining)两种方法的结果。图4和图5展示了在protocol2-fs250(采样率为25mhz)数据集上训练的基础网络迁移到protocol2-fs225(采样率为22.5mhz)数据集和protocol2-fs300(采样率为30mhz)数据集上再训练的结果(下面小圆圈的线是目标网络trainfromscratch的结果)。

可以看到,由于提前获取了先前训练得到的结果,迁移学习方法可以使模型在5个epoch(周期)内基本趋于稳定,然而从零开始训练的方法需要至少20个epoch才能稳定。引入迁移学习极大节省了训练时间,提升了效率,使不同采样率训练得到的模型可以方便地移植到其他数据集。

附图说明:

图1迁移学习模型。

图2迁移学习流程图

图3深度学习架构a。

图4在protocol2-fs250(采样率为25mhz)数据集上训练的基础网络迁移到protocol2-fs225(采样率为22.5mhz)数据集上。

图5在protocol2-fs250(采样率为25mhz)数据集上训练的基础网络迁移到protocol2-fs300(采样率为30mhz)数据集上。

具体实施方式

本发明旨在设计一种外辐射源信号识别的迁移学习方法,可解决不同采样率信号训练得到的模型的迁移问题。该方法是基于模型的参数迁移,将一个采样频率数据集(源域)上训练的模型和参数,迁移到另外一个采样频率的数据集(目标域)上,只需少量有标签的目标域数据,做短时间的训练,即可得到目标域的识别模型。该方法与源域和目标域采样频率的大小关系无关。

(1)模型结构

本发明的实现模型如图1所示。

首先在一个采样频率数据集上训练得到一个模型作为基础网络,然后将基础网络的模型和参数直接迁移到目标域(不同采样频率的数据集),做微调训练即可。这种方法可以使目标域网络模型参数迅速收敛。基础网络一般使用含3层卷积和两层全连接层的神经网络模型。

(2)数据集产生

本发明使用ni的pxi-5611矢量信号产生仪产生了10种协议信号(fm,gsm,lte,wcdma,bluetooth,wlan-ac,wlan-a/g/j/p,wlan-g,wlan-b/g,wlan-n)作为数据集。发射端和接收端采用全向天线辐射的方式接收信号。接收机采样频率分别设为22.5mhz,25mhz和30mhz,得到三个不同采样率的数据集:protocol1-fs225,protocol2-fs250和protocol3-fs300。由于fm信号带宽较窄,三个数据集对应fm信号采样频率分别设为1.125mhz,1.25mhz和1.5mhz。

存储过程中,每个样本分别包含i路(实部)和q路(虚部)各n个采样点,前期训练模型需要给每个样本添加标签,比如gsm,lte等。数据集包含10种协议信号数据,一共40000个样本,每个样本存储形式为2×400的i&q数据。将20000个样本随机打乱作为训练集,余下的20000个样本随机打乱作为测试集。

(3)参数迁移方法

模型迁移学习的主要步骤如下:

步骤1:选择合适的识别模型,如包含三层卷积和两层全连接层的神经网络架构a,训练网络并保存模型和参数p(基础网络)。

步骤2:将基础网络的前n层(一般选择全连接之前的层)作为目标网络的前n层,用参数p初始化前n层。

步骤3:随机初始化目标网络剩余的层(全连接层)。

步骤4:将目标网络在目标域(不同采样率数据集)重新训练(少量时间)。

实现流程图如图2所示。

如图3所示,a模型的每个卷积层采用修正线性单元rectifiedlinear(relu)作为激活函数,采用softmax(一种多分类激活函数)作为最后一层的激活函数。为了避免过拟合,模型a采用dropout技术防止过拟合,在每层后面加上dropout层,设置dropout为0.5。采用前向传播和反向传播技术来训练模型,更新权重。模型a采用了adam(adaptivemomentestimation)作为优化函数。adam利用梯度的一阶矩估计和二阶矩估计动态调整每个参数的学习率。adam的优点主要在于经过偏置校正后,每一次迭代学习率都有个确定范围,使得参数比较平稳。采用批训练的方式,batchsize(批大小)为256,模型在gpu上训练30个epoch。

神经网络是深度学习其中的一部分。本发明为迁移学习模型,是将源域中训练好的参数和模型迁移到目标域,可以达到很快的收敛效果。比如在一种采样率采集的数据(源域)训练的参数迁移到另一种采样率采集的数据上(目标域)。

本发明是一种外辐射源信号识别的迁移学习方法,可解决不同采样率信号训练得到的模型的迁移问题,可应用到被动雷达领域。该方法是基于模型的参数迁移,将一个采样频率数据集(源域)上训练的模型和参数,迁移到另外一个采样频率的数据集(目标域)上,只需少量有标签的目标域数据。

外辐射源信号识别的迁移学习主要步骤如下:

步骤1:选择合适的深度学习识别模型,如卷积层和全连接层构成的网络架构,训练网络并保存模型和参数p(基础网络)。

步骤2:将基础网络的前n层(一般选择全连接之前的层)作为目标网络的前n层,用参数p初始化前n层。

步骤3:随机初始化目标网络剩余的层。

步骤4:将目标网络在目标域(不同采样率的数据集)重新训练(少量时间)。

当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1