一种基于CWGAN-GP和双流CNN模型的离心泵转子故障诊断方法

一种基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法
技术领域
1.本发明属于旋转机械故障信号分析技术研究领域，具体涉及一种基于cwgan-gp(条件梯度惩罚沃瑟斯坦生成对抗网络)和双流cnn(卷积神经网络)模型的离心泵转子故障诊断方法。


背景技术：

2.不平衡与不对中是离心泵常见的转子故障类型，当发生转子不平衡与不对中故障时，整个机组的安全性都将受到严重威胁。因此，开展离心泵转子故障诊断技术的研究对于确保工业生产安全，降低能源消耗以及维修成本，提高生产效率具有至关重要的作用。目前，随着大数据、云计算、人工智能ai等技术的快速发展，离心泵转子的故障诊断已从传统的依赖专家经验逐渐转向基于深度学习的智能诊断。
3.目前基于深度学习的离心泵转子故障诊断方法还存在以下几个问题有待解决：(1)故障信号预处理。由于振动位移信号中常伴有大量的噪声成分，难以直接从中提取出明显的转子故障特征。因此，如何剔除原始信号中的噪声成分，提取出有用的故障特征信息已经成为离心泵转子故障诊断的研究难点之一。(2)故障识别。基于深度学习模型的故障识别是离心泵转子故障诊断的重要环节，深度学习模型通常具有较高的复杂度和参数量，需要进行有效的模型优化和调参，以提高模型的泛化能力和诊断准确度。(3)故障样本获取困难。由于离心泵长期处于健康状态，所采集的振动位移数据大多为健康状态样本，故障样本少，造成了数据集中不同类别的样本数量非均衡的问题。因此，如何获取足够多的故障样本成为了离心泵转子故障诊断的瓶颈之一。


技术实现要素：

