高级搜索

留言板

尊敬的读者、作者、审稿人, 关于本刊的投稿、审稿、编辑和出版的任何问题, 您可以本页添加留言。我们将尽快给您答复。谢谢您的支持!

姓名
邮箱
手机号码
标题
留言内容
验证码

时变转速下基于改进图注意力网络的轴承半监督故障诊断

邵海东 颜深 肖一鸣 刘翊

邵海东, 颜深, 肖一鸣, 刘翊. 时变转速下基于改进图注意力网络的轴承半监督故障诊断[J]. 电子与信息学报, 2023, 45(5): 1550-1558. doi: 10.11999/JEIT220303
引用本文: 邵海东, 颜深, 肖一鸣, 刘翊. 时变转速下基于改进图注意力网络的轴承半监督故障诊断[J]. 电子与信息学报, 2023, 45(5): 1550-1558. doi: 10.11999/JEIT220303
SHAO Haidong, YAN Shen, XIAO Yiming, LIU Yi. Semi-supervised Bearing Fault Diagnosis Using Improved Graph Attention Network under Time-varying Speeds[J]. Journal of Electronics & Information Technology, 2023, 45(5): 1550-1558. doi: 10.11999/JEIT220303
Citation: SHAO Haidong, YAN Shen, XIAO Yiming, LIU Yi. Semi-supervised Bearing Fault Diagnosis Using Improved Graph Attention Network under Time-varying Speeds[J]. Journal of Electronics & Information Technology, 2023, 45(5): 1550-1558. doi: 10.11999/JEIT220303

时变转速下基于改进图注意力网络的轴承半监督故障诊断

doi: 10.11999/JEIT220303
基金项目: 国家重点研发计划(2020YFB1712100),国家自然科学基金(51905160),湖南省优秀青年科学基金(2021JJ20017),上海市空间导航与定位技术重点实验室开放课题(202105)
详细信息
    作者简介:

    邵海东:男,博士,副教授,博士生导师,研究方向为故障诊断与智能运维、数据挖掘与信息融合

    颜深:男,硕士生,研究方向为图神经网络与半监督学习

    肖一鸣:男,硕士生,研究方向为深度迁移学习与智能故障诊断

    刘翊:男,博士,教授级高级工程师,研究方向为智能制造与工业大数据

    通讯作者:

    邵海东 hdshao@hnu.edu.com

  • 中图分类号: TH133.3; TP183

Semi-supervised Bearing Fault Diagnosis Using Improved Graph Attention Network under Time-varying Speeds

