陈翔, 刘勤明, 胡家瑞. 多源传感器数据下基于注意力机制与长短期记忆网络的轴承故障诊断与寿命预测[J]. 信息与控制, 2024, 53(2): 211-225. DOI: 10.13976/j.cnki.xk.2023.3056
引用本文: 陈翔, 刘勤明, 胡家瑞. 多源传感器数据下基于注意力机制与长短期记忆网络的轴承故障诊断与寿命预测[J]. 信息与控制, 2024, 53(2): 211-225. DOI: 10.13976/j.cnki.xk.2023.3056
CHEN Xiang, LIU Qinming, HU Jiarui. Bearing Fault Diagnosis and Life Prediction Based on Attention Mechanism and Long Short-term Memory Network under Multi-source Sensor Data[J]. INFORMATION AND CONTROL, 2024, 53(2): 211-225. DOI: 10.13976/j.cnki.xk.2023.3056
Citation: CHEN Xiang, LIU Qinming, HU Jiarui. Bearing Fault Diagnosis and Life Prediction Based on Attention Mechanism and Long Short-term Memory Network under Multi-source Sensor Data[J]. INFORMATION AND CONTROL, 2024, 53(2): 211-225. DOI: 10.13976/j.cnki.xk.2023.3056

多源传感器数据下基于注意力机制与长短期记忆网络的轴承故障诊断与寿命预测

Bearing Fault Diagnosis and Life Prediction Based on Attention Mechanism and Long Short-term Memory Network under Multi-source Sensor Data

  • 摘要: 针对滚动轴承故障在噪声环境下诊断效果不佳的问题, 提出了一种多源传感器数据下基于注意力机制与长短期记忆(long short-term memory, LSTM)网络的轴承故障诊断新方法。首先, 将多源传感器采集的1维数据进行归一化处理, 通过构建带AdaBN(Adaptive Batch Normalization)技术的双通道孪生卷积网络提取有效特征并进行数据融合。其次, 将融合数据输入具有同时考虑通道间关系和位置信息功能的改进1维CoordAtt(Coordinate Attention)中。再次, 通过LSTM层提取时间特征, 通过标签平滑正则化改进后的损失函数来评估诊断效果, 使用新型优化器Adan进行优化。最后, 将得到的诊断模型应用于测试集, 输出故障类别诊断结果。将模型在不同测试集比例下进行诊断精度对比实验, 判断出最佳比例为0.3, 并在噪声环境下进行测试。实验结果表明, 所提方法能更好地对抗噪声环境的影响。在C-MAPSS数据集上的实验结果验证了CoordAtt-LSTM模型在寿命预测中的有效性。

     

    Abstract: To solve the problem of the poor diagnosis effect of rolling bearing faults in noisy environments, we propose a new bearing fault diagnosis method based on an attention mechanism and a long short-term memory (LSTM) network using multisource sensor data. First, we normalize the one-dimensional (1D) data collected by multisource sensors and then construct a double-channel twin convolutional network with adaptive batch normalization technology to extract effective features and perform data fusion. Second, the fusion data are input into the improved 1D coordinate attention (CoordAtt) which is capable of considering the relationship between channels and position information simultaneously. Third, we use the LSTM layer to extract time features and evaluate the diagnosis effect by the loss function after label smoothing regularization. We use the new optimizer Adan for optimization. Finally, the diagnosis model is applied to a test set, and the diagnosis results of fault categories are output. We compare the diagnostic accuracy of the model under different test set ratios, determine that the optimal ratio is 0.KG-*53, and test it in a noisy environment. The experimental results show that the proposed method can better resist the influence of noisy environments. The validity of the CoordAtt-LSTM model in life prediction is verified by experimental results on the C-MAPSS dataset.

     

/

返回文章
返回