《计算机应用》唯一官方网站 ›› 2022, Vol. 42 ›› Issue (4): 1207-1215.DOI: 10.11772/j.issn.1001-9081.2021071243

• CCF第36届中国计算机应用大会 (CCF NCCA 2021) • 上一篇    

基于改进一维卷积和双向长短期记忆神经网络的故障诊断方法

董永峰1,2, 孙跃华1, 高立超1, 韩鹏3, 季海鹏2,4,5()   

  1. 1.河北工业大学 人工智能与数据科学学院,天津 300401
    2.河北省数据驱动工业智能工程研究中心(河北工业大学),天津 300401
    3.中信戴卡股份有限公司,河北 秦皇岛 066011
    4.天津开发区精诺瀚海数据科技有限公司,天津 300401
    5.河北工业大学 材料科学与工程学院,天津 300401
  • 收稿日期:2021-07-16 修回日期:2021-09-15 接受日期:2021-09-22 发布日期:2021-09-27 出版日期:2022-04-10
  • 通讯作者: 季海鹏
  • 作者简介:董永峰(1977—),男,河北定州人,教授,博士,CCF高级会员,主要研究方向:知识图谱、智能信息计算
    孙跃华(1996—),男,河北沧州人,硕士研究生,主要研究方向:大数据、智能计算、深度学习
    高立超(1997—),男,山东昌邑人,硕士研究生,主要研究方向:大数据、智能计算、知识图谱
    韩鹏(1989—),男,河北秦皇岛人,工程师,主要研究方向:表面处理、表面涂装、智能制造
  • 基金资助:
    河北省自然科学基金资助项目(F2019202062)

Fault diagnosis method based on improved one-dimensional convolutional and bidirectional long short-term memory neural networks

Yongfeng DONG1,2, Yuehua SUN1, Lichao GAO1, Peng HAN3, Haipeng JI2,4,5()   

  1. 1.School of Artificial Intelligence,Hebei University of Technology,Tianjin 300401,China
    2.Hebei Data Driven Industrial Intelligent Engineering Research Center (Hebei University of Technology),Tianjin 300401,China
    3.CITIC Dicastal Company Limited,Qinhuangdao Hebei 066011,China
    4.Tianjin Development Zone Jingnuo Data Technology Company Limited,Tianjin 300401,China
    5.School of Materials Science and Engineering,Hebei University of Technology,Tianjin 300401,China
  • Received:2021-07-16 Revised:2021-09-15 Accepted:2021-09-22 Online:2021-09-27 Published:2022-04-10
  • Contact: Haipeng JI
  • About author:DONG Yongfeng, born in 1977, Ph. D., professor. His research interests include knowledge graph, intelligent information processing.
    SUN Yuehua, born in 1996, M. S. candidate. His research interests include big data, intelligent computing, deep learning.
    GAO Lichao, born in 1997, M. S. candidate. His research interests include big data, intelligent computing, knowledge graph.
    HAN Peng, born in 1989, engineer. His research interests include surface treatment, surface coating, intelligent manufacturing.
  • Supported by:
    Hebei Natural Science Foundation(F2019202062)

摘要:

针对工业领域中故障诊断数据存在时序性和夹杂强噪声的特点导致的收敛速度慢以及诊断精度低的问题,提出了一种基于改进一维卷积和双向长短期记忆(1DCNN-BiLSTM)神经网络融合的故障诊断方法。该方法包括故障振动信号的预处理、特征的自动提取以及振动信号的分类。首先,采用自适应白噪声的完整经验模态分解(CEEMDAN)技术对原始振动信号进行预处理;其次,构建1DCNN-BiLSTM双通道模型,将处理后信号输入双向长短期记忆(BiLSTM)神经网络模型和一维卷积神经网络(1DCNN)模型两个通道,从而对信号的时序相关性特征、局部空间的非相关性特征和弱周期性规律进行充分提取;然后,针对信号夹杂强噪声的问题,对压缩与激励网络(SENet)模块进行改进并将其作用于两个不同的通道;最后,输入全连接层将双通道提取的特征进行融合并借助Softmax分类器实现对设备故障的精确识别。使用凯斯西储大学轴承数据集进行实验,结果表明改进后的SENet模块同时作用于1DCNN通道和stacked BiLSTM通道,1DCNN-BiLSTM双通道模型在保证快速收敛的情况下有最高诊断精度96.87%,优于传统单通道模型,有效提高了机械设备故障诊断效率。

关键词: 注意力机制, 一维卷积神经网络, 双向长短期记忆神经网络, 双通道, 故障诊断

Abstract:

Aiming at the problems of the slow model convergence and low diagnosis accuracy due to the time-series fault diagnosis data with strong noise in the industrial field, an improved one-Dimensional Convolutional and Bidirectional Long Short-Term Memory(1DCNN-BiLSTM) Neural Network fault diagnosis method was proposed. The method includes preprocessing of fault vibration signals, automatic feature extraction and vibration signal classification. Firstly, Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN) technology was used to preprocess the original vibration signal. Secondly, the 1DCNN-BiLSTM dual channel model was constructed, and the processed signal was input into the Bidirectional Long Short-Term Memory (BiLSTM) model channel and the One-dimensional Convolution Neural Network (1DCNN) model channel to fully extract the timing correlation characteristics, the non-correlation characteristics of the local space and the weak periodic laws of the signal. Thirdly, in response to the problem of strong noise in the signal, the Squeeze and Excitation Network (SENet) module was improved and applied to the two different channels. Finally, the features extracted from the two channels were fused by putting them into the fully connected layer, and the accurate identification of equipment faults was realized by the help of the Softmax classifier. The bearing dataset of Case Western Reserve University was used for experimental comparison and verification. The results show that after applying the improved SENet module to the 1DCNN channel and the stacked BiLSTM channel at the same time, the 1DCNN-BiLSTM dual channel model performs the highest diagnosis accuracy 96.87% with fast convergence, which is better than traditional one-channel models, thereby effectively improving the efficiency of equipment fault diagnosis.

Key words: attention mechanism, one-Dimensional Convolution Neural Network (1DCNN), Bidirectional Long Short-Term Memory (BiLSTM) Neural Network, dual channel, fault diagnosis

中图分类号: