APP下载

基于深度学习的智能化高精度测向方法

2023-02-06赵伟豪张君毅

计算机测量与控制 2023年1期
关键词:信噪比卷积神经网络

赵伟豪,张君毅,2,李 淳

(1.中国电子科技集团公司 第五十四研究所,石家庄 050081;2.河北省电磁频谱认知与管控重点实验室,石家庄 050081)

0 引言

波达方向(direction of arrival,DOA)定位技术是天线阵列信号处理领域内的重要分支,利用天线阵列接收到的来波信号就能够有效实现准确的定位。在现有的测向方法中,以多重信号分类(multiple signal classification,MUSIC)为代表的算法具有较高的分辨力、估计精度、稳定性等优点[1]。但需要对阵列矩阵进行数学分解,如特征值分解或奇异值分解。但矩阵分解计算复杂,且运算量大,不适合实时处理,在不同阵列误差环境下或缺少阵列流型误差的先验信息时进行DOA估计鲁棒性较差。而相关干涉仪等算法抗干扰能力一般,在低信噪比与高幅相误差条件下的DOA估计精度较差[2]。

基于模型的DOA估计算法性能依赖于输入数据的响应准确程度,而基于深度学习的数据驱动架构已显示出具有更高的估计精度,并且显示出更低计算复杂性,增强在不同信噪比(SNR)、幅相误差的环境下适应性。Wen lizhu等人提出了基于卷积神经网络(convolutional neutral network,CNN)宽带信号DOA估计模型,比DNN结构的估计时间更少,但对低信噪比信号的估计效果有待提高[3]。Ahmet M.Elbir提出了一种用于多重信号分类的深度学习框架(DeepMUSIC),设计了多个CNN,具有更高的估计精度,并且显示出更低计算复杂性[4],但对于幅相误差较高条件下的DOA估计的性能较差。B.Shi等人提出了一种复值卷积神经网络(CCNN)来更好地拟合具有复杂包络的电磁无人机信号。利用量子概率构造了一些映射函数,并进一步分析了一些可能影响复值神经网络收敛性的因素。数值仿真表明,所提出的CCNN比卷积神经网络收敛更快,DOA估计结果更准确、更稳健[5]。H.Huang等人通过离线学习和在线学习,提出了一种新的超分辨率DOA估计和信道估计的DNN框架。其中离线学习是利用不同信道条件下的模拟数据进行训练,在线学习是基于当前输入数据得到相应的输出数据。实验证实,基于深度学习的方法可以获得比传统方法更好的DOA估计。肖等人提出了一种类似于深度残差网络的DeepFPC网络结构。DeepFPC在低信噪比下具有较高的稀疏信号恢复性能和良好的DOA估计性能[6]。

对于非理想的传感器设计和制造工艺、阵列安装误差和传感器间相互干扰、背景辐射等原因,以及信号接收环境和宽带信号等苛刻条件的时候,阵列系统中存在各种缺陷,建立分析信号传播的模型复杂困难,对DOA估计的性能产生显著的负面影响。基于数据驱动、不需要对复杂场景和各种缺陷进行人为假设和简化。为解决现有阵列测向方法计算复杂度高、存在阵列误差条件下测向性能恶化严重等问题,提出一种基于深度学习高精度快速的DOA估计算法。设计了基于卷积神经网络的PhaseDOA-Net网络模型,基于深度学习的数据驱动架构通过神经网络学习输入和输出数据之间的非线性关系,学习不同环境条件下信号数据与DOA之间的映射关系,将DOA预测问题转化为回归拟合问题,通过回归算法找到最优拟合,提高低信噪比与存在阵列误差时测向模型的泛化能力和容错能力,达到更高的测向精度和更快的处理速度[5-6]。

1 阵列信号接收模型

设空间远场有一个窄带信号s(t)入射到天线阵列上,由M个单元组成,窄带信号的入射角度为θ。信号接收阵列模型如图1所示,假设参考点收到的入射波信号为:

x0(t)=s(t)ejωt

(1)

式中,ω是信号的角频率,s(t)为信号的复振幅。设τm是阵元m收到的入射波相对于参考信号点收到的延时,有:

xm(t)=s(t-τm)ejω(t-τm)

