基于原型和双重域对抗的旋转机械跨工况故障诊断方法技术

一种基于原型和双重域对抗的旋转机械跨工况故障诊断方法，先使用采集设备从待监测旋转机械采集各工况运行条件下的振动信号，建立不同域(工况)的数据集；然后构建基于原型和双重域对抗的域适应网络，并使用梯度下降法训练域适应网络；再将需要诊断的目标域输入到训练好的域适应网络中；最后获取旋转机械的跨工况故障诊断结果；本发明专利技术同时对齐了边缘分布和条件分布，能够更实现细粒度的类级别结构匹配，实现了诊断知识从源域到目标域的迁移，有效实现端到端的跨工况旋转机械的状态识别。现端到端的跨工况旋转机械的状态识别。现端到端的跨工况旋转机械的状态识别。

【技术实现步骤摘要】
基于原型和双重域对抗的旋转机械跨工况故障诊断方法


[0001]本专利技术属于机械设备智能诊断
，具体涉及基于原型和双重域对抗的旋转机械跨工况故障诊断方法。

技术介绍

[0002]旋转机械作为现代制造系统中的主要机械设备，为实现现代制造智能化、快速化、高效化的重要保障。当前，旋转机械正朝着高度复杂化和集成化的方向快速发展，一旦其中的关键部件发生故障，将可能造成巨大的经济损失和人员伤亡。因此，面向旋转机械的智能诊断方法具有重大的意义。
[0003]近年来，基于深度学习的智能故障诊断方法逐渐受到学术界和工业界的关注。然而，应用传统的监督式的深度学习方法(申请号CN201910991173.7、名称为：基于深度卷积域对抗迁移学习的故障诊断方法)，对旋转机械进行故障诊断时，存在以下两个缺点：1，旋转机械的工况变化会导致设备在某一工况的运行中采集的样本和另一工况下采集的样本存在明显的分布差异，造成某一工况下训练的深度学习模型在另一工况的应用效果不佳；2，单一工况下的样本下缺少足够的标签，造成实际应用中不可能对每一工况进行建模。因此，域适应方法(申请号CN201910463197.5、名称为：一种对抗判别域适应一维卷积神经网络智能故障诊断方法)受到研究人员的关注，在机械设备诊断领域中，域适应方法能够利用源域(带标签信息的某一工况)中的诊断知识，解决目标域(无标签信息的另一工况)中的诊断任务；但是，单一的域适应方法的故障诊断精度不高，无法有效地对旋转机械的状态进行辨识。

技术实现思路

