改进的WOA-VMD算法在水声信号去噪中的应用❋
2023-02-21张伟萍张海燕
张伟萍,付 民,张海燕,李 欣
(中国海洋大学信息科学与工程学部,山东 青岛 266100)
水声数据采集对海洋通信、导航定位、目标识别等研究领域具有重要意义[1],但是海洋环境噪声以及其他干扰声源的复杂多变使得目标信号的检测和提取变得异常困难。因此进行水声去噪处理,提高信号质量非常重要。
水声信号具有典型的非平稳性,相比于传统的傅里叶分析,小波变换具有多分辨率特性,能够对信号进行多尺度的细化处理[2]。由于鱼群、风浪和船舶等影响,水声信号频率成分复杂,利用变分模态分解(Variational mode decomposition,VMD)可以实现复杂信号在频域的有效分离[3]。相比于传统的经验模态分解(EMD)及其改进的算法[4],VMD具有分解速度快、精度高等优点,在信号处理方面得到了广泛应用。例如,朵慕社等[5]利用VMD和卷积神经网络来诊断轴承故障,李宏等[6]将VMD应用于语音信号的去噪,杨洁等[7]提出一种基于VMD和熵特征的雷达辐射源分类方法。
VMD算法进行信号分解时需要确定分解模态数和惩罚因子。许多文献中仅凭经验或者不能综合考虑二者的相关性以及不同输入信号的特点来选取参数,都会导致信号分解效果变差。
在小波分析方面,小波阈值(WT)去噪算法由于原理简单,易于实现的特点,成为信号去噪领域的重要方法。但是传统的软、硬阈值函数具有不连续和存在固定阈值偏差的缺陷,容易导致重构信号出现伪吉布斯现象以及边缘模糊等失真问题。此外,在传统的阈值选取方式下,不同小波分解层数对应的阈值固定不变,并没有考虑到噪声小波系数随着分解尺度增加而表现出的衰减特性。文献[8]提出一种改进的小波阈值函数,并且结合中值滤波改善了去噪效果;文献[9]提出利用VMD和小波阈值去噪的联合算法(VMD-WT),进一步提升了去噪性能,但是没有考虑到信号VMD分解的参数设置问题,仅凭经验选取参数模型,存在很大的偶然性与误差问题,并没有对阈值区间的小波系数进行处理,容易丢失信号的细节信息;文献[10]对小波阈值函数与阈值选取方式都进行了改进,但是调节参数过多,去噪效率还有待提高。根据上述算法的优缺点,本文提出一种优化VMD和改进小波阈值处理相结合的去噪方法。通过改进的鲸鱼优化算法自适应地选取VMD的最佳参数组合,将水声信号分解成多个相对平稳且具有不同特性的模态分量。通过计算各模态分量的中心频率以及与原始信号的互相关系数,确定有效信号分量,最后使用改进的可调阈值函数和分层阈值对有效分量进行小波去噪处理。实验结果表明,本文算法与传统FIR、IIR滤波方法、EMD-WT(经验模态分解-小波阈值去噪)、EEMD-WT(集合经验模态分解-小波阈值去噪)、文献[8]所提的小波阈值与中值滤波结合的算法、文献[9]提出的VMD-WT方法、文献[10]中的改进小波阈值去噪算法相比,去噪效果更好且能够有效地提升水声信号的信噪比。
1 VMD算法
利用VMD将输入信号f(t)分解成K个具有不同频率和带宽的模态分量[11],实现信号在频域内的有效分离。其对应的带宽约束变分问题为:
(1)
式中:{μk}={μ1,…,μK}和{ωk}={ω1,…,ωK}分别表示各IMF分量及其对应的中心频率;*表示卷积,∂t为梯度运算。
求解式(1)时,引入Lagrange乘子λ和惩罚因子α,得到非约束的Lagrange表达式:
L({μk},{ωk},λ)=
(2)
最后迭代求出上述模型的最优解,具体步骤为:
(b)对μk、ωk和λ进行更新,公式分别为:
(3)
(4)
(5)
(c)判断是否满足终止条件:
(6)
式中ε表示精度。满足则停止迭代,否则回到步骤(b)。
2 鲸鱼优化算法
WOA是一种新提出的群智能优化算法,模拟座头鲸的狩猎行为并建立数学模型[12]。座头鲸在捕食时主要包含以下三种机制:
(1)缩小包围机制
鲸群发现猎物后会不断收缩包围圈,并根据当前种群最优个体更新位置,对应公式为:
D=C·X*(t)-X(t)。
(7)
X(t+1)=X*(t)-A·D。
(8)
其中:t是当前迭代次数;X(t)代表鲸鱼个体的位置矢量;X*(t)代表当前最优个体的位置矢量;系数A和C定义为:
A=2mr-m。
(9)
C=2r。
(10)
(11)
式中:r为[0,1]的随机值;Tmax表示最大迭代次数。
(2)螺旋气泡攻击
座头鲸在围困猎物的同时,还会有50%的概率制造出螺线形气泡网实施攻击,数学模型对应为:
D′=X*(t)-X(t)。
(12)
X(t+1)=D′·ebl·cos(2πl)+X*(t)。
(13)
式中:D′表示鲸鱼个体到当前最佳解的距离矢量;l是[-1,1]区间的任意值;b是常数,本文取值为1。
(3)随机搜索机制
在搜索猎物阶段,当系数|A|≥1时,鲸群将通过随机搜索在全局范围内寻找优质猎物(候选解),其位置更新表达式为:
D=C·Xrand(t)-X(t)。
(14)
X(t+1)=Xrand(t)-A·D。
(15)
上式中,Xrand(t)是选的随机个体的位置矢量。当|A|<1时,鲸群会选择在局部范围内进行更加精细的搜索,同时根据式(8)更新个体位置。
2.1 WOA的改进策略
传统的WOA通过系数A来调节全局和局部搜索,A的值主要与m有关,然而线性变化的收敛因子m并不利于算法的非线性寻优过程。对此,本文提出一种余弦收敛因子,公式如下:
(16)
m的变化情况如图1所示。在整个迭代过程中,收敛因子非线性递减。前期和中期阶段,m的值较大且变化较快,可以扩大|A|≥1的比例且能够尽快地在全局范围内确定优质解,增强算法的全局寻优性能,同时较大步长更有利于摆脱局部最优极值[13]。算法后期,m以较小步长进行局部的细致搜索,可以提高算法收敛精度。
图1 余弦收敛因子曲线图
此外,引入自适应权重β控制当前最优个体对鲸群位置更新的影响。前期较大的权值可以防止陷入局部最优,后期较小的权值更利于鲸群对猎物的精确定位,改进WOA的局部搜索能力。β定义为:
(17)
同时,式(8)、(13)将更新为:
X(t+1)=β·X*(t)-A·D。
(18)
X(t+1)=D′·ebl·cos(2πl)+β·X*(t)。
(19)
3 改进的小波阈值去噪
小波阈值去噪[14]的主要步骤为:
(1)确定小波基函数和分解层数,对含噪信号进行小波变换,得到各尺度下的小波系数。
上述过程中,阈值、阈值函数的选取对最终去噪效果具有重要影响。本文在传统算法的基础上,引入一种新的阈值函数和分层确定阈值的方法。
3.1 改进的小波阈值函数
针对传统的硬、软阈值函数存在的不连续和恒定差值问题,本文提出一种可调阈值函数,表达式为:
(20)
式中:j表示分解尺度;λ表示阈值;k为系数标号;a是可调参数且0≤a≤1。
图2 基于参数调整的阈值函数曲线图
3.2 分层阈值
(21)
式中:σ表示噪声标准差;N表示序列长度;L表示小波分解尺度。
4 基于改进WOA-VMD的小波阈值去噪
使用VMD分解信号时需要预先设定参数:分解模态个数K和惩罚因子α。本文利用鲸鱼优化算法选取最佳[K,α]参数组合,选取功率谱熵作为WOA的适应度函数,假设在第t次迭代过程中,输入信号f(t)被分解出K个模态分量,则功率谱熵的计算公式如下:
(22)
式中:N为信号序列的长度;k为标号;X(k)是第k个模态分量x(k)对应的离散傅里叶变换;pk表示第k个功率谱值占总功率谱的比重;H表示对应的功率谱熵。
如图3所示,本文算法的实现步骤如下:
图3 算法流程图
(1)WOA优化VMD参数。将需要寻优的参数组合(K,α)作为鲸鱼个体的位置,根据每个鲸鱼的位置对输入信号f(t)进行VMD分解,计算对应的功率谱熵值并记录当前功率谱熵值最小时对应的个体位置(K,α)。随着迭代次数的增加,适应度值的变化将逐渐趋于平稳,最后在满足终止条件后输出最优位置(K,α),作为输入信号f(t)进行VMD分解的最佳参数组合。
(2)使用上述得到的最佳参数模型对输入信号f(t)进行VMD分解,得到对应的K个模态分量IMFs。
(3)计算各IMF对应的中心频率以及与输入信号f(t)的相关系数,根据设定的相关系数阈值,最终确定有效分量和噪声分量。
(4)舍弃噪声分量,使用本文改进的小波阈值算法对有效分量进行去噪处理。
(5)最后重构有效模态分量得到去噪后信号。
5 实验验证
5.1 算法仿真实验
在MATLAB软件平台下进行仿真实验分析。本文采用的仿真信号如下:
(23)
式中:仿真信号f(t)代表含噪输入信号;由各谐波分量与高斯白噪声组成;x0(t)代表目标信号,主频为200 Hz;x1(t)、x2(t)、x3(t)和x4(t)分别表示频率为50、100、250和500 Hz的干扰信号;n(t)代表高斯白噪声。设置采样频率为2 000 Hz,采样点数为1 024,当含噪信号的信噪比为-2.975 0 dB时,仿真信号及其对应的频谱如图4所示,原始信号包含明显的频率干扰和噪声。
图4 目标信号、含噪信号时频图
利用本文算法进行去噪处理。首先,通过WOA获得含噪信号VMD分解的最优参数模型。设置种群数为50,最大迭代次数为100,k∈[2,9],α∈[1 000,9 000]。仿真信号的VMD参数寻优过程如图5所示,最终获得的最佳参数组合是K=8,α=7 230。
图5 WOA优化VMD参数迭代曲线
然后,利用该参数模型对含噪信号f(t)进行VMD分解,最终得到的8个模态分量对应的时频波形如图6所示。观察分解结果可以发现,经本文WOA-VMD算法选取的最佳参数模型能够实现复杂信号的有效分离,仿真信号中的多个频率相近的分量都被很好地分解到不同的频段,每一个模态分量谱中的频率都是单一的,没有出现频谱混叠问题。计算各IMF分量与输入信号f(t)的相关系数,结果分别为0.287 5、0.390 5、0.599 4、0.349 5、0.348 7、0.182 6、0.169 4、0.190 9,其中IMF3分量对应的相关系数最高且中心频率在200 Hz左右,最接近目标信号,因此选定IMF3为有效分量并进行小波阈值去噪。
图6 IMF分量及其对应频谱
选择db4小波基,采用本文方法与传统FIR、IIR滤波方法、EMD-WT、EEMD-WT、文献[8]所提的小波阈值与中值滤波结合的算法、文献[9]提出的VMD-WT方法、文献[10]中的改进小波阈值去噪算法进行去噪对比实验,去噪后信号如图7所示。
((a)FIR滤波 Finite Impulse Response filter;(b)IIR滤波 Infinite Impulse Response filter;(c)EMD-WT;(d)EEMD-WT;(e)文献[8]算法 The algorithm in literature[8];(f)文献[9]算法 The algorithm in literature[9];(g)文献[10]算法 The algorithm in literature[10];(h)本文算法The algorithm proposed in this paper.)
采用信噪比和均方根误差对最终去噪效果进行衡量,结果如表1所示。
观察表1与图7可以发现,与其他算法相比,本文算法的去噪效果最好,去噪后信号的信噪比最大,均方误差最小,本文算法重构后的波形光滑,没有出现明显的毛刺,与目标信号的波动偏差也是最小的,说明本文算法去噪的信号能够更好地拟合目标信号,在去噪方面更具优势。
表1 不同算法的去噪性能指标
为进一步验证本文算法的效果,分别加入不同强度的噪声并进行去噪对比实验,记录对应算法的SNR和RMSE,结果如表2 所示。从表2 可以看出,在不同的噪声强度下,本文算法的去噪性能指标均优于其他五种算法,能够有效提升目标信号的信噪比。图8记录了SNR=-4.514 1时的不同算法的去噪效果。
((a)FIR滤波 Finite Impulse Response filter;(b)IIR滤波 Infinite Impulse Response filter;(c)EMD-WT;(d)EEMD-WT;(e)文献[8]算法 The algorithm in literature[8];(f)文献[9]算法 The algorithm in literature[9];(g)文献[10]算法 The algorithm in literature[10];(h)本文算法The algorithm proposed in this paper.)
表2 八种算法在不同信噪比条件下的去噪性能指标
5.2 实测信号去噪
将水声信号采集板集成到海洋牧场环境监测系统中,然后投放到莱州湾附近的海域,整个系统采用原位坐底的布放方式,距离岸线约26 km,水深约10 m。
水下控制系统通过光电复合缆和接驳盒与岸基监测平台进行数据传输,通过岸基监测站发送配置命令并控制数据采集,采样频率为250 kHz,采样精度为16bit,数据实时上传到岸基服务器。选择10月24日保存的一段数据进行去噪实验,实测数据如图9所示。对比本文算法与其他不同算法的去噪效果,结果如图10所示。其中目标信号的频率是62 kHz,对比几种算法去噪后的波形与频谱,可以看出:VMD-WT算法优于EMD-WT、EEMD-WT,说明了VMD方法在复杂信号分解方面更具优势。FIR与IIR滤波的效果比较接近,去噪效果仍低于本文优化的WOA-VMD-WT方法。利用本文算法可以更好地去除信号中的次要频率干扰,去噪后的信号波形更加平稳,波动偏差更小,进一步证明了该算法的有效性与实用价值。
图9 实测信号波形与频谱
图10 不同算法的实测信号去噪效果图
6 结语
本文根据水声信号的特点,提出一种基于改进WOA-VMD和小波阈值处理结合的去噪方法。在进行小波阈值处理之前,引入VMD算法对含噪信号进行分解,将复杂的信号划分到不同的频段进行分析,并根据各模态分量的中心频率以及与含噪信号的互相关系数,最终确定有效分量。从而有利于根据应用场景和目标信号的特点来进一步提取有用信息,提高目标信号的信噪比。
此外,针对信号VMD处理过程中参数选择的关键影响,将需要选择的参数组合(K,α)作为种群寻优的位置,选取功率谱熵作为适应度函数,然后利用WOA算法获得含噪信号VMD分解的最佳参数模型。为了提高模型的精度,引入余弦收敛因子和指数权重,改进传统WOA的寻优能力和收敛速度。实验证明,采用本文改进的WOA-VMD方法可以实现复杂信号在频域的有效分离,不会出现频谱混叠的现象。
在对有效分量进行小波阈值处理时,考虑到噪声小波系数随着分解尺度的不断增加而呈现出的衰减特性,本文提出改进的可调阈值函数和分层自适应阈值,改善了传统阈值量化过程中存在的固定偏差问题,可以根据含噪信号的特点,灵活调整阈值函数的形状以及设置合适的阈值。在不同噪声强度下进行仿真实验,结果表明了本文算法的去噪性能指标均优于其他七种算法,去噪的同时能够保留更多的细节信息。实测信号的去噪对比分析,也进一步验证本文算法可以有效去除水声信号中的次要频率干扰,去噪后的波形比较光滑,更加接近目标信号,说明该算法在信号去噪方面具有一定的实用性与优越性。