(2)

图1 信号接收阵列模型

对于窄带信号,时间上慢变化,所以有:

s(t-τm)≈s(t)

(3)

同时考虑复基带信号,可表示为:

xm(t)=s(t)e-jωτm

(4)

将M个全向阵元收到的信号数字化表示,可得到矢量表示为:

(5)

式中,

a=[e-jωτ0,…,e-jωτM-1]T

(6)

它被称为阵列对信号源的方向矢量或导向矢量,与入射波到达方向和阵列单元的几何位置有关[7]。

信号到达方向为θl,l=1,2,…,L,L为信号源数目。阵列接收数据模型为:

X(t)=A(θ)S(t)+n(t)

(7)

式中,阵列的接收数据X(t)=[x0(t),x1(t),…,xM-1(t)]T,信号数据矩阵S(t)=[s1(t),s2(t),…,sL(t)]T,导向矢量矩阵A(θ)=[a(θ1),a(θ2),…,a(θL)],以均匀线阵为例,式中a(θl) = [1 ,e(-j2πdsin(θl )/λ),…,e(-j2πd(M-1)sin(θl )/λ)]T,λ为信号波长,d为阵元间距,噪声数据矩阵为n(t)=[n0(t),n2(t),…,nM-1(t)]T[8]。

为模拟真实的信号接收环境,在接收模型中引入阵列误差。存阵列误差在于不同的接收通道,可引用一个通道误差矢量r,

r=[(1+ε0)ejϑ0(1+ε1)ejϑ1… (1+εM-1)ejϑM-1]T

(8)

ε0,ε1,…,εM-1表示通道幅度误差,ϑ0,ϑ1,…,ϑM-1表示通道相位误差。

引入幅相误差后的阵列接收数据改写为:

X′(t)=X(t)r

(9)

针对复杂环境下对高精度快速测向的需求,提出使用卷积神经网络方法提取信号数据的内在特征,以智能化的方法来学习不同环境条件下信号数据与DOA之间的映射关系。

2 基于PhaseDOA-Net模型的DOA估计

2.1 PhaseDOA-Net神经网络模型

分类网络模型通常将连续值预测为属于每个输出类的给最高概率的类标签,可以将预测概率转化为类值。分类是预测离散类标签的任务,分类模型将角度划为不同类值,所预测的角度划入分好的区间,并对划分的准确度进行评估,对角度进行了“定性输出”,模型的性能与DOA精度受限于所划分区间。回归模型是将输入变量x的映射函数f近似为连续输出变量y的任务。是预测连续数量的任务。对角度进行“定量输出”,输出连续性变量。显然回归模型更适合DOA预测任务。

本文中,我们可以将问题表述为在给定阵列接收数据时,估计目标到达角θ。为此,我们引入了如图2所示的模型处理流程图,该网络模型以阵列相位差矩阵φM为输入,以DOA为输出。

图2 模型处理流程图

图3 PhaseDOA-Net网络结构

为学习不同环境条件下信号数据与DOA之间的映射关系,并通过模型拟合出信号角度,在本研究中提出了一个回归网络模型PhaseDOA-Net。测向系统性能的好坏取决于特征提取的充分程度以及系统学习效率的高低,故针对测向精度与效率的需求,通过网络结构模块化的设计优化上述问题。模型的网络结构由一个过滤器大小为5*5的卷积层和两个模块IdentifityBlock、FPB(feature pyramid block)组成。Conv1是核大小为5*5卷积层,允许网络以成本密集型的方式从输入信号阵列中提取低级特征。模块IdentifityBlock、FPB被设计用于处理从上层卷积层单元接收到的输出特征图,通过创建更深层的网络来提取多维纠缠特征,提高学习效率与性能。在图像问题中,某个区域在进行卷积运算后得到的特征可能同样存在于其他区域。对一个特征图中的不同区域,利用其统计学特征来替代该区域的总体特征。利用统计特征代替总体特征的做法可以显著地减少网络参数,降低运算量,这种操作被称为池化。由于池化层通过压缩数据与参数量以减少过拟合,而对于神经网络测向问题来说,池化操作会造成多路接收信号中细微特征的丢失,故网络没有使用池化层。网络模型结构如图3所示。

