基于TVFEMD-CNN-CBAM-BiLSTM的滚动轴承故障诊断研究

Research on Fault Diagnosis of Rolling Bearing Based on TVFEMD-CNN-CBAM-BiLSTM

  • 摘要: 针对滚动轴承故障诊断中传统经验模态分解存在的模态混叠、端点效应以及复合故障场景下内外圈故障特征混淆问题,提出一种基于时变滤波经验模态分解-卷积神经网络-卷积注意力模块-双向长短期记忆网络(TVFEMD-CNN-CBAM-BiLSTM)的智能诊断框架。首先,选取TVFEMD分解生成的最优本征模态函数并计算其时域统计量的九类物理意义明确的指标,构建高判别性特征集作为输入;然后,利用CNN层提取空间局部模式,CBAM模块的通道-空间双路注意力机制增强故障敏感特征并抑制噪声,BiLSTM层建立故障特征的长期时序依赖关系,最后,经Softmax函数生成六类故障状态(健康、单一故障、复合故障)的概率分布,完成故障诊断。基于德国帕德博恩大学轴承数据对模型进行验证,结果表明,在单一工况和变工况下,TVFEMD-CNN-CBAM-BiLSTM模型故障诊断准确率均可达94%以上,且与传统CNN、BiLSTM以及CNN-BiLSTM模型相比较准确率更高,收敛速度更快,拟合情况更好。

     

    Abstract: Aiming at the problems of modal aliasing, endpoint effect and confusion of inner and outer ring fault features in traditional empirical mode decomposition in rolling bearing fault diagnosis, an intelligent diagnosis framework based on time-varying filtering empirical mode decomposition-convolutional neural network-convolutional attention module-bidirectional long short-term memory network (TVFEMD-CNN-CBAM-BiLSTM) is proposed. Firstly, the optimal intrinsic mode function generated by TVFEMD decomposition is selected and nine kinds of indexes with clear physical meaning of time domain statistics are calculated to construct a high discriminant feature set as input. Then, the CNN layer is used to extract the spatial local pattern. The channel-space dual attention mechanism of the CBAM module enhances the fault sensitive features and suppresses the noise. The BiLSTM layer establishes the long-term temporal dependence of the fault features. Finally, the Softmax function is used to generate the probability distribution of six types of fault states (health, single fault, composite fault) to complete the fault diagnosis. Based on the bearing data of Paderborn University in Germany, the model is verified. The results show that the fault diagnosis accuracy of TVFEMD-CNN-CBAM-BiLSTM model can reach more than 94 % under single working condition and variable working condition. Compared with the traditional CNN, BiLSTM and CNN-BiLSTM models, the accuracy is higher, the convergence speed is faster, and the fitting situation is better.

     

/

返回文章
返回