《计算机应用》唯一官方网站

• •    下一篇

基于多路层次化混合专家模型的轴承故障诊断方法

徐欣然1,张绍兵2,成苗2,张洋3,曾尚4   

  1. 1. 中国科学院 成都计算机应用研究所
    2. 深圳市中钞科信金融科技有限公司
    3. 中科院成都计算机应用研究所
    4. 中国科学院大学
  • 收稿日期:2024-01-17 修回日期:2024-03-13 发布日期:2024-05-09 出版日期:2024-05-09
  • 通讯作者: 徐欣然

Bearings fault diagnosis method based on multi-pathed hierarchical mixture-of-experts

  • Received:2024-01-17 Revised:2024-03-13 Online:2024-05-09 Published:2024-05-09

摘要: 针对滚动轴承故障诊断中处理复杂工况准确率较低的问题,提出了一类多任务学习模型以及对应的层次化训练模式。该模型通过结合多阶段、多任务联合训练,实现了层次化的信息共享模式,在普通多任务学习模式的基础上进一步提升了模型的泛化性和故障识别准确率,使模型能够同时在复杂数据集与简单数据集上出色地完成任务。与此同时,结合了一维ResNet的瓶颈层(bottleneck)结构,保证了网络的深度的同时,也规避了梯度爆炸与梯度消失等问题,能够充分地提取数据集的相关特征。最后,以帕德博恩大学轴承故障数据集为测试数据集设计了实验。实验结果表明,在不同工况复杂度下,与不使用多任务学习的OMoE(One-gate Mixture-of-Experts)-ResNet模型相比,准确率提升达到了5.45个百分点到9.30个百分点,与集成经验模态分解的Hilbert谱方法(EEMD-Hilbert)、MMoE(Multi-gate Mixture-of-Experts)、多尺度多任务注意力卷积神经网络(MSTACNN)等模型相比,准确率提升区间在5.47个百分点到18.26个百分点。

Abstract: In response to the issue of low accuracy in handling complex work conditions in rolling bearing fault diagnosis, a multi-task learning model and corresponding hierarchical training mode is proposed. This model combines multi-stage, multi-task-united training to achieve a hierarchical information sharing mode. It further improves the model's generalization and fault recognition accuracy beyond the ordinary multi-task learning mode, enabling the model to perform excellently on both complex and simple datasets. Meanwhile, by incorporating the bottleneck layer structure of one-dimensional ResNet, the depth of the network is ensured while avoiding issues such as gradient explosion and vanishing, fully extracting relevant features of the dataset. Lastly, experiments are designed using the Paderborn University bearing fault dataset as the test dataset. The results demonstrate that under varying degrees of operational complexity, compared to the OMoE (One-gate Mixture-of-Experts)-ResNet model without multi-task learning, accuracy improvement ranges from 5.45 to 9.30 percentage points. Compared to Ensemble Empirical Mode Decomposition Hilbert spectral transform (EEMD-Hilbert), MMoE (Multi-gate Mixture-of-Experts), and Multi-Scale multi-Task Attention Convolutional Neural Network (MSTACNN), accuracy improvement ranges from 5.47 to 18.26 percentage points.

中图分类号: