基于DR-VAE与自注意力机制的电机轴承故障诊断  

Fault Diagnosis for Motor Bearings Based on DR-VAE and Self-Attention Mechanism

作　　者：杨青 刘彦俏 吴东升 崔宝才 YANG Qing;LIU Yanqiao;WU Dongsheng;CUI Baocai(School of Automation and Electrical Engineering,Shenyang Ligong University,Shenyang 110159,China)

机构地区：[1]沈阳理工大学自动化与电气工程学院,沈阳110159

出　　处：《轴承》2023年第9期72-79,共8页Bearing

基　　金：辽宁省教育厅科学研究计划资助项目(LG201917);沈阳理工大学高水平成果建设计划资助项目(SYLUXM202109)。

摘　　要：为提高电机轴承故障诊断的精度,提出一种深度残差变分自编码器与自注意力机制相结合(DR-VAE-SAM)的电机轴承故障诊断方法。首先,在数据预处理中使用矢量量化变分自编码(VQ-VAE)对轴承故障数据进行增强,训练完成后将生成的故障样本混入原始样本中以平衡和增强数据集;然后,将深度残差网络与变分自编码器结合,增加了迁移学习中最大均值差异(MMD)作为融合标准;最后,在编、解码过程中引入自注意力机制提取特征的关键信息。通过西储大学和东南大学的轴承数据进行试验验证,结果表明,在故障数据少、样本不平衡以及变工况情况下,DR-VAE-AM模型能够有效提高故障诊断精度,并有较好的泛化能力。In order to improve the accuracy of fault diagnosis for motor bearings,an fault diagnosis method is proposed based on deep resnet-variational automatic encoder and self-attention mechanism(DR-VAE-SAM).Firstly,the vector quantised variational autoencoder(VQ-VAE)is used to enhance the bearing fault data during data preprocessing,and the generated fault samples are mixed into original samples to balance and enhance the data set after training.Then,the deep resnet is combined with variational self-encoder,and maximum mean discrepancy(MMD)in transfer learning is added as fusion criterion.Finally,the self-attention mechanism is introduced to extract the key information of features during process of encoding and decoding.The test verification is carried out by bearing data from Western Reserve University and Southeast University,the results show that DR-VAE-SAM method can effectively improve the accuracy of fault diagnosis and has good generalization ability under few fault data,unbalanced samples and variable working conditions.

关 键 词：滚动轴承 故障诊断 深度学习 自注意力机制 变工况 特征融合 

分 类 号：TH133.33[机械工程—机械制造及自动化] TP183[自动化与计算机技术—控制理论与控制工程]