基于MCNN-LSTM和交叉熵损失函数的轴承故障诊断

Bearing fault diagnosis based on MCNN-LSTM and cross entropy loss function

  • 摘要: 滚动轴承广泛应用于各类大型机械设备中,发挥着支撑机械旋转体、降低摩擦阻力、保证回转精度的重要作用。由于滚动轴承使用的广泛性和关键性,滚动轴承的实时状态分析对于保证机械设备的正常运行具有重要意义,目前基于振动信号分析的智能故障诊断方法在该领域被广泛使用。采用多尺度卷积神经网络和长短时记忆网络(multi-scale convolutional neural network and long shor-term memory, MCNN-LSTMMCNN-LSTM)模型进行轴承故障诊断。但在分类问题中,均方误差损失函数会发生梯度消失,导致模型初期训练学习速率较低。因此对其损失函数进行改进,采用交叉熵损失函数进行训练,可有效避免梯度消失问题,提高模型学习速率。实验采用凯斯西储大学(Case Western Reserve University, CWRU)公共数据集进行模型的训练与测试,实验结果说明采用改进的交叉熵损失函数不仅能够提高模型的学习速率,还实现了更高的预测准确率与F1分数。

     

    Abstract: Rolling bearings are widely used in all kinds of large machinery and equipment, which play an important role in supporting the mechanical rotating body, reducing friction resistance and ensuring rotation accuracy. Due to the universality and criticality of rolling bearings, real-time state analysis of rolling bearings is of great significance to ensure the normal operation of mechanical equipment. At present, intelligent fault diagnosis method based on vibration signal analysis is widely used in this field. MCNN-LSTM model is used for bearing fault diagnosis. However, in the classification problem, the use of mean square error loss function leads to gradient vanishing, resulting in a low learning rate in the initial training of the model. Therefore, improving its loss function and using cross entropy loss function for training can effectively avoid the problem of gradient vanishing and improve the learning rate of the model. CWRU public data set is used to train and test the model. The experimental results show that the improved cross entropy loss function can not only improve the learning rate of the model, but also achieve higher prediction accuracy and F1 score.

     

/

返回文章
返回