第38卷第5期2021年5月
机㊀㊀电㊀㊀工㊀㊀程
JournalofMechanical&ElectricalEngineering
Vol.38No.5May2021
收稿日期:2020-09-09
基金项目:辽宁省自然科学基金资助项目(2019
BS186)
作者简介:刘红军(1971-)ꎬ男ꎬ辽宁沈阳人ꎬ副教授ꎬ硕士生导师ꎬ主要从事数字化制造技术方面的研究ꎮE ̄mail:133****8635@163.com
DOI:10.3969/j.issn.1001-4551.2021.05.010
基于GADF与卷积神经网络的滚动
轴承故障诊断研究
刘红军ꎬ魏旭阳
(沈阳航空航天大学机电工程学院ꎬ辽宁沈阳110000)
摘要:为充分发挥深度学习识别二维图像在滚动轴承故障检测方面的优势ꎬ提出了一种格拉姆角差场(GADF)结合改进卷积神经网络(CNN)的智能故障诊断模型ꎮ首先ꎬ将一维时序振动信号通过格拉姆角差场转化为二维图像ꎬ提取了图像特征ꎬ并输入了改进后的CNN模型ꎻ其次ꎬ改进的CNN模型采用全局池化层替代了传统的全连接层ꎬ有效地解决了传统CNN模型参数爆炸的问题ꎻ最后ꎬ进行了试验研究ꎬ通过Adam小批量优化法进行了迭代训练ꎬ达到了理想的检测精度ꎮ试验与研究结果表明:该诊断方法在特征提取方面更快速㊁准确ꎬ充分展现了CNN模型的非线性表达能力ꎬ检测精度优于其他智能诊断算法ꎮ关键词:格拉姆角差场ꎻ故障诊断ꎻ卷积神经网络ꎻ深度学习中图分类号:TH133.33㊀㊀㊀㊀文献标识码:A
文章编号:1001-4551(2021)05-0587-06
RollingbearingfaultdiagnosisbasedonGADFand
convolutionalneuralnetwork
LIUHong ̄junꎬWEIXu ̄yang
(CollegeofElectromechanicEngineeringꎬShenyangAerospaceUniversityꎬShenyang110000ꎬChina)
Abstract:Inordertogivefullplaytotheadvantagesofdeeplearningrecognitionintwo ̄dimensionalimageinrollingbearingfaultdetectionꎬanintelligentfaultdiagnosismodelbasedonGramangledifferencefield(GADF)andimprovedconvolutionalneuralnetwork(CNN)wasproposed.Firstlyꎬtheone ̄dimensionaltimeseriesvibrationsignalwastransformedintoatwo ̄dimensionalimagethroughthegramangle
differencefieldꎬandthetargetimagefeatureswereextractedandinputintotheimprovedCNNmodel.Secondlyꎬtheglobalpoolinglayerin ̄steadofthetraditionalfullconnectionlayerwasusedbytheimprovedCN
Nmodelꎬwhicheffectivelysolvedtheproblemofparameterexplo ̄sionofthetraditionalCNNmodel.FinallytheidealdetectionthroughiterativetrainingwithAdamsmallbatchoptimizationmethodaccuracy
wasachieved.Thetestandcomparisonresultsshowthatthediagnosismethodismorerapidandaccurateinfeatureextractionꎬfullydemon ̄stratesthenonlinearexpressionabilityofCNNmodelꎬandthedetectionaccuracyisbetterthantheotherintelligentdiagnosisalgorithms.Keywords:Gramangledifferenceꎻfieldfaultdiagnosisꎻconvolutionneuralnetworkꎻdeeplearning
0㊀引㊀言
在现今机械设备中ꎬ旋转机械是机械设备中最常见和最重要的组成部件之一ꎮ滚动轴承作为旋转设备最基本的结构之一ꎬ对机械设备的寿命与稳定性有着决定性的影响ꎮ
因此ꎬ国内外许多学者对滚动轴承的故障检测进
行了深入的研究[1]ꎬ并提出了许多有价值的方法ꎮ大部分检测方法多采用对滚动轴承的故障信号进行分析ꎬ提取有效特征作为输入ꎬ导入各种分类器中进行识别ꎮ如杨宇等[2]通过EMD和神经网络结合的故障诊断方法ꎻ刘韬等[3]通过KPCA和耦合隐马尔科夫模型
的轴承故障诊断ꎮ
但此类方法存在过程复杂㊁缺少灵活性㊁对专家的经验与知识具有依赖性ꎬ以及分类器缺少非线性拟合能力等问题ꎮ
近些年来ꎬ深度学习的非线性拟合能力在轴承故障检测方面得到了充分的展示ꎬ并获得了理论上的高精度诊断水平ꎮ其中ꎬ卷积神经网络(CNN)因其自身具有共享权值与偏重㊁局部感受野和子采样测量等特点ꎬ与深度信念网络(DBN)[4]和限制玻尔兹曼机(RBM)[5]等其他深度学习方法相比ꎬ在语音以及图像分类任务方面具有较大的优势ꎮ同时ꎬ其在故障诊断领域的应用也日益增多ꎮ
CNN在执行数据分类任务时ꎬ作为输入的数据集
可分为一维数据和二维数据两类ꎮ如孙曙光等[6]提出的一维CNN的低压万能式断路器附件故障诊断ꎻ肖雄等[7]的一种二维CNN优化轴承故障诊断方法ꎮWUY等[8]对两种类型数据进行了比较ꎬ认为在
已有的大多数CNN模型中ꎬ二维图像数据更适用于神经网络的模型的训练ꎮ而如何将一维振动信号清晰地转化为二维图像数据ꎬ成为智能故障检测的重要难点ꎮ
基于以上分析ꎬ笔者提出一种新的故障诊断方式ꎬ由格拉姆角差场(GADF)时间序列编码方式[9]
将一维
时序数据转换为二维图像数据ꎬ并结合改进的CNN模
型进行预测ꎬ以达到高精度诊断的效果ꎮ
1㊀格拉姆角差场图像编码转换
GADF图像编码的本质是通过格拉姆角场
(GAF)ꎬ将获取到的一维时序数据转化为二维图像数据的方法ꎻ将时间序列X缩放入区间[-1ꎬ1]ꎬ再将其极坐标化得到xꎮ
其过程如下:
xi-1=
xi-maxX()()+xi-minX()()
maxX()-minX()
(1)θ=arccosxi()ꎬ-1ɤxiɤ1ꎬxiɪXr=ti/NꎬtiɪN
(2)
经过上述变换过程ꎬ即可将一维时间序列数据转换为对称的带有颜㊁点㊁线的对称特征图像ꎮ当GAF基于正弦函数时生成格拉姆角差场(GADF)ꎬ如下式所示:
GADF=sin(θ1+θ1) sin(θ1+θn)⋮
⋮sin(θn+θ1)
sin(θn+θn)éëê
êê
ù
û
ú
úú(3)通过编码转化为二维图像ꎬGADF编码的示意图
如图1所示
图1㊀GADF编码示意图
在图1中ꎬ当第一个波峰出现时ꎬ在两种特征图中出现颜较浅的对称交叉线ꎻ而当峰值波动较大时ꎬ特征图中与其对应的对应交叉线其颜更为明显ꎻ
与此同时ꎬ其余较小的震动在GADF特征图中也完整地表现出来ꎮ
由此可见ꎬGADF可以将滚动轴承的震动信号完整地映射为二维特征图像ꎬ并作为CNN模型的输出ꎮ
2㊀卷积神经网络
卷积神经网络(CNN)是一种前馈式神经网络ꎬ其卷积层与池化层具有强大的数据特征提取能力ꎬ适用于处理图像与声音等网状结构数据ꎬ神经网络模型中滤波器通过对输入数据的迭代和分析来获得更多的有效特征和隐藏信息ꎻ同时结合稀疏式全连接和权重偏置的共享机制ꎬ在时间与空间上进行采样ꎬ降低了数据的复杂程度ꎬ相比其他神经网络结构类型减少了训练量ꎬ有效避免了算法过拟合ꎮ
CNN模型结构简图如图2所示ꎮ
图2㊀CNN模型结构简图
2.1㊀卷积层
卷积层是整个神经网络模型的核心所在ꎬ它通过对输入图像数据的卷积运算提取图像中的特征信息ꎮ
卷积层的具体运算如下:
xlj=f(ðiɪMJ
Xl-1I∗klij+blj)
(4)
式中:f 激活函数ꎻl 网络模型中的第l层ꎻxlj 第l层中的第j个特征图像ꎻklij 卷积核的权值ꎻblj 运算所需的偏置ꎻMJ 输入模型的二维图像数据集ꎮ
2.2㊀池化层
池化层的主要目的是降低卷积层处理后的数据维度ꎮ因为卷积后的数据虽然具有了之前数据的特征信
885 机㊀㊀电㊀㊀工㊀㊀程第38卷
息ꎬ但是其特征维度也急剧增长ꎮ为了汇总特征平面的输出ꎬ通常采用两种池化方式ꎬ即平均池化与最大池化方式ꎮ
其中ꎬ池化层的具体运算如下:xlj=f(βljdown(xl-1j+blj))
(5)
2.3㊀全局池化层
正则化可以产生稀疏权值
与传统的全连接层相比ꎬ全局池化层降低了运算的参数量ꎬ提高了模型的泛化能力ꎬ减少了过拟合情况的发生ꎮ与Dropout[10]相比ꎬ全局池化层提高了正则化效果ꎮ在现代卷积架构中ꎬ全局池化已经逐渐取代了
Dropout的位置ꎮ全局池化[11]的核心思想是ꎬ将每张图片的所有特征值融合为一个特征值ꎬ实现张量的降维转换ꎬ使特征值的维数等于最后一层卷积层的数目ꎮ
设最后层卷积层特征图大小为mnꎬ其中第n张特征图的值用x
l()ij
表示ꎬ则经过不同类型的全局池化操
作后ꎬ该特征值y
l()
可以表示为:
l()
=max[x
l()ij
](6)y
l()
m∗nðm
i=0ðn
j=0
xl()ij[](7)
其中ꎬ式(6)代表的是最大全局池化ꎬ式(7)为全局平均池化ꎮ
3㊀试验研究
3.1㊀数据采集及编码
为评估GADF ̄CNN算法对滚动轴承故障智能诊断的有效性和准确性ꎬ此次采用的试验数据集为美国凯西储大学(casewesternreserveuniversityꎬCWRU)公开的深沟球轴承故障数据集ꎮ
电机负载状态为1797r/min㊁1772r/min㊁1750r/min㊁1797r/min4种ꎬ故障分为外圈㊁内圈与流动体ꎻ采集频率为12kHzꎬ采用传感器置于电机的驱动
端的12个位置进行采集ꎬ采集时长为10sꎻ
因传感器采集信号时ꎬ转轴每圈所采集到的采样点约为400个ꎻ
为确保数据集的稳定性ꎬ每种故障分类样本长度设置为500个采样点ꎬ各200个样本ꎮ
本文选取1797r/min负载下ꎬ包括正常状态与11种故障状态的共12种情况作为数据集ꎬ分别由0~11数字表示ꎬ取前80%为训练集ꎬ输入已建立好的CNN模型ꎻ其余为测试集对训练好的神经网络模型进行检验ꎮ
二维故障图分类图如图3所示ꎮ3.2㊀卷积神经网络模型超参数优选
超参数的选择决定了整个卷积神经网络模型在故图3㊀二维故障图分类图
障诊断的精准度ꎬ通常CNN中主要的超参数有卷积核㊁池化核㊁优化器㊁激活函数㊁学习率等ꎮ
本次实验为了掌握不同超参数对CNN模型图像处理性能的不同影响经过多次调参ꎬ确立了一种由3卷积层与双池化层穿插连接ꎬ采用全局池化层替代传统卷积层的神经网络模型ꎮ该模型搭建平台为tensor ̄flow2.0.0GPU版ꎬ编译语言为Python3.7.3ꎬ计算机
CPU为i7㊁Windows10ꎮ
改进CNN模型结构表如表1所示ꎮ
表1㊀改进CNN模型结构表
模型参数卷积层C1池化层S2卷积层C3池化层S4全局池化层输出层数目64646464641大小
3∗3
2∗2
3∗3
2∗2
12
3.2.1㊀激活函数的选择
激活函数决定了CNN模型非线性表达能力ꎬ而该
激活函数通常根据经验来设置ꎮ但是这种方式随机性较大[12]ꎮ为此本文选择了常用的3种常用的激活函数进行对比试验ꎮ
3种常用的激活函数对比如表2所示ꎮ
表2㊀激活函数对比表实验
激活函数准确率训练时长1Relu
0.963
0.3552Sigmoid0.5630.2613
Tanh
0.9450.32
㊀㊀由表2可以看出:
实验1准确率最高ꎬ虽然在训练用时方面用时最
长ꎬ但为保证模型的准确度ꎬ在改进神经网络模型中ꎬ本文选用Relu激活函数ꎮ
3.2.2㊀优化器的选择在优化器选择实验中继续采用表1中的基准模型
参数设置ꎬ该实验的卷积层采用Relu激活函数ꎮ该实
985 第5期刘红军ꎬ等:基于GADF与卷积神经网络的滚动轴承故障诊断研究
验对目前最常用的SGDꎬRMSProp和Adam共3种优化器进行测试ꎮ
当学习速率为0.0001和0.001时ꎬAdam与RM ̄
SProp的准确率远高于SGDꎻ而当学习速率升高时ꎬRMSProp准确率急剧下降ꎬAdam优化器准确率依然稳
定在97%左右ꎮ
为此ꎬ本文选用Adam小批量迭代优化器ꎮ优化器准确率对比图如图4所示
图4㊀优化器准确率对比图
4㊀结果分析
4.1㊀改进的CNN模型评估
为对GADF图像编码下改进的CNN算法的准确性进行有效的评估ꎬ本文引用精确率与召回率作为评测指标ꎮ其中ꎬ精确率是指预测结果中符合该类故障的样本占结果总体的占有率ꎻ召唤率是指该类型故障真实样本中被成功预测的样本占类型总体的占有率ꎮ
其具体数学表达式如下:
P=TP/(TP+FP)R=TP/(TP+FN)F1=2TP/(2TP+FP+FN)
ìîíïï
ï
(8)
式中:P 精确率ꎻR 召回率ꎻTP 预测类型与其真实类型相符ꎻFP 预测类型为真ꎬ但实际类型为假ꎻFN 实际类型为真ꎬ但预测结果为假ꎻF1 精确率与召回率的调和均值ꎮ
根据改进的CNN的实验结果ꎬ并按公式计算得到的精确率和召回率ꎬ如表3所示ꎮ
表3㊀改进CNN模型诊断结果表
故障序号
精准率召回率F1调和均值
样本01.00001.00001.00004010.97500.97500.97504021.00001.00001.0000403
0.9744
0.9500
0.9620
40
(续表)故障序号
精准率召回率F1调和均值
样本41.00001.00001.00004050.97561.00000.98774061.00000.97500.98734071.00000.97560.98774080.97501.00000.9873409
0.95120.97500.963040100.97500.97500.975040111.0000
1.00001.000040
平均值/总数0.9855
0.9854
0.9667
480㊀㊀为了更深一层地了解改进CNN模型的故障判别情况ꎬ笔者引入了多分类混淆矩阵对故障结果进行全面分析ꎻ其根据故障的不同等级ꎬ样本的真实值与预测值ꎮ
故障被误判为何种类型故障等信息如图5所示
图5㊀混淆矩阵量化图
由图5可知:在480份测试集样本中ꎬ含有7个误
测样本ꎬ分别为:
将标号1类微小故障判断为标号9类轻型故障ꎻ将标号3类轻微故障判断为标号9类轻型故障和7类显著故障ꎻ将标号6类轻型故障判断为标号10类轻型故障ꎻ将标号8类中度故障判断为标号5类显著故障ꎻ将标号9类轻型故障判定为标号3类轻微故障ꎻ将标号10类轻型故障判定为标号1类微小故障ꎮ
通过分析以上故障类型可以发现ꎬ故障误诊类型多为轻型故障ꎬ误诊为中大型故障或同级故障类别发生错判ꎬ具有一定的预警作用ꎬ改进后的CNN模型在正常0㊁故障2㊁4和11的情况下诊断准确率为100%ꎬ说明这类故障特征较为明显ꎬ对旋转设备的危害最大ꎮ
经过试验验证可知ꎬ该算法综合故障识别率达
98.55%ꎬ对滚动轴承中的GADF编码图像具有精准的识别能力和优越的诊断能力ꎮ4.2㊀对比分析
为验证笔者提出的诊断方法的先进性与优越性ꎬ
095 机㊀㊀电㊀㊀工㊀㊀程
第38卷
本文引用格式:
刘红军ꎬ魏旭阳.基于GADF与卷积神经网络的滚动轴承故障诊断研究[J].机电工程ꎬ2021ꎬ38(5):587 ̄591ꎬ622.
LIUHong ̄junꎬWEIXu ̄yang.RollingbearingfaultdiagnosisbasedonGADFandconvolutionalneuralnetwork[J].JournalofMechanical&ElectricalEngineer ̄ingꎬ2021ꎬ38(5):587 ̄591ꎬ622.
«机电工程»杂志:http://www.meem.com.cn
此处将GADF ̄CNN算法与其余3种主流的智能故障诊断算法进行比较ꎮ3种主流智能故障诊断算法分别为BPNN㊁SAE和
SVM
[13 ̄15]
智能算法ꎮ
笔者将二维图像数据测试集按0~11故障诊断序号共12种情况ꎬ输入进其余智能算法模型进行训练ꎮ
智能算法的对比图如图6所示
图6㊀智能算法对比图
由图6可以看出:
BP神经网络模型(BPNN)与支持向量机(SVM)
的准确率一直维持在90%以下ꎬ准确率较低ꎻ而堆栈自动编码器(SAE)的准确率略低于GADF ̄CNNꎮ
由此可见ꎬ改进的CNN模型更适用与对GADF编码图像数据的处理ꎮ
详细的准确率如表4所示ꎮ
表4㊀详细的准确率
故障序号
BPNNSVMSAEGADF ̄CNN087.5%92.5%95%
100%
185%
87.5%92.5%97.55%282.5%90%
90%100%378.5%87.5%95%95%485%92.5%97.5%100%587.5%90%95%100%682.5%85%92.5%
97.5%785%87.5%95%100%882.5%87.5%95%97.5%9
85%85%
87.5%
97.5%1080%87.5%95%97.5%1182.5%
90%90%100%平均值/总数83.63%
88.54%93.33%
98.55%5㊀结束语
为充分发挥深度学习识别二维图像在滚动轴承故障检测方面的优势ꎬ笔者提出了一种新的故障诊断方式ꎬ即格拉姆角差场(GADF)结合改进卷积神经网络(CNN)的智能故障诊断模型ꎬ并结合改进的CNN模型对
滚动轴承故障进行了预测ꎬ达到了高精度诊断的效果ꎮ与其他智能诊断算法相比ꎬGADF ̄CNN具有以下优势:(1)在对诊断信号的特征提取方面ꎬ相比其他算
法的一维时序输入或模态分解等方法ꎬ采用领先的格拉姆角差场法将诊断信号转化为具有有效特征的二维图像ꎬ对诊断信号的特征提取更为全面ꎬ同样也更适合CNN(卷积神经网络)的训练与测试ꎻ
(2)在卷积神经网络模型的搭建方面ꎬ与其他智能算法的诊断相比ꎬ由于输入数据转化为真正的二维图像ꎬ使得卷积神经网络强大的线性表达能力可以全
面地发挥出来ꎬ为此改进CNN模型添加了全局池化层㊁Adam优化器㊁交叉熵损失函数等先进的超参数ꎬGADF ̄CNN算法鲁棒性和泛化能力ꎮ
本文的算法已经可以在不同电机负载状态下ꎬ对
来自于电机驱动端的振动信号做出有效诊断ꎮ但由于数据库的有限性ꎬ本文没有获取数据库以外的滚动轴承数据进行诊断ꎬ该算法仍然存在一定的局限性ꎮ
获取数据库以外的数据将成为笔者进行下一步实验研究的重点ꎮ
参考文献(References):
[1]㊀许㊀凯ꎬ许黎明ꎬ周大朝ꎬ等.基于Stockwell变换的滚动
轴承故障诊断方法[J].机械制造ꎬ2019ꎬ57(4):98 ̄102.[2]㊀杨㊀宇ꎬ于德介ꎬ程军圣.基于EMD与神经网络的滚动轴承故障诊断方法[J].振动与冲击ꎬ2005ꎬ24(1):85 ̄88.[3]㊀刘㊀韬ꎬ陈㊀进ꎬ董广明.KPCA和耦合隐马尔科夫模型
在轴承故障诊断中的应用[J].振动与冲击ꎬ2014ꎬ33(21):85 ̄89.
[4]㊀张㊀鑫ꎬ郭顺生ꎬ李益兵ꎬ等.基于拉普拉斯特征映射和深
度置信网络的半监督故障识别[J].机械工程学报ꎬ2020ꎬ
56(1):69 ̄81.
(下转第622页)
195 第5期刘红军ꎬ等:基于GADF与卷积神经网络的滚动轴承故障诊断研究

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。