图3中IdentifityBlock用于处理从卷积层中接收到的输出特征图。IdentifityBlock的原理是通过创建深层次的网络来提取多维纠缠特征,提高学习效率,它由两个卷积层组成,每个卷积层中滤波器个数为64,大小为3*3,步幅(Padding)为2,每个卷积层之后是批归一化层(Batch_Normalization,BN)和修正线性单元层(ReLU),BN层加速了网络训练的收敛性,降低了输入信号的初始化灵敏度,修正线性单元层定义为ReLU(x)=max(0,x),解决了正区间梯度消失的问题,易于计算。随着网络层数的增加,梯度消失的现象会加重,且网络训练的效果不会很好,网络退化等问题严重,故引入残差网络结构。在两层或两层以上的节点两端短路连接,添加了一条“捷径”,原网络输出的F(x)就变成了F(x)+x。当神经网络无法再学习到高一维度特征时,该模块的输出变为上一层的输入。这一改动用传统反向传播训练法对深层网络进行训练的收敛速度快,效果更好[9]。

图3中FPB由两个卷积层组成,每个卷积层中滤波器个数为256,大小为3*3,步幅为1,在不牺牲特征学习效率的情况下,显著减少可训练参数的数量和计算成本。在FPB中采用相同的残差连通结构,防止网络深入时消失梯度问题。残差结构中加入一个卷积层,滤波器个数为128,大小为1*1,步幅为1,此卷积层对特征映射进行线性投影,通过1*1的卷积得到与上一层特征层相同的通道数,发挥了通道级特征对齐和降维的作用。

两个模块线性连接,通过连接(串联)层组合沿深度(或通道)维度的所有导出特征。后经过全连接(full connection,FC)层与线性(Linear)层并输出。FC层结合了前一层学习到的所有输入特征,以连接该层中所有预定义的神经元(或节点)。线性层将特征回归为所期望的信号到达方向角。引入特定模块对输入信号进行特征提取和处理,提高网络模型的拟合效果,用所提网络模型自主学习相位差矩阵与DOA之间的映射关系;引入残差网络结构,解决了卷积神经网络层数加深导致网络退化的问题。

2.2 输入数据

对阵列接收数据进行离散傅里叶变换,即:

(10)

在频域内通过被测信号所在频点处对应的相位值可得到通道相位的估计值,进而求得各通道相位差的估计值。

对于数据生成,设定信号源数目L=1,信号频率为1 200 MHz,估计接收信号的方向为[-35°,35°],步长为0.5°,信号由间距d=0.085 m,阵元数M=16线性阵接收,快拍数为1 024。为了尽可能地仿真模拟真实的无线信号环境,在信号数据中加入高斯白噪声,信噪比范围[-20,20]dB,步进为5 dB,引入相位误差,范围[-20°,20°],幅度误差2 dB,且服从均匀分布。该数据集包含141×9×100个信号,覆盖141个DOA角度,9个不同的信噪比层级,其中每个信号有100个样本,每个样本由输入数据数组I∈φ2×16组成。数据集被随机分割,其中80%用于训练,其余的用于测试。

2.3 网络训练

网络训练的主要配置如下:优化器为Adam优化器,最大训练轮数(epochs)为100,批大小(Batchsize)为1 024,初始学习率为0.01,并设置动态下降学习率Patience为5(当网络训练的损失值连续5个epochs不再下降后,学习率减少90%,保证模型在训练后期不会有太大的波动,从而更加接近最优解,以实现更好的学习收敛)。数据集的生成和模拟是使用MATLAB R2016a完成仿真,本文提出的深度网络在Python3.7中实现和训练。计算服务器具有Inter Xeon Gold 6 144 CPU和两个NVIDIA Quadro RTX 6 000 GPU。

图5 DOA测向均方根误差随信噪比变化对比图

在训练神经网络的过程中,一般都是先将神经网络前向传播,在进行反向梯度传播更新参数。在处理多维矩阵数据时,采用传统的神经网络可能会由于参数过多而无法计算,而CNN通过局部感受野和权值共享减少了参数的数量以及冗余,简化了网络结构。网络训练过程中将损失函数(Loss)设置为均方误差函数(MSE),将网络结构输出的θpredict和相应的标签θlable之间的MSE通过最小化目标函数来优化网络参数。