4.针对现有技术中存在不足，本发明提供了一种基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，从而提高离心泵转子故障诊断的效率和准确性。
5.本发明是通过以下技术手段实现上述技术目的的。
6.一种基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法：
7.采集离心泵转子正常状态、转子不平衡故障、转子不对中故障三种状态下的振动位移信号，并经过重构后得到新的振动位移信号，组成cwgan-gp模型的训练集；
8.将所述训练集与各类故障对应的标签信息输入cwgan-gp模型进行训练，得到训练好cwgan-gp模型；再将各类故障对应的标签信息和随机噪声输入训练好的cwgan-gp模型，生成对应故障类型的数据，并将生成数据添加至转子不平衡故障、转子不对中故障的振动位移信号集中，得到均衡数据集；
9.利用所述均衡数据集对双流cnn模型进行训练，再将振动位移信号与二维时频图像数据输入训练好的双流cnn模型中，实现离心泵转子不同类型故障的分类。
10.进一步地，所述cwgan-gp模型是基于cgan与wgan-gp模型建立的：采用
wasserstein距离代替传统gan中的交叉熵损失函数；利用gp对判别器进行惩罚，强制判别器满足lipschitz连续条件；增加额外的数据标签信息。
11.更进一步地，cwagn-gp模型中的鉴别器由12个卷积层、2个最大池化层、6个bn层以及1个全连接层构成；所述卷积层的卷积核大小均为5
×
1，各卷积层的卷积核的数量依次为8、8、16、16、32、32、64、64、128、128、256、256，卷积层1、卷积层3、卷积层5、卷积层7、卷积层9、卷积层11的卷积计算的步长为2，卷积层2、卷积层4、卷积层6、卷积层8、卷积层10、卷积层12的卷积计算的步长为1，所有卷积层均使用leakyrelu激活函数；所述最大池化层设置在bn层2、bn层4之后，最大池化层1与最大池化层2的池化核大小均为5
×
1、步长为2；所述全连接层删除激活函数且不加入偏置计算；bn层设置在卷积层2、卷积层4、卷积层6、卷积层8、卷积层10和卷积层12之后。
12.更进一步地，cwagn-gp模型中的生成器由1个全连接层、13个卷积层、3个上采样层和6个bn层构成；所述全连接层由256个神经元组成；卷积层1、卷积层2均由256个大小为5
×
1的卷积核组成，卷积层3、卷积层4均由128个大小为5
×
1的卷积核组成，卷积层5、卷积层6均由64个大小为5
×
1的卷积核组成，卷积层7、卷积层8均由32个大小为5
×
1的卷积核组成，卷积层9、卷积层10均由16个大小为5
×
1的卷积核组成，卷积层11、卷积层12均由8个大小为5
×
1的卷积核组成，卷积层13由1个大小为5
×
1的卷积核组成，所有卷积层的卷积步长为1，卷积层13之后采用than激活函数，其余卷积层之后均采用relu激活函数；上采样层大小均设置为2；bn层设置在卷积层2、卷积层4、卷积层6、卷积层8、卷积层10和卷积层12之后。
13.进一步地，所述双流cnn模型由输入层、改进1d lenet-5模型、改进2d lenet-5模型、连接层、全连接层以及输出层构成。
14.更进一步地，所述改进1d lenet-5模型，包括：
15.输入层的大小改为1024
×
1；
16.将3
×
3、5
×
5大小的二维卷积核改为3
×
1、5
×
1大小的一维卷积核，第一层卷积中采用64个卷积核，第二层卷积中采用128个卷积核。
17.更进一步地，所述改进2d lenet-5模型，包括：
18.选择64
×
64大小的rgb图像，第一层卷积中采用64个卷积核、卷积核的大小为3
×
3，第二层卷积中采用128个卷积核、卷积核的大小为5
×
5；
19.删除第二个全连接层，并在全连接层之后增加dropout操作，在每个激活层后增加bn操作；
20.采用relu激活函数代替传统lenet-5中sigmoid激活函数。
21.进一步地，对振动位移信号进行重构的具体过程为：
22.1)通过ipso算法对vmd分解振动位移信号时的参数k与a进行寻优，获得最佳参数组合；
23.2)采用优化后的vmd分解原始振动位移信号，得到k个imf分量；
24.3)计算各个imf分量的kld值，并将kld值进行升序排序；
25.4)筛选出kld值最小的imf分量作为有效imf分量用于重构新的振动位移信号。
26.更进一步地，所述ipso算法是对pso算法中的tent混沌映射进行改进、对惯性权重与学习因子进行调整后的算法。
27.更进一步地：
28.tent混沌映射采用如下迭代公式进行改进：
[0029][0030]
其中，xi表示为第i个粒子，α、β为[0，0.1]间的随机数；
[0031]
对惯性权重与学习因子进行调整：
[0032][0033][0034][0035]
其中，w为惯性权重，c1为个体学习因子，c2为群体学习因子，δ1、δ2、δ3为常数，ε为指定迭代次数，k为当前迭代次数，n
max
为最大迭代次数，c
1up
、c
1low
为粒子个体学习因子的上下界，c
2up
、c
2low
为粒子群体学习因子的上下界，ρ为[0,1]之间的随机数。
[0036]
本发明的有益效果为：
[0037]
(1)本发明基于ipso-vmd-kld从原始振动位移信号中提取离心泵转子故障特征，实现了对离心泵转子故障信号的有效重构，提高了转子故障特征提取的准确性。
[0038]
(2)本发明将各类故障对应的标签信息和随机噪声输入训练好的cwgan-gp模型，生成对应故障类型的数据，并将生成数据添加至转子不平衡故障、转子不对中故障的振动位移信号集中，得到均衡数据集；解决了离心泵转子故障数据样本有限、获取困难，导致数据集中正常状态样本与故障状态样本数量之间非均衡的问题。
[0039]
(3本发明利用均衡数据集对双流cnn模型进行训练，再将振动位移信号与二维时频图像数据输入训练好的双流cnn模型中，实现离心泵转子不同类型故障的分类；解决了传统故障诊断方法费时费力以及现有深度学习模型只能从单一维度提取特征的问题。
附图说明
[0040]
图1为本发明所述cwagn-gp模型的结构示意图；
[0041]
图2为本发明所述鉴别器结构示意图；
[0042]
图3为本发明所述生成器结构示意图；
[0043]
图4为本发明所述基于cwgan-gp的离心泵转子故障数据生成流程图；
[0044]
图5为本发明所述双流cnn故障诊断模型结构示意图；
[0045]
图6为本发明所述基于cwgan-gp与双流cnn的离心泵转子故障诊断验证方案流程图；
[0046]
图7为本发明所述数据集扩充后双流cnn模型故障诊断结果图；
[0047]
图8为本发明所述双流cnn模型对扩充后达到平衡的数据集故障诊断结果图；
[0048]
图中：图中：1-1d输入层，2、5-1d卷积层，3、6-1d bn层，4、7-1d最大池化层，8、17-flatten层，9、18、20-全连接层，10-2d输入层，11、14-2d卷积层，12、15-2d bn层，13、16-2d最大池化层，19-连接层，21-输出层。
具体实施方式
[0049]
下面结合附图以及具体实施例对本发明作进一步的说明，但本发明的保护范围并不限于此。
[0050]
本发明一种基于cwgan-gp与双流cnn模型的离心泵转子故障诊断方法，具体包括如下步骤：
[0051]
步骤一、基于ipso-vmd-kld从原始振动位移信号中提取离心泵转子故障特征
[0052]
(1)为了提高pso算法的收敛速度与收敛精度，一方面对pso算法中的tent混沌映射进行改进，另一方面对pso算法中的惯性权重与学习因子进行调整，通过上述改进和调整后的算法，称为ipso算法，具体地：
[0053]
tent混沌映射采用如下迭代公式进行改进：
[0054][0055]
其中，xi表示为第i个粒子，α、β为[0，0.1]间的随机数，当xi+1》1时，返回值等于1，当xi+1《0时，返回值等于0。由此提高pso算法种群中粒子初始化后的质量。
[0056]
对惯性权重与学习因子进行调整，具体采用(2)-(4)所示：
[0057][0058][0059][0060]
其中，w为惯性权重，c1为个体学习因子，c2为群体学习因子，δ1、δ2、δ3为常数，ε为指定迭代次数，k为当前迭代次数，n
max
为最大迭代次数，c
1up
、c
1low
为粒子个体学习因子的上下界，c
2up
、c
2low
为粒子群体学习因子的上下界，ρ为[0,1]之间的随机数。
[0061]
ipso算法流程，具体包括：
[0062]
1)设置ipso算法中粒子群的规模、最大迭代次数、指定迭代次数、学习因子，并基于改进的tent混沌映射初始化粒子群中粒子的速度和位置；
[0063]
2)根据式(2)～(4)计算粒子的惯性权重、个体学习因子和群体学习因子；
[0064]
3)根据待求解的适应度函数计算粒子群中每个粒子的适应度值；
[0065]
4)更新每个粒子的历史最优位置和群体最优值；
[0066]
5)判断ipso算法是否达到了终止迭代条件，如果满足，则停止迭代计算，并且返回最优值，如果不满足，则返回2)继续迭代，直至满足终止条件为止。
[0067]
(2)采用ipso算法对vmd(变分模态分解)的参数进行自动寻优，实现振动位移信号的自适应分解，具体流程包括：
[0068]
1)初始化ipso算法的参数值与vmd参数k、a的取值范围，生成初始粒子种群，并将排列熵作为ipso算法的适应度函数；
[0069]
2)对原始振动位移信号进行vmd分解，得到k个imf分量；
[0070]
3)计算k个imf分量的排列熵，将计算所得到的排列熵进行排序，得到该粒子种群下的最小排列熵，并获取到最小排列熵所处的粒子的位置；
[0071]
4)更新粒子种群中粒子的位置和速度，以及粒子的历史最优位置和群体最优位置；
[0072]
5)重复步骤2)至步骤4)，直至达到迭代终止条件；
[0073]
6)迭代完成后，输出最佳的粒子位置，即为最佳参数组合(k,a)。
[0074]
(3)基于kld(相对熵)建立有效imf分量筛选准则，通过ipso-vmd-kld实现对振动位移信号中转子故障特征的提取
[0075]
有效imf分量筛选准则具体为：挑选出kld值最小即与原始振动位移信号最相近且包含故障特征丰富的imf分量作为重构新的振动位移信号的有效imf分量，kld的计算公式如式(5)所示：
[0076][0077]
其中，p(x)为目标分布，q(x)为近似分布，m是分布的长度，xk为时域序列，i＝1，2，3，
…
，n，n为序列大小。
[0078]
利用ipso-vmd-kld实现对振动位移信号中转子故障特征的提取，具体包括：
[0079]
1)通过ipso算法对vmd分解振动位移信号时的参数k与a进行寻优，获得最佳参数组合；
[0080]
2)采用优化后的vmd分解原始振动位移信号，得到k个imf分量；
[0081]
3)计算各个imf分量的kld值，并将kld值进行升序排序；
[0082]
4)筛选出kld值最小的imf分量作为有效imf分量用于重构新的振动位移信号，最终完成对转子故障特征的提取目的。
[0083]
步骤二、建立cwgan-gp模型，改善故障特征数据与非故障数据之间的非均衡问题
[0084]
(1)结合cgan(条件生成对抗网络)与wgan-gp模型，建立条件梯度惩罚沃瑟斯坦生成对抗网络(cwgan-gp)
[0085]
首先，采用wasserstein距离代替传统gan(生成对抗网络)中的交叉熵损失函数，以优化生成器和判别器的训练；
[0086]
然后，利用gp对判别器进行惩罚，强制判别器满足lipschitz连续条件，使判别器更好地区分真实数据(原始振动位移信号)和生成数据(经cwgan-gp模型生产的数据)，减少模式塌陷问题；
[0087]
最后，增加额外的数据标签信息，使cwgan-gp更好地控制与指导生成器生成的数据，cwagn-gp模型的结构示意图如图1所示，目标函数为：
[0088][0089]
其中，d为鉴别器，g为生成器，v(d,g)为目标函数，x为真实数据，为真实数据x与生成数据g(z)的线性插值，pr为真实数据分布，pg为生成数据分布，z表示随机噪声，λ为梯度惩罚系数，y为数据标签(例如正常为0，不平衡数据为1)，e[]表示数学期望，为梯度算子。
[0090]
cwagn-gp模型中的鉴别器的输入层为1024维真实数据或生成数据与各故障类型标签信息融合构成的1024维数据，输出为对真实数据和生成数据的判别结果，鉴别器的结构如图2所示。具体地，鉴别器由12个卷积层、2个最大池化层、6个bn层以及1个全连接层构成。为了从输入数据中获得更多的局部信息，卷积层的卷积核大小均为5
×
1，各卷积层的卷积核的数量依次为8、8、16、16、32、32、64、64、128、128、256、256，卷积层1、卷积层3、卷积层5、卷积层7、卷积层9、卷积层11的卷积计算的步长为2，卷积层2、卷积层4、卷积层6、卷积层8、卷积层10、卷积层12的卷积计算的步长为1，所有卷积层均使用leakyrelu激活函数。在bn层2、bn层4后采用最大池化层对特征进行降维以降低鉴别器的训练难度，其中最大池化层1与最大池化层2的池化核大小均为5
×
1、步长为2。最后一层全连接层去除激活函数且不加入偏置计算。为了加快生成器模型的训练速度，卷积层2、卷积层4、卷积层6、卷积层8、卷积层10以及卷积层12后加入了bn层。
[0091]
生成器中输入层为140维均匀分布随机噪声与各故障类型标签融合组成的140维数据，最终的输出为1024维生成数据，生成器的结构如图3。具体地，全连接层由256个神经元组成，将随机噪声数据扩展至256维。卷积层1由256个大小为5
×
1的卷积核组成，卷积层2由256个大小为5
×
1的卷积核组成，卷积层3由128个大小为5
×
1的卷积核组成，卷积层4由128个大小为5
×
1的卷积核组成，卷积层5由64个大小为5
×
1的卷积核组成，卷积层6由64个大小为5
×
1的卷积核组成，卷积层7由32个大小为5
×
1的卷积核组成，卷积层8由32个大小为5
×
1的卷积核组成，卷积层9由16个大小为5
×
1的卷积核组成，卷积层10由16个大小为5
×
1的卷积核组成，卷积层11由8个大小为5
×
1的卷积核组成，卷积层12由8个大小为5
×
1的卷积核组成，卷积层13由1个大小为5
×
1的卷积核组成。所有卷积层的卷积步长为1，在卷积层13之后采用than激活函数，其余卷积层之后均采用relu激活函数。上采样层1、上采样层2、上采样层3大小均设置为2。与鉴别器相同，卷积层2、卷积层4、卷积层6、卷积层8、卷积层10以及卷积层12后增加了bn层。
[0092]
(2)将训练集与各类故障对应的标签信息输入cwgan-gp模型进行训练，将输入标签信息和随机噪声输入生成器，生成对应故障类型的数据，完成数据生成任务。
[0093]
将不同故障类型的原始振动位移信号进行信号重构后得到新的振动位移信号，利用新的振动位移信号构建cwgan-gp模型训练集。
[0094]
将训练集与各类故障对应的标签信息输入cwgan-gp模型进行训练。在训练过程中，采用rmsprop优化器进行参数优化，生成器的初始学习率设置为0.0005，鉴别器的初始学习率设置为0.0001，模型梯度惩罚项的系数λ为10。采用生成器和鉴别器分开训练的策
略。首先设置生成器为不可训练状态，对鉴别器进行训练实现参数更新，随后固定鉴别器，训练生成器。每次迭代过程中鉴别器训练8次，生成器训练2次。通过不断地迭代训练，最终鉴别器与生成器达到纳什均衡，得到训练好cwgan-gp模型。
[0095]
将标签信息和随机噪声输入生成器，生成对应故障类型的数据，完成数据生成任务。基于cwgan-gp模型的离心泵转子数据生成流程如图4。
[0096]
步骤三、建立双流cnn模型
[0097]
(1)改进2d lenet-5模型和1d lenet-5模型
[0098]
1)改进2d lenet-5模型，包括：
[0099]
首先，更改输入层大小，增加卷积核的数量。具体地，选择64
×
64大小的rgb图像。第一层卷积中采用64个卷积核，卷积核的大小为3
×
3，第二层卷积中采用128个卷积核，卷积核的大小为5
×
5。
[0100]
然后，减少全连接层层数，具体地，删除第二个全连接层，并在全连接层之后增加dropout操作，在每个激活层后增加批量归一化(batch normalization，bn)操作。
[0101]
最后，采用relu激活函数代替传统lenet-5中sigmoid激活函数。
[0102]
2)改进1d lenet-5模型，包括：
[0103]
首先，更改输入层尺寸大小，具体地，将输入层的大小改为1024
×
1；
[0104]
然后，更改卷积核的尺寸，具体地，将3
×
3、5
×
5大小的二维卷积核改为3
×
1、5
×
1大小的一维卷积核，且各卷积层的卷积核数量与改进2d lenet-5模型相同。
[0105]
(2)采用双流架构建立双流cnn故障诊断模型
[0106]
基于改进1d lenet-5模型与改进2d lenet-5模型，采用了双流架构建立双流cnn故障诊断模型，结构示意图如图5，详细结构如表1所示。双流cnn模型主要由输入层、改进1dlenet-5模型、改进2d lenet-5模型、连接层、全连接层以及输出层构成。其中输入层的数据分别为一维振动位移数据(原始振动位移信号)与二维时频图像数据。连接层由全连接层构成，用于融合改进1d lenet-5模型与改进2d lenet-5模型提取到的深度不变特征，通过连接层将不同维度的特征进行双流融合，兼顾了不同维度的数据，能够提取到数据中更多的特征信息，提高模型的故障诊断精度。输出层采用softmax分类器对不同类型的故障进行分类，最终实现故障诊断任务。
[0107]
表1双流cnn模型结构参数
[0108][0109]
(3)对优化器、学习率、训练批量以及dropout值等超参数进行设置
[0110]
具体的，优化器为adam，学习率为0.0001，训练批量为8，dropout值为0.4。
[0111]
步骤四、将cwgan-gp模型与双流cnn模型相结合，实现数据非均衡状态下离心泵转子的故障诊断
[0112]
通过训练好的cwgan-gp模型对原始非均衡数据集(原始振动位移信号)进行扩充，使得各类故障数据与正常状态数据达到数量间的平衡；
[0113]
采用不同的非均衡数据集与扩充后达到平衡的数据集对双流cnn模型进行模型训练，并运用训练后的模型对离心泵转子故障进行诊断。
[0114]
基于cwgan-gp与双流cnn的离心泵转子故障诊断效果验证实例，具体流程图如图6所示：
[0115]
(1)采用非均衡数据集中的故障训练集对cwgan-gp模型进行训练，通过不断地迭代训练使得模型达到纳什均衡状态；
[0116]
(2)通过训练好的cwgan-gp模型生成指定类型的故障数据，并将其添加至原始非均衡数据集中，对原始非均衡数据集进行扩充，使得各类故障数据与正常状态数据达到数量间的平衡；
[0117]
(3)搭建双流cnn模型，并选择最佳超参数组合初始化双流cnn模型；
[0118]
(4)采用不同的非均衡数据集与扩充后达到均衡的数据集对双流cnn模型进行训练；
[0119]
(5)通过验证集(表3)对训练好的双流cnn模型进行验证，得到不同的非均衡数据集与扩充后达到平衡的数据集的故障诊断准确率，并进行对比分析。
[0120]
离心泵转子故障模拟试验，具体如下：
[0121]
在离心泵转子故障模拟实验过程中，离心泵按照其额定转速(2900r/min)进行运转，将离心泵的运行流量分别调节为7.42m3/h、9.01m3/h、10.6m3/h、12.72m3/h。每种运行流量下，使用信号采集系统分别采集正常状态、转子不平衡故障、转子不对中故障三种状态下的振动位移信号。在数据采集时，为了符合奈奎斯特定律，即采样频率要大于分析频率的2倍，本实验的采样频率设置为25600hz，采样时间设置为1s。转子不平衡故障与转子不对中故障的模拟实验方案如下，每种故障类型下采集的数据量如表2所示。
[0122]
(1)转子不平衡故障模拟方案
[0123]
转子不平衡故障模拟实验是通过在弹性联轴器的联结螺栓处增加垫片来模拟转子不平衡故障，其中垫片的重量分别为2.5g、6.5g、9g；加入垫片后，拧紧弹性联轴器螺栓，模拟离心泵在运行过程中产生转子不平衡故障。
[0124]
(2)转子不对中故障模拟方案
[0125]
转子不对中故障模拟实验是通过移动电机端的位置来模拟转子平行不对中故障，并采用千分表对转子平行不对中量进行测量，造成离心泵运行过程中的转子平行不对中故障，其中平行不对中量分别为30丝、40丝、50丝。
[0126]
表2离心泵转子故障模拟实验数据采集数量表
[0127][0128]
构建非均衡数据集：
[0129]
按照30：1、12：1、6：1、3：1、2：1的非均衡数据比例设置a、b、c、d、e共五个数据集(采集原始振动位移信号并重构)来模拟不同程度的数据非均衡情况，构建不同非均衡数据比例数据集，如表3所示。
[0130]
表3非均衡数据集
[0131][0132]
基于cwgan-gp与双流cnn和故障诊断结果验证：
[0133]
在实验过程中，采用双流cnn模型对非均衡数据集和cwgan-gp模型扩充后的数据集分别进行10次独立重复训练与验证。双流cnn模型在非均衡数据集上的10次实验结果如图7所示。将采用cwgan-gp模型扩充后的数据集分别记为a'、b'、c'、d'、e'，在cwgan-gp模型扩充后的数据集上的10次实验结果如图8所示。
[0134]
根据图7和图8可以得出，采用生成数据对非均衡数据集进行扩充之后，双流cnn模型的验证准确率有了明显的提升，且波动范围显著缩小。对比非均衡数据集，数据集扩充后，模型的最高验证准确率分别提高了16％、17.33％、8.66％、4.33％、1.33％。
[0135]
所述实施例为本发明的优选的实施方式，但本发明并不限于上述实施方式，在不背离本发明的实质内容的情况下，本领域技术人员能够做出的任何显而易见的改进、替换或变型均属于本发明的保护范围。

