基于注意力循环胶囊网络的滚动轴承故障诊断  

Fault Diagnosis of Rolling Bearing Based on Attention Recurrent Capsule Network

作　　者：瞿红春[1] 朱伟华 高鹏宇 王超[1] 周大鹏 丁凯 QU Hongchun;ZHU Weihua;GAO Pengyu;WANG Chao;ZHOU Dapeng;DING Kai(College of Aeronautical Engineering,Civil Aviation University of China Tianjin,300300,China)

机构地区：[1]中国民航大学航空工程学院,天津300300

出　　处：《振动．测试与诊断》2022年第6期1108-1114,1242,1243,共9页Journal of Vibration,Measurement & Diagnosis

基　　金：中国民航大学科研基金资助项目(05yk08m);中央高校基本科研业务费资助项目(ZXH2010D019)。

摘　　要：针对滚动轴承工作工况复杂、载荷大及测得的振动信号信噪比(signal-to-noise ratio,简称SNR)低的特点,提出了一种利用注意力循环机制(attention recurrent,简称AR)构建数字胶囊并与胶囊网络(capsule network,简称Caps)相融合的微弱故障诊断模型。首先,在构建初级胶囊时引入双向长短时记忆网络(bidirectional long short time memory neural network,简称Bi-LSTM),对时频图中的时序特征进行提取,并建立胶囊间的非线性关联;其次,引入注意力循环机制构建数字胶囊,提高时频图中不同时间和频带的能量强度变化的影响力;然后,通过3D卷积与动态路由机制构建的数字胶囊进行自适应融合,实现特征的多样提取;最后,利用softmax分类器将融合特征映射到输出层,实现高噪声环境下的滚动轴承故障诊断。结果表明,该方法对小样本、低信噪比的微弱故障信号较其他诊断模型有更高的诊断精度,并能够有效减小过拟合问题。使用不同负载下的数据做测试集验证了该模型有较强的泛化能力。In view of the complex working conditions,large load and low signal-to-noise ratio(SNR)of vibra⁃tion signal of rolling bearing,a weak fault diagnosis model based on attention recurrence(AR)is proposed to construct digital capsule and fuse with capsule network(Caps).In this model,the bidirectional long-short time memory neural network(Bi-LSTM)is introduced to extract the temporal features of the time-frequency dia⁃gram,and establish nonlinear association between capsules.Secondly,we use the attention recurrence mecha⁃nism to construct digital capsules to improve the influence of energy intensity changes in different time and fre⁃quency bands of time-frequency diagram.Then the attention recurrence and digital capsules constructed by dy⁃namic routing mechanism are fused by 3D convolution adaptively to realize the diversity of feature extraction.Finally,the softmax classifier is used to map the fusion features to the output layer,to achieve the fault diagnosis of rolling bearing in high noise environment.The results show that this method has higher diagnostic accuracy than other diagnostic models on weak fault signals with small samples and low signal-to-noise ratio.Moreover,the model can effectively reduce the over fitting problem and strong generalization ability.

关 键 词：智能故障诊断 胶囊网络 注意力机制 滚动轴承 

分 类 号：TH17[机械工程—机械制造及自动化] TH113.1