该过程可以用如下公式表示:

(11)

式中,len为输出层特征长度,W和b为神经网络层中各个连接权重与神经元偏置。采用反向传播算法(error back propagation,BP)对进行微调。为防止网络训练出现过拟合,在训练中加入了提前终止算法(Early Stop),即模型在测试集上权重的更新低于所设定的阈值,则提前终止训练,使用上一次迭代结果中的参数作为训练结果。

损失函数以及所预测DOA的均方根误差随训练轮数变化曲线如图4所示。在设置了动态下降学习率以及提前终止法后,初始学习率设置为0.01的学习效果最佳,Loss随训练轮数下降明显,且回归网络收敛较快。

图4 Loss及均方根误差随训练轮数变化曲线

2.4 验证结果

验证集选用的信号方向为[-35°,35°],步长为0.1°,其他条件与训练集相同,用于对不同算法在均匀阵列条件下DOA估计精度对比验证。按照上面描述的算法构建PhaseDOA-Net神经网络DOA估计模型。DeepMUSIC测向算法采用文献[3]中介绍的方法,在其他条件相同的情况下对不同方法所测得的DOA均方根误差(RMSE)进行对比。

DOA均方根误差随信噪比[-20,20]dB范围变化对比如图5(a)所示。在信噪比[0,20]dB范围内,PhaseDOA-Net神经网络算法的性能优于DeepMUSIC与MUSIC算法,干涉仪算法的DOA估计性能最差。如图5(b)所示,在较高信噪比条件下,4种算法的RMSE波动不大趋于平稳。DeepMUSIC的性能损失是由于深度网络中的精度损失所致。这是因为其深层网络作为有偏估计器,无法提供更加准确的测向结果[10]。而PhaseDOA-Net神经网络通过增加各个网络层中的单元数、网络层数以及改变网络结构来缓解此问题。

在其他条件不变的情况下,在训练过程中将输入更改为信噪比为[-20,-5]dB数据集,来观察网络在低信噪比条件下对损坏输入的泛化能力。如图5(c)所示,可以看出PhaseDOA-Net在低SNR情况下(-20 dB)的性能优于MUSIC算法。MUSIC算法在低信噪比的条件下噪声子空间与信号子空间划分不准确情况下,前者与阵列流型导向矢量的正交性会受到影响,降低DOA估计精度。

在提出的PhaseDOA-Net中,所设计的IdentifityBlock和FPB模块提高了特征学习效率,提取数据信息并建立有效的映射关系,提高了模型对不同信号条件的泛化能力,从而在多个信噪比水平回归拟合出高精度DOA。PhaseDOA-Net在不同信噪比水平上优于所有其他模型,尤其在低信噪比情况下测向精度的提升更加明显。

2.5 多频点数据验证

随着无线频谱资源的大规模使用,甚至大量采用跳频、扩频信号来保证信号的加密传输和抗干扰能力,故所提模型需适应多种频段的测向需求,以在多频带场景下使用。

在信噪比30 dB的条件下,将输入信号频率更改为[1 000,2 000]MHz数据集作为训练集,其中频率步进为200 MHz,共11个频点的窄带数据,每个频点的信号方向集合为[-35°,35°],步长为0.5°,并引入与2.2节相同的幅相误差。训练集包含141×11×100个信号,覆盖141类DOA角度,11个不同的频率层级,其中每个信号有100个样本。验证集中信号方向集合[-35°,35°],步长为0.1°,其他条件与训练集相同,计算DOA均方根误差来验证网络对输入信号频率的泛化能力。

DOA预测均方根误差随频率变化对比图如图6所示,可以看出所测信号频率越高,DOA预测均方根误差越小。所提出的PhaseDOA-Net在宽频带DOA估计性能优于其他几种测向算法。

图6 DOA预测均方根误差随频率变化对比图

2.6 测向时间对比

测向时间与算法的计算量相关。对于神经网络模型的计算量可以用浮点运算数FLOPs(floating-point operations)来衡量[11]。在卷积层中,采用滑动窗实现卷积且忽略非线性计算开销,则卷积核的FLOPs为:

