CN116754230A - 基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法 - Google Patents
基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法 Download PDFInfo
- Publication number
- CN116754230A CN116754230A CN202310489061.8A CN202310489061A CN116754230A CN 116754230 A CN116754230 A CN 116754230A CN 202310489061 A CN202310489061 A CN 202310489061A CN 116754230 A CN116754230 A CN 116754230A
- Authority
- CN
- China
- Prior art keywords
- training
- real
- bearing
- discriminator
- generator
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 238000003745 diagnosis Methods 0.000 title claims abstract description 106
- 238000001514 detection method Methods 0.000 title claims abstract description 82
- 238000000034 method Methods 0.000 title claims abstract description 63
- 230000005856 abnormality Effects 0.000 title claims abstract description 47
- 238000012549 training Methods 0.000 claims description 127
- 238000010586 diagram Methods 0.000 claims description 33
- 230000002159 abnormal effect Effects 0.000 claims description 18
- 238000004364 calculation method Methods 0.000 claims description 14
- 238000013527 convolutional neural network Methods 0.000 claims description 12
- 238000002372 labelling Methods 0.000 claims description 9
- 230000009466 transformation Effects 0.000 claims description 8
- 238000013528 artificial neural network Methods 0.000 claims description 3
- 238000013507 mapping Methods 0.000 claims description 3
- 239000013589 supplement Substances 0.000 claims description 3
- 230000004927 fusion Effects 0.000 abstract description 4
- 230000006870 function Effects 0.000 description 71
- 230000008569 process Effects 0.000 description 16
- 238000012360 testing method Methods 0.000 description 14
- ORILYTVJVMAKLC-UHFFFAOYSA-N Adamantane Natural products C1C(C2)CC3CC1CC2C3 ORILYTVJVMAKLC-UHFFFAOYSA-N 0.000 description 7
- 230000004913 activation Effects 0.000 description 7
- 238000004458 analytical method Methods 0.000 description 6
- 238000005096 rolling process Methods 0.000 description 6
- 238000009826 distribution Methods 0.000 description 5
- 238000011160 research Methods 0.000 description 5
- 230000008901 benefit Effects 0.000 description 4
- 239000002131 composite material Substances 0.000 description 4
- 238000002474 experimental method Methods 0.000 description 4
- 238000005457 optimization Methods 0.000 description 4
- 230000003213 activating effect Effects 0.000 description 3
- 238000013145 classification model Methods 0.000 description 3
- 238000010276 construction Methods 0.000 description 3
- 230000000694 effects Effects 0.000 description 3
- 230000001965 increasing effect Effects 0.000 description 3
- 238000004422 calculation algorithm Methods 0.000 description 2
- 238000013135 deep learning Methods 0.000 description 2
- 238000013461 design Methods 0.000 description 2
- 238000011161 development Methods 0.000 description 2
- 238000005516 engineering process Methods 0.000 description 2
- 238000013508 migration Methods 0.000 description 2
- 230000005012 migration Effects 0.000 description 2
- 239000013598 vector Substances 0.000 description 2
- 206010000372 Accident at work Diseases 0.000 description 1
- 238000013459 approach Methods 0.000 description 1
- 230000003190 augmentative effect Effects 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 239000002775 capsule Substances 0.000 description 1
- 230000002708 enhancing effect Effects 0.000 description 1
- 239000000284 extract Substances 0.000 description 1
- 238000000605 extraction Methods 0.000 description 1
- 238000010801 machine learning Methods 0.000 description 1
- 238000012423 maintenance Methods 0.000 description 1
- 238000004519 manufacturing process Methods 0.000 description 1
- 239000000463 material Substances 0.000 description 1
- 238000005259 measurement Methods 0.000 description 1
- 238000007781 pre-processing Methods 0.000 description 1
- 230000009467 reduction Effects 0.000 description 1
- 238000004088 simulation Methods 0.000 description 1
- 239000002689 soil Substances 0.000 description 1
Classifications
-
- G—PHYSICS
- G01—MEASURING; TESTING
- G01M—TESTING STATIC OR DYNAMIC BALANCE OF MACHINES OR STRUCTURES; TESTING OF STRUCTURES OR APPARATUS, NOT OTHERWISE PROVIDED FOR
- G01M13/00—Testing of machine parts
- G01M13/04—Bearings
- G01M13/045—Acoustic or vibration analysis
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/0464—Convolutional networks [CNN, ConvNet]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/0475—Generative networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/048—Activation functions
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Theoretical Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Artificial Intelligence (AREA)
- Biomedical Technology (AREA)
- Biophysics (AREA)
- Computational Linguistics (AREA)
- Health & Medical Sciences (AREA)
- Life Sciences & Earth Sciences (AREA)
- General Health & Medical Sciences (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Acoustics & Sound (AREA)
- Testing Of Devices, Machine Parts, Or Other Structures Thereof (AREA)
Abstract
一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,属于人工智能识别机械部件的技术领域,既可以实现对轴承类部件的异常检测,也可以实现对轴承的故障类别分类。本发明所述的网络模型在结构上做出相应改进,以及在功能上实现了对轴承部件的异常检测和故障诊断功能融合,还可以使用模型生成数据进行数据扩充,并且在异常检测以及故障诊断达到了较高的准确率。
Description
技术领域
本发明公开一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,属于人工智能识别机械部件的技术领域。
背景技术
随着工业技术的不断发展,工程机械设备异常检测和故障诊断受到了广泛的关注。在几乎所有类型的旋转机械中都会使用到滚动轴承,轴承工作环境复杂多变,而且高速运转的机械设备使滚动轴承更容易发生故障。滚动轴承还存在原始振动信号含有噪声和干扰过大、标签样本过少和负载迁移等问题。为了保证设备可靠并稳定运行,及时发现故障点并处理故障零件,可以避免出现大规模的经济损失甚至工业事故,异常检测和故障诊断成为智能制造背景下重要研究内容之一。
传统的基于振动信号的轴承异常检测和故障诊断多使用时域、频域分析或时频分析等方法。然而,轴承故障的振动信号具有复杂的非平稳时变特性,传统时频分析方法使用相对固定的时频分辨率处理非平稳信号,存在计算量大、诊断精度不理想等问题。
轴承异常检测方面旨在减少噪声干扰,提取高效特征精准进行异常检测。中国文献,张祥国为作者的《基于生成对抗网络的风电机组主轴承故障诊断研究》于2021年公开,提出了自编码器与软阈值结合利用快速傅里叶变换得到的频域信息对轴承进异常检测。越来越多的研究利用人工智能算法对原始故障数据集进行特征提取、融合、降维等一系列操作,最后实现异常检测和故障诊断。但是在很多实际应用过程中,正常样本和故障样本数量相差过大,会导致训练样本严重不平衡的问题。生成对抗网络因其强大的数据生成能力,在数据增强方面十分出色,成为了解决无监督和半监督问题的主要办法。
中国文献《基于生成对抗网络的风电机组主轴承故障诊断研究》通过对辅助分类生成对抗网络(ACGAN)的适应性进行改进,引入梯度惩罚,构建了改进ACGAN框架,其GANomaly模型将AE与GAN相结合,通过更高抽象空间中比较原始数据和重构数据的差异来推断异常。
在轴承故障诊断方面,按照轴承故障诊断的诊断手段分类,可分为基于电信号的故障诊断和基于振动信号的故障诊断两大类。第一类主要是通过分析故障电流和故障振动信号之间的内在关联之后,再利用对应的电流特征对轴承的故障类别进行识别鉴定。比如中国文献为2021年河南工业大学公开的作者柴志豪公开的《改进生成对抗网络的转子系统故障诊断方法研究》,利用时频分析方法进行数据预处理,将一维时序振动信号转换为时频图片后输入到卷积神经网络中,提出了二维卷积神经网络故障诊断模型,还提出基于生成数据的二维卷积神经网络故障诊断方法。通过实验验证了利用定子电流特征识别轴承故障的可行性。但是在大量实验研究中表明,利用电流信号实现高效故障诊断比利用振动信号实现故障诊断困难。第二类是通过轴承的振动状态实现故障诊断。
2022年公开在Measurement上,作者为Liu Yunpeng等人的《A conditionalvariational autoencoding generative adversarial networks with self-modulationfor rolling bearing fault diagnosis》,构建了一种集成了条件变分自编码器和条件瓦瑟斯坦生成对抗网络与自调制的新模型将数据扩充为平衡数据集。针对一维机械振动信号在输入卷积神经网络时无法充分提取相对位置关系的问题,提出一种基于格拉姆角场(GAF)和小尺寸卷积的胶囊网络的轴承故障诊断分类方法。
滚动轴承在几乎所有的旋转机械中都会被使用到。轴承故障影响范围大,维修成本高,因此对其做出精准快速的异常检测及故障分类是提高机械设备经济效益的有效措施。滚动轴承存在原始振动信号含有噪声和干扰过大、标签样本过少和负载迁移等问题,使得轴承故障样本数量在各种故障中占比较低,为基于大量训练数据的网络模型带来困难。
综上,现有技术虽然分别在轴承异常检测和轴承故障诊断有一定的研究成果,但是没有网络模型能够一起实现轴承的异常检测和故障诊断,并且异常检测和故障诊断精度都没有特别高。因此,本技术领域亟需一种既可实现对轴承异常检测,也可实现对轴承故障的诊断分类模型,并且利用生成对抗网络强大的数据生成能力进行生成数据,使用生成的数据对原始数据集进行数据扩充,提高异常检测和故障诊断准确率。
发明内容
针对现有技术的问题,本发明公开一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,构建DC-GANomaly异常检测和故障诊断网络模型,对轴承数据实现精准高效的异常检测和故障诊断。
本发明详细的技术内容如下:
一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,其特征在于,包括:
(1)构建基于深度卷积生成对抗网络的模型
构建一种由CNN和生成对抗网络组成的深度卷积网络模型DC-GANomaly,结合了GANomaly和CGAN的特点,在判别器添加了由四层全连接层构成的辅助分类器用于故障类别分类,此设计使用训练好的模型生成逼真数据进行数据扩充并训练分类器,优化损失函数实现轴承故障诊断,和上文连在一起是有监督故障分类模型;
(2)对深度卷积网络模型DC-GANomaly的训练
在异常检测训练阶段,输入轴承的正常一维振动信号及人为标注正常标签进行模型训练,直到整个网络模型达到纳什平衡状态,得到异常检测阶段训练好的深度卷积网络模型DC-GANomaly;
在故障诊断训练阶段,输入轴承的故障一维振动信号及人为标注故障类别标签进行模型训练,直到整个模型达到纳什平衡状态,使用深度卷积网络模型DC-GANomaly生成故障数据并加入初始训练集中,使用扩充后的训练集进行训练分类器,直到损失函数收敛即停止,得到故障诊断阶段训练好的分类器;
(3)对轴承异常检测和故障诊断
将实时采集到的轴承一维振动信号输入至得到异常检测阶段训练好的深度卷积网络模型DC-GANomaly,得到是否发生异常的检测结果;
再将异常的检测结果输入故障诊断阶段训练好的分类器,输出对应的故障类别标签类别。
根据本发明优选的,所述深度卷积网络模型DC-GANomaly包括:生成器、生成样本、编码器和判别器;
所述生成器为蝴蝶结卷积自编码神经网络架构;所述判别器后添加由四层全连接层构成的辅助分类器用于轴承故障诊断,本设计为在原GANomaly网络框架的基础上,使用蝴蝶结卷积自编码器和DCGAN的卷积神经网络架构作为网络主体结构,并于判别器后添加由四层全连接层构成的辅助分类器用于故障诊断,深度卷积网络模型DC-GANomaly如图1所示;深度卷积网络模型DC-GANomaly既能实现对轴承数据的异常检测,也能进行轴承故障类别分类,同时实现GAN生成数据对训练集的数据增强,补充不平衡的样本集,提高轴承故障诊断精度;深度卷积网络模型DC-GANomaly的生成器有两个输入,一个是真实样本,一个是真实样本对应的标注故障类别标签,得到生成样本,通过编码器将输入数据与正常一维振动信号数据进行特征比较,与设置的阈值比较后检测是否发生异常,判别器产生轴承故障分类诊断结果作为输出,根据场景设置所述阈值,判断大于所述阈值时发生异常。
根据本发明优选的,异常检测阶段训练好的深度卷积网络模型DC-GANomaly具体包括:
将全部正常数据的原始一维振动信号经过小波变换转为1*32*32时频图样本/>将嵌入人为标注正常标签的时频图输入到生成器,映射到低维重构图,由生成器生成的新样本/>生成相应的人为标注的正常异常标签/>
新样本被标记为0,并且时频图样本/>被标记为类别标签1;
然后将样本组合装载,鉴定和分类真实性,即将时频图样本和新样本一起输入判别器,由判别器判断样本是真实的原始样本还是由生成器生成的新样本;
此后,时频图样本标签记为lreal,新样本/>标签记为lfake;对时频图样本的预测记为preal,对新样本的预测记为pfake;
通过最小化训练模型的损失函数,完成对判别器的训练;所述判别器的交叉熵损失函数的计算公式如下:
Ld-real=fbce(preal,lreal) (I)
La-fake=fbce(pfake,lfake) (II)
Ld=(Ld-real+Ld-fake)*0.5 (III)
在公式(I)、公式(II)和公式(III)中,Ld-real为时频图样本输入判别器后得到的对时频图样本的预测标签和真实标签之间的损失函数;fbce指二分类交叉熵损失函数(BinaryCrossEntropy Loss);Ld-fake为新样本输入判别器后得到的对生成样本的预测标签和生成样本的生成标签之间的损失函数;Ld判别器的交叉熵损失函数;
新样本的标签被设置为1并发送到判别器,判别器通过标签为非0的新样本;通过最小化Ld来调整和完成生成器的训练,即通过反向更新梯度参数实现最小化;原始一维振动信号经判别器提取的特征记为zd-real,新样本经判别器提取的特征记为zd-fake;原始一维振动信号经生成器产生的瓶颈特征记为zg-raal,新样本经生成器提取的编码特征记为zg-fake;所述生成器的交叉熵损失函数的计算公式如下:
Ladv=fadv(zd-real,zd-fake) (IV)
Lcon=fcon(x,xfake) (V)
Lenc=fenc(zg-real,zg-fake) (VI)
Lg=wadvLadv+wconLcon+wencLenc (VII)
在公式(IV)、(V)、(VI)和(VII)中,Ladv、fadv表示特征匹配损失,计算原始一维振动信号经判别器提取的特征和新样本经判别器提取的特征的特征表示之间的L2距离,降低GAN训练的不稳定性,基于判别器的内部来更新生成器,即通过反向更新梯度参数和激活函数降低GAN训练的不稳定性;Lcon、fcon表示上下文损失,计算原始一维振动信号和新样本之间的L1距离;Lenc、fenc表示编码器损失,最小化输入的瓶颈特征和新样本经生成器提取的编码特征之间的距离;
所述判别器用于:当时频图样本被输入到深度卷积网络模型DC-GANomaly中时,使输出值对时频图样本的预测preal尽可能接近1,其中基于交叉熵损失函数更新网络梯度参数,使得损失函数越来越收敛,便能使得判别器对样本的判断越来越接近1;当输入生成的新样本/>时,判别器试图使对时频图样本的预测preal尽可能接近0,其中基于交叉熵损失函数更新网络梯度参数使得损失函数越来越收敛,就可以试图更准确的判断,即判断新样本数据为假;同时,使对时频图样本的预测preal尽可能接近1,其中,通过设置一个阈值将特征距离打分,大于所述阈值的输出1,小于等于所述阈值的输出0,以使生成器成功通过判别器;
零和博弈发生在判别器和生成器之间,其中最优化目标被转换为最小最大化问题;判别器和生成器在训练期间顺序旋转以被优化:首先固定生成器,然后优化判别器,使其最大化诊断准确率;接下来,与前一步骤相反,固定判别器,然后优化生成器;重复上述过程,直到判别器和生成器达到纳什平衡,完成DC-GANomaly的训练。
根据本发明优选的,在所述步骤(3)对轴承故障诊断的具体方法包括:
全部轴承故障原始一维振动信号经过小波变换转为1*32*32时频图,将嵌入人为标注故障类别标签的时频图样本输入到生成器,映射到低维重构图,由新样本/>生成相应的类别标签/>
新样本被标记为0,并且时频图样本/>被标记为类别标签1,然后将样本组合装载,鉴定和分类真实性,即将时频图样本和新样本一起输入判别器,由判别器判断样本是真实的原始样本还是由生成器生成的新样本;
通过最小化训练模型的损失函数,完成对判别器的训练,所述判别器的交叉熵损失函数的计算公式如下:
Ld=fbce(preal,lreal) (X)
在公式(IX)中,fbce指二分类交叉熵损失函数(Binary CrossEntropy Loss);Ld是指判别器的损失函数;
的标签被设置为1并发送到判别器,判别器不能通过真实性标签为0的新生成样本,通过最小化Ld来调整和完成生成器的训练,基于交叉熵损失函数更新网络梯度参数,使得损失函数越来越收敛,就可以最小化Ld,深度卷积网络模型DC-GANomaly中的生成器的交叉熵损失函数的计算公式如下:
Lg=fadv(zd-real,zd-fake) (XI)
在公式(XI)中,Lg、fadv表示特征匹配损失,计算原始一维振动信号和新样本经判别器的特征表示之间的L2距离;原始一维振动信号经判别器提取的特征记为zd-real,新样本经判别器提取的特征记为zd-fake;
所述判别器用于:当时频图样本被输入到深度卷积网络模型DC-GANomaly中时,使输出值对时频图样本的预测preal尽可能接近1,其中基于交叉熵损失函数更新网络梯度参数,使得损失函数越来越收敛,便能使得判别器对样本的判断越来越接近1;当输入生成的新样本/>时,判别器试图使对时频图样本的预测preal尽可能接近0,其中基于交叉熵损失函数更新网络梯度参数使得损失函数越来越收敛,就可以试图更准确的判断,即判断新样本数据为假;同时,使对时频图样本的预测preal尽可能接近1,其中,通过设置一个阈值将特征距离打分,大于所述阈值的输出1,小于等于所述阈值的输出0,以使生成器成功通过判别器;
零和博弈发生在判别器和生成器之间,其中最优化目标被转换为最小最大化问题;判别器和生成器在训练期间顺序旋转以被优化:首先固定生成器,然后优化判别器,使其最大化诊断准确率;接下来,与前一步骤相反,固定判别器,然后优化生成器;重复上述过程,直到判别器和生成器达到纳什平衡,完成DC-GANomaly的训练。
根据本发明优选的,在步骤(2)中,所述扩充后的训练集对应的扩充方法包括:
深度卷积网络模型DC-GANomaly训练完成后,所述生成器根据给定的标签生成不同故障类型的轴承振动信号数据,以对原始不平衡样本集进行扩充,进而实现数据增强的效果,扩充后的数据集记为扩充后的数据标签记为lehance,分类器产生的分类标签记为lc。
根据本发明优选的,在步骤(2)中,使用扩充后的训练集进行训练分类器的方法,包括:
使用扩充平衡样本集训练分类器,如图5所示;
分类器的交叉墒损失函数的计算公式如下:
Lc=fcrossentropylosslenhance,lc) (XII)
在公式(XII)中,Lc、fcrossentropyloss指多分类交叉熵损失函数。
离线数据增强完成后,能够解决数据不平衡的问题,从而提高故障诊断精度;故障诊断测试阶段,直接使用离线数据增强阶段中训练好的分类器便可对轴承故障数据产生精准、快速的诊断结果作为输出,故障诊断如图3、图4。
本发明的有益技术效果:
本发明所述的一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,既可以实现对轴承类部件的异常检测,也可以实现对轴承的故障类别分类。
当利用本发明进行异常检测时,使用采集到的正常轴承振动信号转化的时频图进行训练本发明中的模型,模型便学会了正常轴承振动信号数据的特征分布。在测试阶段时输入正常轴承振动数据和异常轴承振动数据,模型提取学习数据的特征分布,通过蝴蝶结自编码器和自解码器后,模型能够减少数据中的噪声干扰,更好的学习数据振动分布,将测试的数据特征分布与训练时期学习到的正常数据的特征分布进行比较,超过设置的阈值便认为发生异常。
在异常检测后,将进入故障诊断阶段:将轴承故障数据集输入模型进行训练,所述生成器判别器交替训练完成后达到纳什平衡状态,使用训练好的模型生成故障数据,生成的数据与原始数据具有较强的相似性,将生成的数据加入到训练集中进行数据扩充。使用扩充的数据集重新训练判别器。测试阶段时,将故障数据输入判别器,辅助分类器的最后一层linear层对故障数据做出分类作为结果输出。
本发明所述的网络模型在结构上做出相应改进,以及在功能上实现了对轴承部件的异常检测和故障诊断功能融合,还可以使用模型生成数据进行数据扩充,并且在异常检测以及故障诊断达到了较高的准确率。
附图说明
图1是本发明深度卷积网络模型DC-GANomaly的框架图;
图2是图1具体网络结构示意图;
在图2中,在判别器同时具备判别与分类功能;深度卷积网络模型DC-GANomaly网络大体结构及超参数选择是在GANomaly原始文献的基础上进行实验的结果;深度卷积网络模型DC-GANomaly的生成器判别器优化选用Adam算法;
在图2中,Net G指生成器;net D指判别器;encoder和decoder是一对蝴蝶结自编码自解码器;encoder1是异常检测辅助编码器,encoder2是判别器中的编码器:“异常检测”是指基于深度卷积网络模型DC-GANomaly实现轴承异常检测;“特征值”是指通过encoder2图片潜在的特征表示;“对数据的预测标签”是指在异常检测阶段,判别器对数据做出的一个0-1的一个判别数值,用来判断数据是真实数据还是生成数据进而完成模型的训练;“故障诊断”是指训练分类器时产生故障诊断结果以及分类器训练完成后对故障数据进行测试产生的故障诊断分类结果。
图3是本发明中对轴承异常检测的示意图;
在图3中,Initial data:原始一维振动信号;class:正常类别标签;Generator:生成器;Discriminator:判别器;Convolution:卷积神经网络;Relu:激活函数;Batchnorm:归一化处理;Auxilliary encoder:异常检测辅助编码器;Sigmoid:二分类激活函数;Parameter sharing:参数共享;Anomaly detection:异常检测;Score:分数;Abnormal/normal:异常/正常;
图4是本发明针对轴承故障诊断中扩充数据的流程图;
在图4中,Initial data:原始一维振动信号;class:类别故障标签;Generator:生成器;Discriminator:判别器;Convolution:卷积神经网络;Relu:激活函数;Batchnorm:归一化处理;Sigmoid:二分类激活函数;Data enrichment:数据增强;
图5是本发明针对轴承故障诊断的流程图;
在图5中,Fault Diagnosis:故障诊断;Fault classifier:故障分类;Convolution:卷积神经网络;Relu:激活函数;Flatten:全连接层;Linear:全连接层;Diagnosis Results:诊断结果;
图6是利用本发明所述方法对轴承异常检测第一阶段的示意图,即用训练集训练模型,交替训练生成器和判别器,训练完成后,用模型的生成器生成数据并加入训练集;
图7是利用本发明所述方法对轴承异常检测的示意图;
图8是利用本发明所述方法对轴承异常检测第二阶段的示意图,即用训练集(原始训练数据和生成数据)训练判别器,并用测试集进行验证;
图9是利用本发明所述方法对轴承故障诊断过程框架示意图;
轴承故障诊断过程分为两个阶段,如图4、图5所示:在迭代初始阶段,DC-GANomaly生成器损失较大,但之后快速收敛,与判别器损失形成纳什平衡;在迭代次数接近50次时,诊断精度达到99.375%;
图10本发明轴承异常检测过程框架示意图;
图11是增加故障数据类别的故障诊断准确率的示意图;
图12是本发明所述方法与其他模型准确率相比较的结果图。
具体实施方式
下面结合实施例和说明书附图对本发明做详细的说明,但不限于此。
实施例1、
一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,包括:
(1)构建基于深度卷积生成对抗网络的模型
构建一种由CNN和生成对抗网络组成的深度卷积网络模型DC-GANomaly,结合了GANomaly和CGAN的特点,在判别器添加了由四层全连接层构成的辅助分类器用于故障类别分类,此设计使用训练好的模型生成逼真数据进行数据扩充并训练分类器,优化损失函数实现轴承故障诊断,和上文连在一起是有监督故障分类模型;
(2)对深度卷积网络模型DC-GANomaly的训练
在异常检测训练阶段,输入轴承的正常一维振动信号及人为标注正常标签进行模型训练,直到整个网络模型达到纳什平衡状态,得到异常检测阶段训练好的深度卷积网络模型DC-GANomaly;
在故障诊断训练阶段,输入轴承的故障一维振动信号及人为标注故障类别标签进行模型训练,直到整个模型达到纳什平衡状态,使用深度卷积网络模型DC-GANomaly生成故障数据并加入初始训练集中,使用扩充后的训练集进行训练分类器,直到损失函数收敛即停止,得到故障诊断阶段训练好的分类器;
(3)对轴承异常检测和故障诊断
将实时采集到的轴承一维振动信号输入至得到异常检测阶段训练好的深度卷积网络模型DC-GANomaly,得到是否发生异常的检测结果;
再将异常的检测结果输入故障诊断阶段训练好的分类器,输出对应的故障类别标签类别。
所述深度卷积网络模型DC-GANomaly包括:生成器、生成样本、编码器和判别器;
所述生成器为蝴蝶结卷积自编码神经网络架构;所述判别器后添加由四层全连接层构成的辅助分类器用于轴承故障诊断,本设计为在原GANomaly网络框架的基础上,使用蝴蝶结卷积自编码器和DCGAN的卷积神经网络架构作为网络主体结构,并于判别器后添加由四层全连接层构成的辅助分类器用于故障诊断,深度卷积网络模型DC-GANomaly如图1所示;深度卷积网络模型DC-GANomaly既能实现对轴承数据的异常检测,也能进行轴承故障类别分类,同时实现GAN生成数据对训练集的数据增强,补充不平衡的样本集,提高轴承故障诊断精度;深度卷积网络模型DC-GANomaly的生成器有两个输入,一个是真实样本,一个是真实样本对应的标注故障类别标签,得到生成样本,通过编码器将输入数据与正常一维振动信号数据进行特征比较,与设置的阈值比较后检测是否发生异常,判别器产生轴承故障分类诊断结果作为输出,根据场景设置所述阈值,判断大于所述阈值时发生异常。
其中所述生成器属于现有技术,包括一对蝴蝶结自编码自解码器和异常检测辅助编码器,蝴蝶结自编码器共有四层卷积层,中间两个卷积层后加有batchnorm层,前三层后又都使用relu作为激活函数;蝴蝶结自解码器基本与蝴蝶结自编码器对称,第四层则使用tanh作为激活函数;异常检测辅助编码器也由四层卷积层组成,中间两层加有batchnorm层和relu层;
所述判别器包括编码器和辅助分类器,该判别器是对现有技术做了相应的改进以对故障数据做出诊断分类,即在判别器里面加了由四层全连接层构成的辅助分类器;
所述编码器的结构与蝴蝶结自编码器的结构相同,后使用sigmoid作为激活函数进行分类;
所述辅助分类器包括一层Flatten层和四层linear层,并由最后一层linear层实现分类。
异常检测阶段训练好的深度卷积网络模型DC-GANomaly具体包括:
将全部正常数据的原始一维振动信号经过小波变换转为1*32*32时频图样本/>将嵌入人为标注正常标签的时频图输入到生成器,映射到低维重构图,由生成器生成的新样本/>生成相应的人为标注的正常异常标签/>
新样本被标记为0,并且时频图样本/>被标记为类别标签1;
然后将样本组合装载,鉴定和分类真实性,即将时频图样本和新样本一起输入判别器,由判别器判断样本是真实的原始样本还是由生成器生成的新样本;
此后,时频图样本标签记为lreal,新样本/>标签记为lfake;对时频图样本的预测记为preal,对新样本的预测记为pfake;
通过最小化训练模型的损失函数,完成对判别器的训练;所述判别器的交叉熵损失函数的计算公式如下:
Ld-real=fbce(preal,lreal) (I)
Ld-fake=fbce(pfake,lfake) (II)
Ld=Ld-real+Ld-fake)*0.5 (III)
在公式(I)、公式(II)和公式(III)中,Ld-real为时频图样本输入判别器后得到的对时频图样本的预测标签和真实标签之间的损失函数;fbce指二分类交叉熵损失函数(BinaryCrossEntropy Loss);Ld-fake为新样本输入判别器后得到的对生成样本的预测标签和生成样本的生成标签之间的损失函数;La判别器的交叉熵损失函数;
新样本的标签被设置为1并发送到判别器,判别器通过标签为非0的新样本;通过最小化Ld来调整和完成生成器的训练,即通过反向更新梯度参数实现最小化;原始一维振动信号经判别器提取的特征记为zd-read,新样本经判别器提取的特征记为zd-fake;原始一维振动信号经生成器产生的瓶颈特征记为zg-real,新样本经生成器提取的编码特征记为zg-fake;所述生成器的交叉熵损失函数的计算公式如下:
Ladv=fadv(zd-real,zd-fake) (IV)
Lcon=fcon(x,xfake) (V)
Lenc=fenc(zg-real,zg-fake) (VI)
Lg=wadvLadv+wconLcon+wencLenc (VII)
在公式(IV)、(V)、(VI)和(VII)中,Ladv、fadv表示特征匹配损失,计算原始一维振动信号经判别器提取的特征和新样本经判别器提取的特征的特征表示之间的L2距离,降低GAN训练的不稳定性,基于判别器的内部来更新生成器,即通过反向更新梯度参数和激活函数降低GAN训练的不稳定性;Lcon、fcon表示上下文损失,计算原始一维振动信号和新样本之间的L1距离;Lenc、fenc表示编码器损失,最小化输入的瓶颈特征和新样本经生成器提取的编码特征之间的距离;
所述判别器用于:当时频图样本被输入到深度卷积网络模型DC-GANomaly中时,使输出值对时频图样本的预测preal尽可能接近1,其中基于交叉熵损失函数更新网络梯度参数,使得损失函数越来越收敛,便能使得判别器对样本的判断越来越接近1;当输入生成的新样本/>时,判别器试图使对时频图样本的预测preal尽可能接近0,其中基于交叉熵损失函数更新网络梯度参数使得损失函数越来越收敛,就可以试图更准确的判断,即判断新样本数据为假;同时,使对时频图样本的预测preat尽可能接近1,其中,通过设置一个阈值将特征距离打分,大于所述阈值的输出1,小于等于所述阈值的输出0,以使生成器成功通过判别器:
零和博弈发生在判别器和生成器之间,其中最优化目标被转换为最小最大化问题;判别器和生成器在训练期间顺序旋转以被优化:首先固定生成器,然后优化判别器,使其最大化诊断准确率;接下来,与前一步骤相反,固定判别器,然后优化生成器;重复上述过程,直到判别器和生成器达到纳什平衡,完成DC-GANomaly的训练。
实施例2、
如实施例1所述一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,在所述步骤(3)对轴承故障诊断的具体方法包括:
全部轴承故障原始一维振动信号经过小波变换转为1*32*32时频图,将嵌入人为标注故障类别标签的时频图样本输入到生成器,映射到低维重构图,由新样本/>生成相应的类别标签/>
新样本被标记为0,并且时频图样本/>被标记为类别标签1,然后将样本组合装载,鉴定和分类真实性,即将时频图样本和新样本一起输入判别器,由判别器判断样本是真实的原始样本还是由生成器生成的新样本;
通过最小化训练模型的损失函数,完成对判别器的训练,所述判别器的交叉熵损失函数的计算公式如下:
Ld=fbce(preal,lreal) (X)
在公式(IX)中,fbce指二分类交叉熵损失函数(Binary CrossEntropy Loss);Ld是指判别器的损失函数:
的标签被设置为1并发送到判别器,判别器不能通过真实性标签为0的新生成样本,通过最小化Ld来调整和完成生成器的训练,基于交叉熵损失函数更新网络梯度参数,使得损失函数越来越收敛,就可以最小化Ld,深度卷积网络模型DC-GANomaly中的生成器的交叉熵损失函数的计算公式如下:
Lg=fadv(zd-real,zd-fake) (XI)
在公式(XI)中,Lg、fadv表示特征匹配损失,计算原始一维振动信号和新样本经判别器的特征表示之间的L2距离;原始一维振动信号经判别器提取的特征记为zd-real,新样本经判别器提取的特征记为zd-fake;
所述判别器用于:当时频图样本被输入到深度卷积网络模型DC-GANomaly中时,使输出值对时频图样本的预测preal尽可能接近1,其中基于交叉熵损失函数更新网络梯度参数,使得损失函数越来越收敛,便能使得判别器对样本的判断越来越接近1;当输入生成的新样本/>时,判别器试图使对时频图样本的预测preal尽可能接近0,其中基于交叉熵损失函数更新网络梯度参数使得损失函数越来越收敛,就可以试图更准确的判断,即判断新样本数据为假;同时,使对时频图样本的预测preal尽可能接近1,其中,通过设置一个阈值将特征距离打分,大于所述阈值的输出1,小于等于所述阈值的输出0,以使生成器成功通过判别器;
零和博弈发生在判别器和生成器之间,其中最优化目标被转换为最小最大化问题;判别器和生成器在训练期间顺序旋转以被优化:首先固定生成器,然后优化判别器,使其最大化诊断准确率;接下来,与前一步骤相反,固定判别器,然后优化生成器;重复上述过程,直到判别器和生成器达到纳什平衡,完成DC-GANomaly的训练。
实施例3、
如实施例1、2所述一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,在步骤(2)中,所述扩充后的训练集对应的扩充方法包括:
深度卷积网络模型DC-GANomaly训练完成后,所述生成器根据给定的标签生成不同故障类型的轴承振动信号数据,以对原始不平衡样本集进行扩充,进而实现数据增强的效果,扩充后的数据集记为扩充后的数据标签记为lenhance,分类器产生的分类标签记为lc。
在步骤(2)中,使用扩充后的训练集进行训练分类器的方法,包括:
使用扩充平衡样本集训练分类器,如图5所示;
分类器的交叉熵损失函数的计算公式如下:
Lc=fcrossentropyloss(lenhance,lc) (XII)
在公式(XII)中,Lc、fcrossentropyloss指多分类交叉熵损失函数。
离线数据增强完成后,能够解决数据不平衡的问题,从而提高故障诊断精度;故障诊断测试阶段,直接使用离线数据增强阶段中训练好的分类器便可对轴承故障数据产生精准、快速的诊断结果作为输出,故障诊断如图4、图5。
根据上述实施例所记载的技术内容,以下具体说明:
一、采用离线模型训练,在线异常检测的模式,进行轴承异常检测过程如图10所示。
异常检测阶段在迭代次数接近45次时,检测精度达到92.6%。
1、离线模型训练
基于损失函数,对模型参数进行迭代更新。模型训练采用Adam优化器,判别器与生成器的学习率均设置为0.0002。在每个训练单元中,依次分别交替训练判别器以及生成器。
训练过程可以分为两个步骤:
(1)正常训练样本中的原始一维振动信号经过小波变换转为时频图,将嵌入正常标签的时频图输入到生成器模型中,得到低维重构图;重构图与真实样本数据混合在一起传送至判别器进行训练,基于其损失函数计算出判别器网络的损失值,使用Adam优化器更新网络参数。
(2)训练判别器后,组合结构开始训练。在这一阶段,模型保持判别器网络权重不变,同样根据损失函数计算模型生成器的损失值,使用Adam优化器更新网络参数。在每次更新生成器网络参数之前,先执行判别器网络参数的更新,以提高模型训练速度。
在训练完这个组合结构后,也就是一个epoch已经结束,模型的训练过程会重新开始。经过足够次的迭代训练,判别器和生成器网络达到纳什平衡,模型训练完成。此时网络模型可检测输入振动信号是否发生异常。
2、在线异常检测
离线训练模型完成后,将测试集中的故障数据图像输入训练好的DC-GANomaly模型,编码器的异常重构导致异常检测辅助编码器得到异常特征向量,从而导致与原始输入数据瓶颈特征向量的不相似性,通过最小化特征差异得出是否发生异常的判断。
二、基于本发明轴承故障诊断实现过程,采用离线数据增强,在线故障诊断的模式,如图9所示。
故障诊断阶段在迭代次数接近50次时,诊断精度达到99.375%。
1、离线数据增强
基于损失函数,对模型参数进行迭代更新。模型训练采用Adam优化器,判别器与生成器的学习率均设置为0.0007。在每个训练单元中,依次分别交替训练判别器以及生成器。训练过程可以分为两个步骤:
(1)全部原始一维振动信号(仅故障数据)经过小波变换转为时频图,将嵌入类别标签的时频图输入到生成器模型,得到低维重构图;重构图与真实样本数据混合在一起传送至判别器进行训练,基于其损失函数计算出判别器网络的损失值,使用Adam优化器更新网络参数。
(2)训练判别器后,组合结构开始训练。在这一阶段,模型保持判别器网络权重不变,同样根据损失函数计算模型生成器的损失值,使用Adam优化器更新网络参数。在每次更新生成器网络参数之前,先执行判别器网络参数的更新,以提高模型训练速度。
在训练完这个组合结构后,也就是一个epoch已经结束,模型的训练过程会重新开始。经过足够次的迭代训练,判别器和生成器网络达到纳什平衡,模型训练完成。此时生成器可根据给定的标签生成不同故障类型的轴承振动信号数据,利用生成数据对原始不平衡样本集进行扩充,进而实现数据增强的效果。随后,使用扩充平衡样本集训练分类器。
2、在线故障诊断
离线数据增强完成后,能够解决数据不平衡的问题,从而提高故障诊断精度。在线故障诊断阶段,直接使用离线数据增强阶段中训练好的分类器便可对轴承故障数据做出精准、快速的诊断。
3、算例分析
仿真硬件为11th Gen Intel(R)Core(TM)i5-1135G7@2.40GHz 2.42GHz,8GB运行内存、内存为462GB的LAPTOP-QP4FBG28,改进网络模型基于深度学习框架pytorch和keras构建。
3.1实验样本构建
(1)基于DC-GANomaly的轴承异常检测实验样本构建
为方便模型训练,将所选的试验样本划分为两部分:训练集和测试集。使用训练集对改进DC-GANomaly进行训练,使用测试集对模型进行验证。选取正常试验样本的前三类作为训练集,第四类正常试验样本和三类异常样本作为测试集。epoch设定为100,batch size设定为64,wadv、wcon、wenc分别设置为1、60、1。
(2)基于DC-GANomaly的轴承故障诊断实验样本构建
故障诊断实验样本构建如下:将三类故障数据集随机分开,80%和20%的样本分别作为训练集和测试集的一部分。epoch设定为100,batch size设定为64。使用训练集对改进DC-GANomaly进行训练,训练完成后,对轴承故障样本进行生成,得到DC-GANomaly生成的轴承故障样本。将生成的轴承故障样本加入训练集,扩充轴承故障不平衡样本集。使用测试集对模型进行验证。
3.2模型训练
(1)基于DC-GANomaly的轴承异常检测过程
基于DC-GANomaly的轴承异常检测过程如图3所示。在迭代初始阶段,DC-GANomaly生成器损失较大,但之后快速收敛,与判别器损失形成纳什平衡。在迭代次数接近45次时,检测精度达到92.6%。
综上所述,与原GANomaly相比,本发明深度卷积网络模型DC-GANomaly实现了对轴承振动信号的异常检测和故障分类,并在检测精度和诊断精度均表现更好。
另外,结合图11、图12可知,下面两图分别是增加故障数据类别的故障诊断准确率以及与其他模型准确率相比较的结果图,由图11实验结果可知,随着故障数据类别以及训练次数的增加,模型的故障诊断准确率均有所提升,并接近100%,表明模型对轴承故障诊断的有效性;由图12可知,本文提出的模型相较于传统机器学习模型具有十分明显的优越性。
随着科学技术飞速发展,工业水平不断提高,越来越多机械设备投入使用。几乎所有的机械设备都会用到轴承部件,且轴承发挥着至关重要的作用。多数轴承运行环境恶劣,运行工况复杂多变,轴承发生故障的次数也就增多。因此对轴承故障做出精准的检测及分类可以提高机械设备的经济效益。但是轴承故障数据少,存在严重的数据不平衡问题,给基于大量训练数据的深度学习网络模型带来很大困难。
本发明结合实施例和附图从模型融合及数据增强的观点出发,利用生成对抗网络(GAN)在目前表现出的强大的数据生成能力,基于深度卷积网络模型DC-GANomaly实现对轴承的异常检测以及数据增强后故障诊断,进一步提高检测和诊断精度。在工程实践中提高异常检测和故障诊断效率,节省人力物力资源,提升异常检测和故障诊断准确率,提升机械设备经济效益。
Claims (6)
1.一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,其特征在于,包括:
(1)构建基于深度卷积生成对抗网络的模型
构建一种由CNN和生成对抗网络组成的深度卷积网络模型DC-GANomaly,在判别器添加了由四层全连接层构成的辅助分类器用于故障类别分类;
(2)对深度卷积网络模型DC-GANomaly的训练
在异常检测训练阶段,输入轴承的正常一维振动信号及人为标注正常标签进行模型训练,直到整个网络模型达到纳什平衡状态,得到异常检测阶段训练好的深度卷积网络模型DC-GANomaly;
在故障诊断训练阶段,输入轴承的故障一维振动信号及人为标注故障类别标签进行模型训练,直到整个模型达到纳什平衡状态,使用深度卷积网络模型DC-GANomaly生成故障数据并加入初始训练集中,使用扩充后的训练集进行训练分类器,直到损失函数收敛即停止,得到故障诊断阶段训练好的分类器;
(3)对轴承异常检测和故障诊断
将实时采集到的轴承一维振动信号输入至得到异常检测阶段训练好的深度卷积网络模型DC-GANomaly,得到是否发生异常的检测结果;
再将异常的检测结果输入故障诊断阶段训练好的分类器,输出对应的故障类别标签类别。
2.根据权利要求1所述一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,其特征在于,所述深度卷积网络模型DC-GANomaly包括:生成器、生成样本、编码器和判别器;
所述生成器为蝴蝶结卷积自编码神经网络架构;所述判别器后添加由四层全连接层构成的辅助分类器用于轴承故障诊断;深度卷积网络模型DC-GANomaly既能实现对轴承数据的异常检测,也能进行轴承故障类别分类,同时实现GAN生成数据对训练集的数据增强,补充不平衡的样本集,提高轴承故障诊断精度;深度卷积网络模型DC-GANomaly的生成器有两个输入,一个是真实样本,一个是真实样本对应的标注故障类别标签,得到生成样本,通过编码器将输入数据与正常一维振动信号数据进行特征比较,与设置的阈值比较后检测是否发生异常,判别器产生轴承故障分类诊断结果作为输出,根据场景设置所述阈值,判断大于所述阈值时发生异常。
3.根据权利要求1所述一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,其特征在于,异常检测阶段训练好的深度卷积网络模型DC-GANomaly具体包括:
将全部正常数据的原始一维振动信号经过小波变换转为时频图样本/>将嵌入人为标注正常标签的时频图输入到生成器,映射到低维重构图,由生成器生成的新样本生成相应的人为标注的正常异常标签/>
新样本被标记为0,并且时频图样本/>被标记为类别标签1:
然后将样本组合装载,鉴定和分类真实性;
此后,时频图样本标签记为lreal,新样本/>标签记为lfake;对时频图样本的预测记为preal,对新样本的预测记为pfake;
通过最小化训练模型的损失函数,完成对判别器的训练;所述判别器的交叉熵损失函数的计算公式如下:
Ld-real=fbce(preal,lreal) (I)
Ld-fake=fbce(pfake,lfake) (II)
Ld=(Ld-real+Ld-fake)*0.5 (III)
在公式(I)、公式(II)和公式(III)中,Ld-real为时频图样本输入判别器后得到的对时频图样本的预测标签和真实标签之间的损失函数;fbce指二分类交叉熵损失函数;Ld-fake为新样本输入判别器后得到的对生成样本的预测标签和生成样本的生成标签之间的损失函数;Ld判别器的交叉熵损失函数;
新样本的标签被设置为1并发送到判别器,判别器通过标签为非0的新样本;通过最小化Ld来调整和完成生成器的训练;原始一维振动信号经判别器提取的特征记为zd-eal,新样本经判别器提取的特征记为zd-fake;原始一维振动信号经生成器产生的瓶颈特征记为zg-real,新样本经生成器提取的编码特征记为zg-fake;所述生成器的交叉熵损失函数的计算公式如下:
Ladv=fadv(zd-real,zd-fake) (IV)
Lcon=fcon(x,xfake) (V)
Lenc=fenc(zg-real,zg-fake) (VI)
Lg=wadvLadv+wconLcon+wencLenc (VII)
在公式(IV)、(V)、(VI)和(VII)中,Ladv、fadv表示特征匹配损失,计算原始一维振动信号经判别器提取的特征和新样本经判别器提取的特征的特征表示之间的L2距离,降低GAN训练的不稳定性,基于判别器的内部来更新生成器;Lcon、fcon表示上下文损失,计算原始一维振动信号和新样本之间的L1距离;Lenc、fenc表示编码器损失,最小化输入的瓶颈特征和新样本经生成器提取的编码特征之间的距离;
所述判别器用于:当时频图样本被输入到深度卷积网络模型DC-GANomaly中时,使输出值对时频图样本的预测preal接近1;当输入生成的新样本/>时,判别器试图使对时频图样本的预测preal接近0,即判断新样本数据为假;同时,使对时频图样本的预测preal接近1,以使生成器成功通过判别器。
4.根据权利要求1所述一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,其特征在于,在所述步骤(3)对轴承故障诊断的具体方法包括:
全部轴承故障原始一维振动信号经过小波变换转为1*32*32时频图,将嵌入人为标注故障类别标签的时频图样本输入到生成器,映射到低维重构图,由新样本/>生成相应的类别标签/>
新样本被标记为0,并且时频图样本/>被标记为类别标签1,然后将样本组合装载,鉴定和分类真实性;
通过最小化训练模型的损失函数,完成对判别器的训练,所述判别器的交叉熵损失函数的计算公式如下:
Ld=fbce(preal,lreal) (X)
在公式(IX)中,fbce指二分类交叉熵损失函数;Ld是指判别器的损失函数;
的标签被设置为1并发送到判别器,判别器不能通过真实性标签为0的新生成样本,通过最小化Ld来调整和完成生成器的训练,深度卷积网络模型DC-GANomaly中的生成器的交叉熵损失函数的计算公式如下:
Lg=fadv(zd-real,zd-fake) (XI)
在公式(XI)中,Lg、fadv表示特征匹配损失,计算原始一维振动信号和新样本经判别器的特征表示之间的L2距离;原始一维振动信号经判别器提取的特征记为zd-real,新样本经判别器提取的特征记为zd-fake;
所述判别器用于:当时频图样本被输入到深度卷积网络模型DC-GANomaly中时,使输出值对时频图样本的预测preal接近1:当输入生成的新样本/>时,判别器试图使对时频图样本的预测preal接近0,即判断新样本数据为假;同时,使对时频图样本的预测preal接近1,以使生成器成功通过判别器。
5.根据权利要求1所述一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,其特征在于,在步骤(2)中,所述扩充后的训练集对应的扩充方法包括:
深度卷积网络模型DC-GANomaly训练完成后,所述生成器根据给定的标签生成不同故障类型的轴承振动信号数据,扩充后的数据集记为扩充后的数据标签记为lenhance,分类器产生的分类标签记为lc。
6.根据权利要求5所述一种基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法,其特征在于,在步骤(2)中,使用扩充后的训练集进行训练分类器的方法,包括:
分类器的交叉熵损失函数的计算公式如下:
Lc=fcrossentropyloss(lenhance,lc) (XII)
在公式(XII)中,Lc、fcrossentropyloss指多分类交叉熵损失函数。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202310489061.8A CN116754230A (zh) | 2023-04-25 | 2023-04-25 | 基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202310489061.8A CN116754230A (zh) | 2023-04-25 | 2023-04-25 | 基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN116754230A true CN116754230A (zh) | 2023-09-15 |
Family
ID=87952194
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202310489061.8A Pending CN116754230A (zh) | 2023-04-25 | 2023-04-25 | 基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN116754230A (zh) |
Cited By (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN117407784A (zh) * | 2023-12-13 | 2024-01-16 | 北京理工大学 | 面向传感器数据异常的旋转机械智能故障诊断方法及系统 |
CN117409546A (zh) * | 2023-11-15 | 2024-01-16 | 淮阴工学院 | 一种化工厂环境及有害气体智能检测与预警系统 |
-
2023
- 2023-04-25 CN CN202310489061.8A patent/CN116754230A/zh active Pending
Cited By (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN117409546A (zh) * | 2023-11-15 | 2024-01-16 | 淮阴工学院 | 一种化工厂环境及有害气体智能检测与预警系统 |
CN117407784A (zh) * | 2023-12-13 | 2024-01-16 | 北京理工大学 | 面向传感器数据异常的旋转机械智能故障诊断方法及系统 |
CN117407784B (zh) * | 2023-12-13 | 2024-03-12 | 北京理工大学 | 面向传感器数据异常的旋转机械智能故障诊断方法及系统 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
Zhang et al. | The multiclass fault diagnosis of wind turbine bearing based on multisource signal fusion and deep learning generative model | |
Wang et al. | Multilevel information fusion for induction motor fault diagnosis | |
Zhao et al. | Deep multi-scale convolutional transfer learning network: A novel method for intelligent fault diagnosis of rolling bearings under variable working conditions and domains | |
CN109580215B (zh) | 一种基于深度生成对抗网络的风电传动系统故障诊断方法 | |
Jiang et al. | A GAN-based anomaly detection approach for imbalanced industrial time series | |
Zhao et al. | Improved generative adversarial network for vibration-based fault diagnosis with imbalanced data | |
Zheng et al. | Data synthesis using dual discriminator conditional generative adversarial networks for imbalanced fault diagnosis of rolling bearings | |
CN116754230A (zh) | 基于深度卷积生成对抗网络的轴承异常检测和故障诊断方法 | |
Li et al. | Diagnosis of inter-turn short circuit of permanent magnet synchronous motor based on deep learning and small fault samples | |
Peng et al. | Fault feature extractor based on bootstrap your own latent and data augmentation algorithm for unlabeled vibration signals | |
CN112763214A (zh) | 基于多标签零样本学习的滚动轴承故障诊断方法 | |
CN116451150A (zh) | 一种基于半监督小样本的设备故障诊断方法 | |
Li et al. | A novel method of bearing fault diagnosis in time-frequency graphs using InceptionResnet and deformable convolution networks | |
Qin et al. | Research on intelligent fault diagnosis of mechanical equipment based on sparse deep neural networks | |
CN115962946A (zh) | 基于改进WGAN-GP和Alxnet的轴承故障诊断方法 | |
Han et al. | GAN-SAE based fault diagnosis method for electrically driven feed pumps | |
CN114491823A (zh) | 一种基于改进生成对抗网络的列车轴承故障诊断方法 | |
Xu et al. | Intelligent fault diagnosis of bearings under small samples: A mechanism-data fusion approach | |
Wang et al. | Multi-scale style generative and adversarial contrastive networks for single domain generalization fault diagnosis | |
Zhao et al. | Hybrid semi-supervised learning for rotating machinery fault diagnosis based on grouped pseudo labeling and consistency regularization | |
Liu et al. | Multi-scale residual anti-noise network via interpretable dynamic recalibration mechanism for rolling bearing fault diagnosis with few samples | |
Miao et al. | Sparse-representation-network-based feature learning of vibration signal for machinery fault diagnosis | |
Chen et al. | Relational conduction graph network for intelligent fault diagnosis of rotating machines under small fault samples | |
Hu et al. | An improved metalearning framework to optimize bearing fault diagnosis under data imbalance | |
Zhang et al. | Mechanical fault diagnosis with noisy multisource signals via unified pinball loss intuitionistic fuzzy support tensor machine |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |