基于DRSN-CW和LSTM的轴承故障诊断

Bearing Fault Diagnosis Based on DRSN-CW and LSTM

  • 摘要: 利用深度残差网络中逐通道不同阈值的残差收缩模块(DRSN-CW)的降噪能力和特征提取能力,结合长短时记忆网络(LSTM)和注意力机制,设计了一个端到端的基于振动信号的轴承故障诊断模型DRSNCW-LSTM。其中,LSTM模块很好地利用了信号的时序特点,充分提取振动信号的内部时域特征。同时,注意力机制的引入可以使得模型自动提取出重要的时域特征用于后续的故障类型识别。在凯斯西储大学(CWRU)数据集上对提出的模型进行了测试,实验表明提出的方法在无降噪处理的情况下,相比于最新的MCNN-LSTM模型能更准确地诊断轴承故障。在训练数据不足的情况下,提出的方法依旧能较好地实现轴承故障诊断,平均准确率能达到98.16%,比MCNN-LSTM平均提升了2.62%。

     

    Abstract: In this paper, based on the noise reduction capability and feature extraction capability of deep residual shrinkage networks with channel-wise thresholds (DRSNCW) module, combined with long short-term memory network (LSTM) and attention mechanism module, an end-to-end vibration signal-based bearing fault diagnosis model, named DRSNCW-LSTM, is proposed. In this model, LSTM module makes good use of the time-series characteristics of the signal to sufficiently extract the internal time-domain features of the vibration signal. Moreover, the introduction of attention mechanism enables the model to automatically extract important time-domain features for follow-up fault type recognition. The effectiveness of the proposed model is validated on the authoritative case Western Reserve University (CWRU) dataset, and experiments show that the proposed method can achieve more accurate bearing fault diagnosis than the state-of-the-art multi-scale convolutional neural network-LSTM (MCNN-LSTM) model without noise reduction processing. In particular, in the case of insufficient training data, the proposed method still achieves rather good bearing fault diagnosis with an average accuracy of 98.16%, which is an average improvement of 2.62% over the MCNN-LSTM.

     

/

返回文章
返回