Fconv=2HW(CinK2+1)Cout

(12)

其中:H、W为输入特征的高度和宽度,K为核大小,Cin和Cout为输入输出通道数。全连接层网络FLOPs为:

Ffc=(2I-1)O

(13)

其中:I为输入维度,O为输出维度。

而经典MUSIC算法中计算开销主要来源于M×M的二阶矩阵计算、矩阵的特征分解以及一维谱峰搜索。在Python3.7环境下分别实现PhaseDOA-Net与MUSIC算法,经计算PhaseDOA-Net的FLOPs为2.47 M,参数量为0.18 M,乘加次数MAC(multiply accumulate)是FLOPs的一半约为1.24 M次,而MUSIC算法的MAC约为2.717 M次。可以看出PhaseDOA-Net模型降低了计算复杂度。

在其他条件不变的情况下,比较了PhaseDOA-Net模型与其他DOA估计算法的计算时间。将验证样本数设置为3 000个,样本角度范围为[-30°,30°],得出各模型算法单个样本DOA预测时间如表1所示。

表1 PhaseDOA-Net与其他算法测向时间对比 ms

由仿真结果可知,基于PhaseDOA-Net模型的DOA测向时间低于MUSIC与干涉仪算法,略优于其他智能算法。

3 实验结果与分析

3.1 暗室环境搭建

本节通过采集实际数据来对比验证网络的DOA估计性能。暗室实际采集数据流程图如图7所示。发射天线与槽线接收天线间距为22米,符合信号采集远场条件。发射天线采用对轴发射天线。信号源发射单音信号,发射功率0 dB。所采信号数据频点范围为1~2 GHz,频率步进为200 MHz。角度范围-35°~35°,步进为2°作为训练与测试数据,步进1°作为验证数据。

图7 暗室实际采集数据流程图

暗室数据采集环境如图8所示。接收天线为槽线天线,阵列形式为均匀线阵,阵元个数为6阵元,阵元间距为0.085 m。多通道超短波接收存储单元所存储信号数据为6路I/Q数据。

图8 暗室数据采集环境

3.2 阵列接收信号误差分析

关于信号和噪声、阵列流型、以及信号处理环境均同时理想的假设,在现实中是较难满足和实现的。尽管信号和噪声条件、阵列条件、以及信号处理时所处需求条件都尽可能地追求理想,但是并不能达到最完美的状态,可能因素包括:

1)噪声干扰、噪声相关性和杂波等环境复杂性的影响;

2)传感器老化、阵元间干扰、以及阵列校正存在误差;

3)信号处理过程需要满足现实应用中的高速度等需求。此时,某种非理想情况的发生,会致使理想条件下提出的DOA估计算法受到现实环境和约束的相应影响与干扰,在精度或速度方面无法满足现实需求。通过分析幅相误差阵元位置误差、场地误差和噪声等因素导致的阵列误差,验证智能网络的泛化能力和容错能力。

3.2.1 通道幅相误差

通道误差是影响DOA性能的一种重要因素,这里考虑角度独立的通道误差,主要是由于各个通道的传输函数的不一致导致的。由所用多通道超短波接收机的设备技术指标可知,在常温条件下,6路通道的幅度误差为1 dB,相位抖动为2°。

3.2.2 阵元位置误差

由于阵列中阵元的位置存在偏差,引入的不一致性称之为阵元位置误差,阵元位置误差在一般文献中不针对性进行研究,主要是由于阵元位置误差可归结为通道的相位误差[12-13]。

假设阵元模型中该阵元的位置被认为是dxn,而在实际加工的天线单元上,该阵元的实际位置为dxn+Δdxn,所以相位误差可表示为:

(14)

由上式可得本系统中接收天线阵元位置造成的相位误差为3°。

3.2.3 多径效应与信噪比

暗室环境中墙壁由吸波材料填充,大幅减小了多径效应对系统模型的影响,信源与阵列接收器之间不存在任何障碍,直接被阵列接收器接收。通过对环境噪声的测量与对信号发射源发射功率的调整,信噪比保持在[10 dB,20 dB]范围内。

3.3 基于实采数据的网络训练与性能验证