Funds: The National Key Research and Development of China (2020YFB1712100), The National Natural Science Foundation of China (51905160), The Natural Science Fund for Excellent Young Scholars of Hunan Province (2021JJ20017), The Opening Project of Shanghai Key Laboratory of Space Navigation and Positioning Techniques (202105)
  • 摘要: 新近的基于图神经网络(GNN)的轴承半监督故障诊断研究仍存在标签信息挖掘不充分和诊断场景较理想等问题。工程实际中,轴承经常运行于启停等时变转速工况,且故障标签样本的获取成本越发昂贵。针对以上挑战,该文提出时变转速下基于改进图注意力网络(GAT)的轴承半监督故障诊断新方法。基于K最近邻(KNN)算法和平滑假设(SA)设计伪标签传播策略,将标签信息沿边传播给分布相似的邻域样本,从而充分利用有限样本的标签信息。将每个振动频谱样本视为一个节点,构建基于节点级图注意力网络的半监督学习模型,通过注意力机制进一步挖掘代表性的轴承故障特征。将所提方法用于分析两组时变转速下轴承故障实验数据,结果表明所提方法能够在不超过2%的低标签率情况下,准确诊断轴承的不同故障模式,性能优于其他常用的图神经网络半监督学习方法。
  • 轴承作为机械设备中重要的部件,若发生故障,可能会造成巨大的经济损失。得益于传感器等技术的发展,轴承在运行中产生的海量数据得以采集,如何挖掘其中的有效信息,是近年来轴承故障诊断领域研究的热点[1,2]。智能故障诊断是基于数据驱动的自动化推理过程。近年来,多种深度学习模型已成功应用于智能故障诊断中[3]。然而,它们的有效训练均依赖于大量的标签数据,这在实际故障诊断中是相当困难的[4]。因此,在稀缺标签样本下研究高精度的轴承故障诊断方法具有重要意义。

    半监督学习旨在利用有限的标签数据,从无标签的数据中提取出有效信息,近年来在智能故障诊断中得到了广泛的研究[5]。Yu等人[6]研究了一种基于一致性正则化的数据增强方法,在标注样本有限的情况下,实现了轴承的故障诊断。Tao等人[7]构造了一种基于半监督核局部Fisher判别分析的轴承缺陷检测模型,通过使用伪标签,有效地提取分类特征。Wu等人[8]提出了一种基于混合分类自编码器的半监督故障诊断方法,可以同时使用有标签和无标签的数据训练模型,获得较高的诊断准确率。Nie等人[9]将带噪声标签的故障诊断难题转化为半监督学习过程,并基于此提出了一种两阶段故障诊断框架。

    以上所提及的方法尽管能在一定程度上应对故障诊断任务中标签样本不足的挑战,然而它们均是从欧几里得空间的角度孤立地对多个样本进行分析,致使样本彼此之间隐含的关联信息没有得到有效开发[10]。图神经网络(Graph Neural Network, GNN)能从非欧几里得空间建立模型,建立节点间的关系结构,通过连接的边传播邻域节点的特征信息,以更新每个节点的特征表示。近两年,国内外学者开始逐渐探索GNN在轴承半监督故障诊断领域的应用研究。2021年,Zhao等人[11]构建了一种新型半监督图卷积深度信任网络,结合数据的标记信息和未标记信息,实现了10%标签率情景下的轴承故障诊断。2021年,Gao等人[12]引入了半监督图卷积网络,学习数据间的内在联系,实现了标签率为5%情景下的轴承故障诊断。2021年,Tang等人[13]开发了一种基于半监督条件随机场的图注意力网络算法,在标签率最低为6.67%的情况下,实现了对电机轴承故障严重程度和工作状态的识别。

    上述的GNN新近研究虽在轴承半监督故障诊断中均取得较高的准确率,但仍普遍存在以下局限:(1)上述研究中忽略了稀缺的样本标签信息和分布周边的大量无标签样本之间的联合依赖关系,致使有限的标签信息未得到更充分的利用。理论上,有边连接关系的样本特征分布较为相似,其所属标签类别较大概率是相同的,可倾向于在连接的样本之间共享。因此,已有的稀缺标签信息可被充分挖掘以辅助预测无标签样本。(2)上述研究均局限于轴承稳态运行工况下的理想诊断场景,未涉及信号征兆与故障模式映射复杂的时变转速工况。然而,实际工程中机械设备经常经历启停阶段,或是因具体任务要求需进行速度切换,产生速度波动,因此时变转速场景下的半监督故障诊断研究更具现实意义。(3)上述研究的半监督故障诊断任务中,样本的标签率均不低于5%。工程实际中,由于标记数据的成本越发昂贵以及海量监测数据的长期积累,所能获取的标签样本将越发有限。因此,更低标签率下基于GNN的半监督故障诊断方法亟需进一步研究。

    综上,本文提出时变转速下基于改进图注意力网络(Graph ATtention Network, GAT)的轴承半监督故障诊断新方法。基于K最近邻(K Nearlist Neighbor, KNN)算法和半监督学习的平滑假设(Smoothness Assumption, SA)设计KNN-SA策略,可以通过样本间特征的相似性建立连接关系,充分挖掘样本间标签的联合依赖关系,将标签信息沿边传播给分布相似的邻域样本,以充分利用有限的标签信息。将每个频谱样本视为一个节点,构建基于图注意力网络的轴承半监督学习模型,通过注意力机制学习有效的节点特征表示。在两个轴承运行实验案例中与多种方法对比,结果表明所提方法能够自适应地从时变转速工况的振动信号中提取可用的故障特征,在标签率不超过2%的较极端情况下,仍有准确稳定的诊断效果,性能优于其他常用的GNN半监督学习方法。

    不同于卷积神经网络(Convolutional Neural Network, CNN),GNN是一类基于图结构数据的学习模型,它可以在非欧氏空间基于节点间连接关系定义图卷积,CNN和GNN的卷积区别可直观表示如图1所示。所涉及的图结构可简化为

    图 1  CNN和GNN的卷积示意图
    G=(H,A)
    (1)

    其中,H={h1,h2,,hn}Rn×d代表节点的集合;n是节点的数量;d是输入的节点特征维度;ARn×n代表节点间连接关系的邻接矩阵。

    图卷积层是通过聚合邻域节点特征来更新每个节点的特征,一般是将图G作为输入,经过卷积层后,输出新的节点集合H={h1,h2,,hn}Rn×dd是输出的节点特征维度。图卷积层可表示为

    hi=Γ(hi,Υ({hj|jNi}))
    (2)

    其中,Ni是节点hi的邻域节点数;Γ()代表非线性层;Υ()代表某种节点聚合模式。近两年来,GNN尤其是图卷积神经网络在多个领域受到了较为广泛的研究[14,15]

    将采集的轴承时域振动信号进行快速傅里叶变换(Fast Fourier Transform, FFT),利用欧几里得距离判断所有频谱信号样本之间的邻接关系来构造频谱样本KNN图,能很好地反映样本之间的局部几何性质,构建流程如图2所示。KNN算法的距离度量公式可表示为

    图 2  基于频谱信号的频谱样本KNN图构建流程
    Lij=(dl=1|h(l)ih(l)j|2)12
    (3)

    其中,Lij表示节点hi和节点hj的距离;h(l)i表示节点hi的第l维特征;每个节点代表每个经过FFT的频谱样本。频谱样本KNN图中边的构造可表示为

    Aij=KNN(k,Lij,Ωi), AijA
    (4)

    其中,Ωi={Li1,Li2,,Lin}hi与所有节点的距离集合;k是频谱样本KNN图的超参数;若LijΩi集合中的k个最小值,KNN()=1,否则KNN()=0

    半监督学习中SA认为,相似的样本特征倾向于有着相同的分类标签[16]。综上,设计了基于KNN-SA的伪标签传播策略,它可以表示为

    yj=SA(Aij,(hi,yi),hj)=yi, Aij=1
    (5)

    其中,yi是节点hi的真实标签(独热编码格式);yj是节点hj经过SA得到的伪标签;yi,yjYY为标签样本集(包含真实标签和伪标签)。

    KNN-SA策略可以通过样本间特征的相似性建立彼此的连接关系,充分利用标签间的联合依赖关系,将标签信息沿连接边进行传播,给未标签的样本赋予伪标签,使有限的标签信息得到更充分的挖掘,增加模型可学习的标签信息,其直观流程如图3所示。

    图 3  基于KNN-SA的伪标签传播策略

    GAT[17]是一种先进的GNN模型,可用于图形级或节点级的故障诊断[18]。图形级的故障诊断旨在将每个样本生成一张对应的图,样本之间没有建立联系;节点级的故障诊断将每个样本视为一个节点,多个节点构成一张图,可以通过连接边建立节点间的联系,实现半监督学习。因此,将每个频谱样本视为GAT中的一个节点,通过注意力机制为不同的邻域节点自适应地分配权重,以实现更好的特征提取和故障诊断。

    以下描述节点级图注意力层的具体构造,输入为一组节点特征H,图注意力层能输出一组新的节点特征H。为了获得足够的表达能力,需要进行线性变换将输入特征转化为更高层次的特征。为此,作为初始步骤,对每个节点应用权值矩阵WRd×d参数化的共享线性变换,然后在节点上执行一种共享的注意机制F计算注意力系数。为了使注意力系数在不同节点之间容易比较,使用softmax函数对所有注意力系数进行归一化

    eij=F(Whi,Whj)
    (6)
    αij=softmaxj(eij)=exp(eij)jNiexp(eij)
    (7)

    其中,注意机制F为一种单层前馈神经网络;eij表示节点hj的特征对节点hi的重要性;αij为节点hi和节点hj之间归一化后的注意力系数。

    通过归一化的注意力系数计算对应特征的线性组合,作为每个节点的输出特征,表示为

    hi=σ(jNiαijWhj)
    (8)

    其中,σ为ReLU激活函数。节点级图注意力层的特征更新机制如图4所示,权值矩阵W和注意机制F是图注意层在迭代过程中自动学习的参数,以自适应地为邻域节点分配权重。每个图注意层中均引入了批标准化,以应对时变转速频域信号的频移和幅变特性,加快网络的收敛[19]。输入节点集合H经过两个图注意力层后得到输出节点集合H,再将其输入至全连接层后得到预测标签集合Z={z1,z2,,zn},表示为

    图 4  节点级图注意力层的特征更新机制
    Z=FC(H)
    (9)

    其中,FC()表示全连接层,节点级GAT半监督学习模型的构建过程如图5所示。

    图 5  节点级GAT半监督学习模型构建

    振动信号首先经过KNN-SA策略得到含伪标签的频谱样本KNN图,将其输入节点级GAT半监督学习模型得到预测标签,再与标签样本比较,进行损失函数的计算

    Loss=1IIi=1Cc=1y(c)iln(z(c)i)
    (10)

    其中,I为标签数(包含真实标签和伪标签);C为故障类别的数量;y(c)i为标签yi(独热编码)第c维的值;z(c)i为预测标签zi(独热编码)第c维的值。

    最终,所提方法的算法流程见算法1。需要强调的是,共有两个频谱样本的KNN图,分别作为含伪标签的训练集和无标签的测试集。

    算法1 所提方法的算法流程
     输入:含伪标签的训练集(训练输入),无标签的测试集(测试输入)
     (1)设定k值、图注意力层层数、激活函数、学习率、迭代次数
      T等模型超参数
     (2)随机初始化模型的权重等参数
     (3)基于式(3)求解Lij,基于式(4)构造边,得到频谱样本KNN图
     (4)基于式(5)得到标签样本集Y
     (5)For epoch in range(T):
     (6)   基于式(8)、式(9)得到输出节点集合Z
     (7)   基于式(10)计算损失函数Loss
     (8)   根据损失函数Loss基于Adam优化器更新模型参数
     (9)End For
     (10)将训练好的模型用于测试集的预测
     输出:测试集预测标签的准确率
    下载: 导出CSV 
    | 显示表格

    在两个时变转速的轴承故障诊断案例中验证所提方法的优越性和有效性。

    案例1 数据集源自西安交通大学的机械故障综合模拟试验台[20,21]。电机轴承型号为NSK6203,故障轴承位于电机驱动端。信号采样频率为25.6 kHz,每次实验采集时长为15 s,包含一个完整的从静止状态逐渐加速至3 000 r/min,后保持稳定,最后逐渐减速为0的启停过程。案例1选用了包括6种故障和正常状态共7种不同状态的轴承振动信号,每个样本的长度为1 024个采样点。

    案例2 数据集源自渥太华大学的SpectraQuest机械故障模拟器[22]。安装两个ER16K球轴承支撑转轴,可更换为不同健康状态的轴承。加速计(ICP加速计,623C01型)放置在实验轴承座上进行振动数据采集,增量式编码器(EPC-775型)测量轴转速。信号采样频率为200 kHz,每次实验采样持续时间为10 s,包括加速和减速两个过程。案例2选用了包括4种故障和正常状态共5种不同状态的轴承振动信号,每个样本的长度为10 000个采样点。

    两个实验案例的轴承数据集详细描述如表1所列。对于案例1,轴承处于包含启停阶段的时变转速过程,每类状态的样本总数为300,其中测试集样本数为250,训练集的样本数为50,标签的样本数仅为6,标签率为2%;对于案例2,轴承处于先加速和后减速的随机速度波动过程,每类状态的样本总数为200,其中测试集样本数为190,训练集样本数为10,标签的样本数仅为1,标签率为0.5%。

    表 1  轴承数据集信息
    故障诊断案例时变转速(r/min)轴承状态样本总数标签样本数训练样本数
    案例10-3 000-0正常2 100(300×7)42(6×7)(标签率2%)350(50×7)
    内圈轻度故障
    内圈中度故障
    内圈重度故障
    外圈轻度故障
    外圈中度故障
    外圈重度故障
    案例2882-1 518-1 260正常1 000(200×5)5(1×5)(标签率0.5%)50(10×5)
    906-1 464-1 122内圈故障
    865-1 421-928滚珠故障
    840-1 302-870外圈故障
    795-1 280-766复合故障
    下载: 导出CSV 
    | 显示表格

    为避免实验结果的偶然性,所有方法的实验均执行了5次重复实验。运行配置描述如下:操作系统为Windows10;CPU为i5-10400F;GPU为GTX1650;软件运行版本为Pytorch1.7.1。模型的主要超参数设置如下:迭代次数设置为400次,学习率大小设为0.01,在迭代至200和300次时分别衰减0.1倍。其他超参数可见源码:https://github.com/yanshen210/KNN-SA-GNN

    首先,验证GAT的优越性。将GAT与GraphSage[23], GCN[24], ChebyNet[25]等3种常用的GNNs进行对比。两个案例中4种GNNs的诊断结果如图6所示,案例1中GAT, GraphSage, GCN和ChebyNet的准确率分别为86.07%, 83.82%, 80.63%和80.79%,且GAT在案例2中的准确率达到98.55%,在两个案例中均取得了最高的准确率。

    图 6  两个案例中4种GNNs的诊断准确率

    然后,验证KNN-SA策略的有效性。将上述4种GNNs分别加入KNN-SA策略,并与未加入该策略的诊断结果进行对比,实验具体结果如表2所示。所提的改进图注意力网络(K Nearlist Neighbor- Smoothness Assumption-Graph Attention Network, KNN-SA-GAT)在两个案例中均取得了最高的平均准确率,分别为89.62%和99.58%,较GAT的诊断准确率分别提升了3.55%和1.03%。其原因在于KNN-SA策略可以通过样本特征的相似性建立连接关系,充分利用样本间标签的联合依赖关系,将稀缺的标签信息沿连接边进行传播,给无标签的样本赋予伪标签,以增加模型可学习的标签信息。在两个案例中,KNN-SA策略对于GraphSage的诊断准确率分别提升了1.10%和15.37%,对于GCN的诊断准确率分别提升了3.90%和10.44%,对于ChebyNet的诊断准确率分别提升了5.46%和32.00%。这说明KNN-SA策略对于GNNs效果的提升具有普适性和鲁棒性。所有方法的训练时间和测试时间如表2所示,所提方法在两个案例中的训练时间分别为5.39 s和6.20 s,测试时间均为0.01 s。所有方法的运行时间均在同一数量级,因为它们的网络结构类似,且迭代次数相同。

    表 2  各方法的诊断结果和运行时间
    半监督故障诊断方法准确率和方差 (%)训练时间 (s)测试时间 (s)
    案例1案例2案例1案例2案例1案例2
    改进图注意力网络(KNN-SA-GAT)89.62±3.3699.58±0.455.396.200.010.01
    GAT86.07±4.4298.55±0.815.396.210.010.01
    KNN-SA-GraphSage84.92±0.8782.38±2.224.496.640.010.03
    GraphSage83.82±2.6167.01±12.444.496.630.010.03
    KNN-SA-GCN84.53±2.5984.99±4.444.595.240.010.01
    GCN80.63±5.1174.55±7.584.555.230.010.01
    KNN-SA-ChebyNet86.25±5.5879.85±4.395.407.440.020.05
    ChebyNet80.79±2.1747.85±10.485.397.450.020.05
    下载: 导出CSV 
    | 显示表格

    为了进一步展示所有方法对不同状态的诊断效果,将案例1各方法第5次实验的混淆矩阵可视化,如图7所示。每类状态的测试数据为250个,共7类状态,0-6分别与算法1中案例1轴承状态一一对应,横坐标为预测标签,纵坐标为真实标签。根据混淆矩阵可知,各种方法对于处理不同状态类别的轴承优势和不足不尽相同,其原因在于各种方法的内在原理机制存在差异,且学习参数的初始化是随机的。但总体而言,在包含启停阶段的时变转速工况下,加入KNN-SA策略的4种GNNs均优于对应未加入的GNNs,且所提方法的诊断效果最佳。

    图 7  案例1各方法第5次实验的混淆矩阵图

    为更好地展现所有方法的特征提取性能,通过t分布邻域嵌入算法,将案例2各方法第5次实验的输出特征向量进行2维可视化,如图8所示。由图可知,所提方法和GAT学习的特征最具代表性,说明了即使在信号征兆不明显的时变转速工况和低标签率的情况下,图注意力机制仍然是有效的,能够自适应分配邻域节点的权重,学习有效的特征表示。

    图 8  案例2各方法第5次实验的特征2维可视化

    其次,探讨频谱样本KNN图构造中参数k的取值对于诊断结果和伪标签的影响。图9为所提方法在不同k值下对于两个案例的诊断准确率和伪标签准确率。可以直观地看出,案例2的诊断准确率和伪标签准确率均优于案例1,其主要原因在于案例1的轴承振动信号包含完整的启停过程,加减速跨度更大。两个案例中,伪标签的准确率随k值增加呈现波动向下趋势,主要是因为k值增加会带来伪标签数量的增加,导致错误的概率增大(伪标签数量=k值×标签样本)。在两个案例中,k值为3时均有最高的诊断准确率,且对应的伪标签准确率分别为93.65%和100%,可信度高。

    图 9  两个案例中不同k值下所提方法的诊断准确率和伪标签准确率

    最后,探讨低标签率对所提方法的影响,案例1中5次实验的诊断准确率和平均准确率如图10所示。标签率分别设置为2%, 3%, 4%, 5%,每种状态训练集样本数均为50,其中标签样本数分别为6, 9, 12, 15。对于案例1包含启停全过程的时变转速轴承数据,在标签率仅为2%的情况下,所提方法仍有89.62%的平均准确率,且随着标签率的增加,所提方法的诊断准确率呈现上升趋势,在标签率为4%的情况下,所提方法平均准确率可达94.73%。

    图 10  案例1中所提方法在不同低标签率下的诊断效果

    本文工作探索了时变转速下低标签率的轴承半监督故障诊断新途径。从图结构的角度,设计KNN-SA伪标签传播策略,能充分利用样本间的标签相关性,进一步挖掘标签信息。将每个振动频谱样本视为一个节点,构建基于节点级GAT的半监督学习模型,通过注意力机制学习代表性的诊断特征表示。在时变转速和标签率不超过2%的两个轴承半监督故障诊断案例中验证了所提方法的优越性和有效性。诊断结果表明:(1)设计的KNN-SA策略对于不同GNNs的效果提升具有普适性和鲁棒性。(2)构建的节点级GAT能够更好地从时变转速下的样本中提取故障特征。(3)所提方法能较为准确地诊断出轴承不同故障模式,性能优于其他常用的GNN半监督学习方法。研究工作仍存在以下不足:(1)对于包含启停阶段的时变转速轴承故障诊断效果仍有一定的提升空间。(2)因获取到的数据较为有限,所提方法未能在实际工业场景中进一步得以验证。极低标签率和时变转速等涉及工程实际的研究具有良好的应用价值,值得深入研究。在未来,我们将继续努力探索,并将其应用于更为复杂的实际工业场景中。

  • 图  1  CNN和GNN的卷积示意图

    图  2  基于频谱信号的频谱样本KNN图构建流程

    图  3  基于KNN-SA的伪标签传播策略

    图  4  节点级图注意力层的特征更新机制

    图  5  节点级GAT半监督学习模型构建

    图  6  两个案例中4种GNNs的诊断准确率

    图  7  案例1各方法第5次实验的混淆矩阵图

    图  8  案例2各方法第5次实验的特征2维可视化

    图  9  两个案例中不同k值下所提方法的诊断准确率和伪标签准确率

    图  10  案例1中所提方法在不同低标签率下的诊断效果

    算法1 所提方法的算法流程
     输入:含伪标签的训练集(训练输入),无标签的测试集(测试输入)
     (1)设定k值、图注意力层层数、激活函数、学习率、迭代次数
      T等模型超参数
     (2)随机初始化模型的权重等参数
     (3)基于式(3)求解Lij,基于式(4)构造边,得到频谱样本KNN图
     (4)基于式(5)得到标签样本集Y
     (5)For epoch in range(T):
     (6)   基于式(8)、式(9)得到输出节点集合Z
     (7)   基于式(10)计算损失函数Loss
     (8)   根据损失函数Loss基于Adam优化器更新模型参数
     (9)End For
     (10)将训练好的模型用于测试集的预测
     输出:测试集预测标签的准确率
    下载: 导出CSV

    表  1  轴承数据集信息

    故障诊断案例时变转速(r/min)轴承状态样本总数标签样本数训练样本数
    案例10-3 000-0正常2 100(300×7)42(6×7)(标签率2%)350(50×7)
    内圈轻度故障
    内圈中度故障
    内圈重度故障
    外圈轻度故障
    外圈中度故障
    外圈重度故障
    案例2882-1 518-1 260正常1 000(200×5)5(1×5)(标签率0.5%)50(10×5)
    906-1 464-1 122内圈故障
    865-1 421-928滚珠故障
    840-1 302-870外圈故障
    795-1 280-766复合故障
    下载: 导出CSV

    表  2  各方法的诊断结果和运行时间

    半监督故障诊断方法准确率和方差 (%)训练时间 (s)测试时间 (s)
    案例1案例2案例1案例2案例1案例2
    改进图注意力网络(KNN-SA-GAT)89.62±3.3699.58±0.455.396.200.010.01
    GAT86.07±4.4298.55±0.815.396.210.010.01
    KNN-SA-GraphSage84.92±0.8782.38±2.224.496.640.010.03
    GraphSage83.82±2.6167.01±12.444.496.630.010.03
    KNN-SA-GCN84.53±2.5984.99±4.444.595.240.010.01
    GCN80.63±5.1174.55±7.584.555.230.010.01
    KNN-SA-ChebyNet86.25±5.5879.85±4.395.407.440.020.05
    ChebyNet80.79±2.1747.85±10.485.397.450.020.05
    下载: 导出CSV
  • [1] 雷亚国, 贾峰, 孔德同, 等. 大数据下机械智能故障诊断的机遇与挑战[J]. 机械工程学报, 2018, 54(5): 94–104. doi: 10.3901/JME.2018.05.094

    LEI Yaguo, JIA Feng, KONG Detong, et al. Opportunities and challenges of machinery intelligent fault diagnosis in big data era[J]. Journal of Mechanical Engineering, 2018, 54(5): 94–104. doi: 10.3901/JME.2018.05.094
    [2] 邵海东, 张笑阳, 程军圣, 等. 基于提升深度迁移自动编码器的轴承智能故障诊断[J]. 机械工程学报, 2020, 56(9): 84–90. doi: 10.3901/JME.2020.09.084

    SHAO Haidong, ZHANG Xiaoyang, CHENG Junsheng, et al. Intelligent fault diagnosis of bearing using enhanced deep transfer auto-encoder[J]. Journal of Mechanical Engineering, 2020, 56(9): 84–90. doi: 10.3901/JME.2020.09.084
    [3] 文成林, 吕菲亚. 基于深度学习的故障诊断方法综述[J]. 电子与信息学报, 2020, 42(1): 234–248. doi: 10.11999/JEIT190715

    WEN Chenglin and LV Feiya. Review on deep learning based fault diagnosis[J]. Journal of Electronics &Information Technology, 2020, 42(1): 234–248. doi: 10.11999/JEIT190715
    [4] LIANG Pengfei, DENG Chao, WU Jun, et al. Intelligent fault diagnosis via semisupervised generative adversarial nets and wavelet transform[J]. IEEE Transactions on Instrumentation and Measurement, 2020, 69(7): 4659–4671. doi: 10.1109/tim.2019.2956613
    [5] RAZAVI-FAR R, HALLAJI E, FARAJZADEH-ZANJANI M, et al. A semi-supervised diagnostic framework based on the surface estimation of faulty distributions[J]. IEEE Transactions on Industrial Informatics, 2019, 15(3): 1277–1286. doi: 10.1109/tii.2018.2851961
    [6] YU Kun, MA Hui, LIN Tianran, et al. A consistency regularization based semi-supervised learning approach for intelligent fault diagnosis of rolling bearing[J]. Measurement, 2020, 165: 107987. doi: 10.1016/j.measurement.2020.107987
    [7] TAO Xinmin, REN Chao, LI Qing, et al. Bearing defect diagnosis based on semi-supervised kernel Local Fisher discriminant analysis using pseudo labels[J]. ISA Transactions, 2021, 110: 394–412. doi: 10.1016/j.isatra.2020.10.033
    [8] WU Xinya, ZHANG Yan, CHENG Changming, et al. A hybrid classification autoencoder for semi-supervised fault diagnosis in rotating machinery[J]. Mechanical Systems and Signal Processing, 2021, 149: 107327. doi: 10.1016/j.ymssp.2020.107327
    [9] NIE Xiaoyin and XIE Gang. A two-stage semi-supervised learning framework for fault diagnosis of rotating machinery[J]. IEEE Transactions on Instrumentation and Measurement, 2021, 70: 3521212. doi: 10.1109/tim.2021.3091489
    [10] WU Zonghan, PAN Shirui, CHEN Fengwen, et al. A comprehensive survey on graph neural networks[J]. IEEE Transactions on Neural Networks and Learning Systems, 2021, 32(1): 4–24. doi: 10.1109/TNNLS.2020.2978386
    [11] ZHAO Xiaoli, JIA Minping, and LIU Zheng. Semisupervised graph convolution deep belief network for fault diagnosis of electormechanical system with limited labeled data[J]. IEEE Transactions on Industrial Informatics, 2021, 17(8): 5450–5460. doi: 10.1109/tii.2020.3034189
    [12] GAO Yiyuan, CHEN Mang, and YU Dejie. Semi-supervised graph convolutional network and its application in intelligent fault diagnosis of rotating machinery[J]. Measurement, 2021, 186: 110084. doi: 10.1016/j.measurement.2021.110084
    [13] TANG Yao, ZHANG Xiaofei, ZHAI Yujia, et al. Rotating machine systems fault diagnosis using semisupervised conditional random field-based graph attention network[J]. IEEE Transactions on Instrumentation and Measurement, 2021, 70: 3521010. doi: 10.1109/tim.2021.3091212
    [14] 向敏, 饶华阳, 张进进, 等. 基于图卷积神经网络的软件定义电力通信网络路由控制策略[J]. 电子与信息学报, 2021, 43(2): 388–395. doi: 10.11999/JEIT190971

    XIANG Min, RAO Huayang, ZHANG Jinjin, et al. Software-defined power communication network routing control strategy based on graph convolution network[J]. Journal of Electronics &Information Technology, 2021, 43(2): 388–395. doi: 10.11999/JEIT190971
    [15] 盛晓光, 王颖, 钱力, 等. 基于图卷积半监督学习的论文作者同名消歧方法研究[J]. 电子与信息学报, 2021, 43(12): 3442–3450. doi: 10.11999/JEIT200905

    SHENG Xiaoguang, WANG Ying, QIAN Li, et al. Author name disambiguation based on semi-supervised learning with graph convolutional network[J]. Journal of Electronics &Information Technology, 2021, 43(12): 3442–3450. doi: 10.11999/JEIT200905
    [16] CHEN Ke and WANG Shihai. Semi-supervised learning via regularized boosting working on multiple semi-supervised assumptions[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2011, 33(1): 129–143. doi: 10.1109/TPAMI.2010.92
    [17] VELIČKOVIĆ P, CUCURULL G, CASANOVA A, et al. Graph attention networks[EB/OL]. https://arxiv.org/abs/1710.10903, 2018.
    [18] LI Tianfu, ZHOU Zheng, LI Sinan, et al. The emerging graph neural networks for intelligent fault diagnostics and prognostics: A guideline and a benchmark study[J]. Mechanical Systems and Signal Processing, 2022, 168: 108653. doi: 10.1016/j.ymssp.2021.108653
    [19] WANG Jinrui, LI Shunming, HAN Baokun, et al. Construction of a batch-normalized autoencoder network and its application in mechanical intelligent fault diagnosis[J]. Measurement Science and Technology, 2019, 30(1): 015106. doi: 10.1088/1361-6501/aaf319
    [20] LIU Shen, CHEN Jinglong, HE Shuilong, et al. Subspace network with shared representation learning for intelligent fault diagnosis of machine under speed transient conditions with few samples[J]. ISA Transactions, 2022, 128: 531–544.
    [21] SHI Zhen, CHEN Jinglong, ZI Yanyang, et al. A novel multitask adversarial network via redundant lifting for multicomponent intelligent fault detection under sharp speed variation[J]. IEEE Transactions on Instrumentation and Measurement, 2021, 70: 3511010. doi: 10.1109/tim.2021.3055821
    [22] HUANG Huan and BADDOUR N. Bearing vibration data collected under time-varying rotational speed conditions[J]. Data in Brief, 2018, 21: 1745–1749. doi: 10.1016/j.dib.2018.11.019
    [23] HAMILTON W L, YING R, and LESKOVEC J. Inductive representation learning on large graphs[C]. The 31st International Conference on Neural Information Processing Systems, Long Beach, USA, 2017: 1025–1035.
    [24] DEFFERRARD M, BRESSON X, and VANDERGHEYNST P. Convolutional neural networks on graphs with fast localized spectral filtering[C]. The 30th International Conference on Neural Information Processing Systems, Barcelona, Spain, 2016: 3844–3852.
    [25] KIPF T N and WELLING M. Semi-supervised classification with graph convolutional networks[C]. The 5th International Conference on Learning Representations, Toulon, France, 2017.
  • 期刊类型引用(7)

    1. 荆新岚,黄民,马超. 基于特征融合与改进半监督学习的变转速轴承故障诊断. 组合机床与自动化加工技术. 2025(01): 132-137 . 百度学术
    2. 刘馨雅,马超,黄民,张占一. 利用WDCNN-GRU模型的变转速轴承故障诊断技术研究. 组合机床与自动化加工技术. 2025(01): 138-142+149 . 百度学术
    3. 伍章俊,许仁礼,方刚,邵海东. 一种面向旋转机械多传感器故障诊断的模态融合深度聚类方法. 电子与信息学报. 2025(01): 244-259 . 本站查看
    4. 严颖,蔡骏,吴奇,张欣,杨溢. 基于投票机制的暖通空调空气处理单元传感器故障诊断. 电子与信息学报. 2024(01): 258-266 . 本站查看
    5. 韩延,林志超,黄庆卿,向敏,文瑞,张焱. 正交约束域适应的跨工况滚动轴承剩余使用寿命预测方法. 电子与信息学报. 2024(03): 1043-1050 . 本站查看
    6. 石明宽,丁传仓,王锐,黄伟国,朱忠奎. 面向变工况下工业流数据故障诊断的持续迁移学习系统. 仪器仪表学报. 2024(04): 10-16 . 百度学术
    7. 朱威,马小明,张纲,周忠志,薛原,邓艾东. 基于GraphSage和自注意力机制的滚动轴承半监督故障诊断方法. 信息化研究. 2023(04): 48-54 . 百度学术

    其他类型引用(14)

  • 加载中
图(10) / 表(3)
计量
  • 文章访问数:  1425
  • HTML全文浏览量:  634
  • PDF下载量:  323
  • 被引次数: 21
出版历程
  • 收稿日期:  2022-03-21
  • 修回日期:  2022-06-22
  • 网络出版日期:  2022-06-25
  • 刊出日期:  2023-05-10

目录

/

返回文章
返回