技术特征：
1.一种基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于：采集离心泵转子正常状态、转子不平衡故障、转子不对中故障三种状态下的振动位移信号，并经过重构后得到新的振动位移信号，组成cwgan-gp模型的训练集；将所述训练集与各类故障对应的标签信息输入cwgan-gp模型进行训练，得到训练好cwgan-gp模型；再将各类故障对应的标签信息和随机噪声输入训练好的cwgan-gp模型，生成对应故障类型的数据，并将生成数据添加至转子不平衡故障、转子不对中故障的振动位移信号集中，得到均衡数据集；利用所述均衡数据集对双流cnn模型进行训练，再将振动位移信号与二维时频图像数据输入训练好的双流cnn模型中，实现离心泵转子不同类型故障的分类。2.根据权利要求1所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，所述cwgan-gp模型是基于cgan与wgan-gp模型建立的：采用wasserstein距离代替传统gan中的交叉熵损失函数；利用gp对判别器进行惩罚，强制判别器满足lipschitz连续条件；增加额外的数据标签信息。3.根据权利要求2所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，cwagn-gp模型中的鉴别器由12个卷积层、2个最大池化层、6个bn层以及1个全连接层构成；所述卷积层的卷积核大小均为5
×
1，各卷积层的卷积核的数量依次为8、8、16、16、32、32、64、64、128、128、256、256，卷积层1、卷积层3、卷积层5、卷积层7、卷积层9、卷积层11的卷积计算的步长为2，卷积层2、卷积层4、卷积层6、卷积层8、卷积层10、卷积层12的卷积计算的步长为1，所有卷积层均使用leakyrelu激活函数；所述最大池化层设置在bn层2、bn层4之后，最大池化层1与最大池化层2的池化核大小均为5
×
1、步长为2；所述全连接层删除激活函数且不加入偏置计算；bn层设置在卷积层2、卷积层4、卷积层6、卷积层8、卷积层10和卷积层12之后。4.根据权利要求2所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，cwagn-gp模型中的生成器由1个全连接层、13个卷积层、3个上采样层和6个bn层构成；所述全连接层由256个神经元组成；卷积层1、卷积层2均由256个大小为5
×
1的卷积核组成，卷积层3、卷积层4均由128个大小为5
×
1的卷积核组成，卷积层5、卷积层6均由64个大小为5
×
1的卷积核组成，卷积层7、卷积层8均由32个大小为5
×
1的卷积核组成，卷积层9、卷积层10均由16个大小为5
×
1的卷积核组成，卷积层11、卷积层12均由8个大小为5
×
1的卷积核组成，卷积层13由1个大小为5
×
1的卷积核组成，所有卷积层的卷积步长为1，卷积层13之后采用than激活函数，其余卷积层之后均采用relu激活函数；上采样层大小均设置为2；bn层设置在卷积层2、卷积层4、卷积层6、卷积层8、卷积层10和卷积层12之后。5.根据权利要求1所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，所述双流cnn模型由输入层、改进1d lenet-5模型、改进2d lenet-5模型、连接层、全连接层以及输出层构成。6.根据权利要求5所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，所述改进1d lenet-5模型，包括：输入层的大小改为1024
×
1；将3
×
3、5
×
5大小的二维卷积核改为3
×
1、5
×
1大小的一维卷积核，第一层卷积中采用64个卷积核，第二层卷积中采用128个卷积核。
7.根据权利要求5所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，所述改进2d lenet-5模型，包括：选择64
×
64大小的rgb图像，第一层卷积中采用64个卷积核、卷积核的大小为3
×
3，第二层卷积中采用128个卷积核、卷积核的大小为5
×
5；删除第二个全连接层，并在全连接层之后增加dropout操作，在每个激活层后增加bn操作；采用relu激活函数代替传统lenet-5中sigmoid激活函数。8.根据权利要求1所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，对振动位移信号进行重构的具体过程为：1)通过ipso算法对vmd分解振动位移信号时的参数k与a进行寻优，获得最佳参数组合；2)采用优化后的vmd分解原始振动位移信号，得到k个imf分量；3)计算各个imf分量的kld值，并将kld值进行升序排序；4)筛选出kld值最小的imf分量作为有效imf分量用于重构新的振动位移信号。9.根据权利要求8所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于，所述ipso算法是对pso算法中的tent混沌映射进行改进、对惯性权重与学习因子进行调整后的算法。10.根据权利要求8所述的基于cwgan-gp和双流cnn模型的离心泵转子故障诊断方法，其特征在于：tent混沌映射采用如下迭代公式进行改进：其中，x
i
表示为第i个粒子，α、β为[0，0.1]间的随机数；对惯性权重与学习因子进行调整：对惯性权重与学习因子进行调整：对惯性权重与学习因子进行调整：其中，w为惯性权重，c1为个体学习因子，c2为群体学习因子，δ1、δ2、δ3为常数，ε为指定迭代次数，k为当前迭代次数，n
max
为最大迭代次数，c
1up
、c
1low
为粒子个体学习因子的上下界，c
2up
、c
2low
为粒子群体学习因子的上下界，ρ为[0,1]之间的随机数。

技术总结
本发明公开了一种基于CWGAN-GP和双流CNN模型的离心泵转子故障诊断方法，首先采集振动位移信号，并经过重构后得到新的振动位移信号，组成CWGAN-GP模型的训练集；将训练集与各类故障对应的标签信息输入CWGAN-GP模型进行训练，再将各类故障对应的标签信息和随机噪声输入训练好的CWGAN-GP模型，生成对应故障类型的数据，并将生成数据添加至转子不平衡故障、转子不对中故障的振动位移信号集中，得到均衡数据集；利用均衡数据集对双流CNN模型进行训练，再将振动位移信号与二维时频图像数据输入训练好的双流CNN模型中，实现离心泵转子不同类型故障的分类。本发明提高离心泵转子故障诊断的效率和准确性。断的效率和准确性。断的效率和准确性。


技术研发人员：董亮 周润泽 陈泽宇 刘厚林 代翠
受保护的技术使用者：江苏大学
技术研发日：2023.07.14
技术公布日：2023/10/20