在空域角度范围中,把空间角度均匀划分,形成离散角度集合。对于一维阵列,把空域角度范围均匀划分为离散的角度,角度间隔划分过小会增大训练难度,并且使实际采集的工作量急剧增大,而划分过大会降低网络的DOA估计精度。实采数据集角度范围设置为[35°,35°],间隔为2°,包含36×11×100个信号,覆盖36个DOA角度,11个不同的频率层级,将实际采集存储的6路I/Q数据进行数字信号处理。首先对数据进行采样,采样点设为1 024,得到信号数据X6×1 024,可得到相位差数据φx作为算法训练与验证数据。数据集被随机分割,其中80%用于训练,其余的用于测试。

网络训练的主要配置如下:优化器为Adam优化器,最大训练轮数(epochs)为80,批大小(Batchsize)为1 024,初始学习率为0.01,并设置动态下降学习率Patience为5。基于实采数据的深度网络在Python3.7中实现和训练。

验证实验一:将验证集中信号频率设置为1 200 MHz,快拍数为1 024,信噪比为15 dB。角度范围设置为-35°~35°,角度间隔设置为1°为了避免实验的偶然结果,采用100次蒙特卡洛仿真实验,包含71×100个信号,覆盖71个DOA角度。在100次蒙特卡洛仿真实验后得到的估计角度与实际角度的平均误差作为最终验证的角度误差结果。

基于实采数据的DOA精度验证结果如图9所示,图中根据DOA真实值与预测值的误差进行对比,查看估计结果与真实位置之间的差距。

图9 基于实采数据的DOA精度验证结果

上述实验条件下,干涉仪方法的预测结果与真实值的位置差距较大,估计结果与真实角度的差值范围在0.09和0.94之间,且波动较大,多个位置预测角度严重偏离正确位置,DOA估计精度较差。通过MUSIC方法预测的角度虽然也有部分预测值位置远远偏离真实位置,但对比干涉仪方法,数量上有所减少。DeepMUSIC、DNN测向误差较传统方法较小,而使用PhaseDOA-Net神经网络模型预测的结果误差最小,估计结果与真实角度的差值范围大多集中在0.02和0.47之间,准确度优于其他方法。

验证实验二:将验证集中信号频率设置为[1 000,2 000]MHz,频率步进为200 MHz,每个频点下快拍数为1 024。角度范围设置为-35°~35°,角度间隔设置为1°,同样采用100次蒙特卡洛仿真实验,统计各个频点下DOA估计均方根误差,对网络性能进行验证。通过验证得到各算法随频率的均方根误差如图10所示。

图10 实采信号测向均方根误差随频率变化对比图

由图10可以观察到,PhaseDOA-Net网络模型在各个频率范围的DOA估计精度均优于其他方法,且频点频率越高,所测得的均方根误差越小,与仿真得出结果一致。

4 结束语

针对复杂环境下对高精度快速测向的需求,为解决现有阵列测向体制计算复杂度高、存在阵列误差条件下测向性能恶化严重等问题,开展基于智能方法的高精度测向技术研究。本文提出基于PhaseDOA-Net网络模型的高精度快速DOA估计算法,提高了低信噪比环境下DOA估计方法的鲁棒性,改善了神经网络的泛化能力。通过实际验证比对,所提算法性能在低信噪比环境中不仅优于现有的MUSIC算法与相关干涉仪算法,与其他基于深度学习的DOA估计算法相比也具有一定优势。而且在阵列误差条件下也表现出了很强的鲁棒性,验证了PhaseDOA-Net测向方法对阵列误差的泛化能力和容错能力,在保证了测向精度的同时,在测向时间上也表现出优异的性能。

猜你喜欢

信噪比卷积神经网络
两种64排GE CT冠脉成像信噪比与剂量对比分析研究
基于3D-Winograd的快速卷积算法设计及FPGA实现
卷积神经网络的分析与设计
神经网络抑制无线通信干扰探究
基于深度学习的无人机数据链信噪比估计算法
从滤波器理解卷积
基于神经网络的中小学生情感分析
基于傅里叶域卷积表示的目标跟踪算法
低信噪比下基于Hough变换的前视阵列SAR稀疏三维成像
基于神经网络的拉矫机控制模型建立