[0004]为了克服现有技术的缺点，本专利技术的目的是提供一种基于原型和双重域对抗的旋转机械跨工况故障诊断方法，采用的域适应网络结构简单，能够同时适应不同工况下的边缘分布和条件分布，在不同类型的旋转机械下准确地进行故障辨识，实现对旋转机械的智能故障诊断。
[0005]为了达到上述目的，本专利技术采用的技术方案为：
[0006]一种基于原型和双重域对抗的旋转机械跨工况故障诊断方法，包括以下步骤：
[0007]步骤一：通过信号采集设备采集旋转机械在不同工况下的振动数据，每个工况下对应不同的域；
[0008]步骤二：搭建基于原型和双重域对抗的域适应网络，该域适应网络包括一个特征学习器F及可训练参数θ
F
、一个状态分类器C及可训练参数θ
C
、一个原型学习器P、一个边缘域对抗器D
M
及可训练参数θ
DM
，和多个条件域对抗器D
C
及对应的可训练参数；其中，特征学习器提取源域和目标域的高层次隐含特征，状态分类器学习一个类分离的特征空间，原型学习器用于学习表征源域中每一类样本的原型，边缘域对抗器从整体上对齐源域分布和目标域分布，多个条件域对抗器对齐特征空间中的源域和目标域每一个类的结构；边缘域对抗器
与条件域对抗器和特征学习器通过梯度反转层(Gradient reversal layer，GRL)连接，且条件域对抗器的个数和旋转机械的状态类别数一致；
[0009]对于一个旋转机械的跨工况诊断任务，源域数据集为其中n
s
分别表示源域样本、源域样本标签、源域样本数量，目标域数据集为其中n
t
分别表示目标域样本、目标域样本数量；利用交叉熵损失函数L
cel
，状态分类器C的分类损失L
cls
表述为：
[0010][0011]其中F(x
i
)为特征学习器提取样本x
i
的特征，C(F(x
i
))为分类器的softmax输出；
[0012]原型学习器被用于学习源域每一类样本的原型特征，第k类样本的原型p
k
的学习过程如下所示：
[0013][0014]其中n
sk
表示源域数据集中k类样本的数量，K为诊断任务的总类别数；
[0015]利用边缘域对抗器对齐源域和目标域的边缘分布，且利用源域的原型来代表源域的所有样本，并且对源域和目标域的样本打上域标签，源域d
i
(i＝1,2,
…
,K)和目标域d
i
(i＝1,2,
…
,n
t
)的域标签分别是1和0，边缘域对抗器D
M
的对抗损失L
m
如下所示：
[0016][0017]其中f
i
为源域原型特征和目标域特征的集合，L
bcel
为二元交叉熵损失函数；
[0018]同理，利用条件域对抗器对齐源域和目标域的子类结构，条件域对抗器D
C
的对抗损失L
c
如下所示：
[0019][0020]其中v
i
为源域原型特征和目标域特征与状态分类器softmax乘积的集合；
[0021]随着损失L
cls
的减小，域适应网络能够习得一个类分离的特征空间；随着双重域对抗损失L
m
和L
c
的增大，域适应网络不能够区分特征是来自源域或目标域，即习得域不变的特征空间，因此，域适应网络的总优化目标函数L如下所示：
[0022]L＝L
cls
‑
αL
m
‑
βL
c
[0023]其中，α和β分别是边缘分布对齐和条件分布对齐的权衡参数；
[0024]该目标函数通过随机梯度下降Adam算法优化，具体的参数更新规则如下：
[0025][0026][0027][0028][0029]其中，γ和分别为学习率和偏导；
[0030]步骤三：将目标域的样本输入到训练好的特征学习器和状态分类器中；
[0031]步骤四：获取旋转机械的跨工况故障诊断结果。
[0032]本专利技术的优点为：
[0033](1)本专利技术同时采取边缘域对抗器和条件域对抗器，结合双重域对抗学习，实现了源域和目标域特征空间的边缘分布对齐和条件分布对齐，促进域适应网络学习类别分离和域不变的特征空间；
[0034](2)本专利技术将原型学习和域对抗结合在一起，实现了源域特征空间和目标域特征空间的细粒度匹配，构造了一个能在不同工况中实现知识迁移的故障诊断模型；
[0035](3)本专利技术采用深度卷积神经网络为骨干结构，实现了端到端的特征提取和故障诊断，解决了目标域无标签预测的问题，提高了旋转机械跨工况诊断的精度。
附图说明
[0036]图1为本专利技术方法的流程图。
[0037]图2为实施例滚动轴承的实验平台。
[0038]图3为实施例滚动轴承(源域为1hp，目标域为2hp)故障诊断结果的评价指标示意图。
具体实施方式
[0039]下面结合附图和实施例对本专利技术进行详细说明。
[0040]参照图1，一种基于原型和双重域对抗的旋转机械跨工况故障诊断本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于原型和双重域对抗的旋转机械跨工况故障诊断方法，其特征在于，包括以下步骤：步骤一：通过信号采集设备采集旋转机械在不同工况下的振动数据，每个工况下对应不同的域；步骤二：搭建基于原型和双重域对抗的域适应网络，该域适应网络包括一个特征学习器F及可训练参数θ
F
、一个状态分类器C及可训练参数θ
C
、一个原型学习器P、一个边缘域对抗器D
M
及可训练参数θ
DM
，和多个条件域对抗器D
C
及对应的可训练参数；其中，特征学习器提取源域和目标域的高层次隐含特征，状态分类器学习一个类分离的特征空间，原型学习器用于学习表征源域中每一类样本的原型，边缘域对抗器从整体上对齐源域分布和目标域分布，多个条件域对抗器对齐特征空间中的源域和目标域每一个类的结构；边缘域对抗器与条件域对抗器和特征学习器通过梯度反转层(Gradient reversal layer，GRL)连接，且条件域对抗器的个数和旋转机械的状态类别数一致；对于一个旋转机械的跨工况诊断任务，源域数据集为其中n
s
分别表示源域样本、源域样本标签、源域样本数量，目标域数据集为其中n
t
分别表示目标域样本、目标域样本数量；利用交叉熵损失函数L
cel
，状态分类器C的分类损失L
cls
表述为：其中F(x
i
)为特征学习器提取样本x
i
的特征，C(F(x
i
))为分类器的softmax输出；原型学习器被用于学习源域每一类样本的原型特征，第k类样本的原型p
k
的学习过程如下所示：其中n
sk
表示源域数据集中k类样本的数量，K为诊断任务的总类别数；利用边缘域对抗器对...

【专利技术属性】
技术研发人员：徐光华，况佳臣，张四聪，陶唐飞，韩丞丞，吴庆强，魏帆，陈晟超，
申请(专利权)人：西安交通大学，
类型：发明
国别省市：