An Improved Fault Diagnosis Method for Rolling Bearings Based on 1D_CNN Considering Noise and Working Condition Interference

Huang,Zhu,Ren,Lin,Zeng,Wan,Zhu
DOI: https://doi.org/10.3390/machines12060383
IF: 2.899
2024-06-03
Machines
Abstract:Rolling bearings are prone to failure due to the complexity and serious operational environment of rotating equipment. Intelligent fault diagnosis based on convolutional neural networks (CNNs) has become an effective tool to ensure the reliable operation of rolling bearings. However, interference caused by environmental noise and variable working conditions can affect the data. To solve this problem, we propose an improved fault diagnosis method called deep convolutional neural network based on multi-scale features and mutual information (MMDCNN). In our approach, a multi-scale convolutional layer is placed at the front end of a 1D_CNN to maximize the retention of the multi-scale initial features. Meanwhile, the key fault features are further enhanced adaptively by introducing a self-attention mechanism. Then, the composite loss function is constructed by maximizing mutual information as an auxiliary loss based on cross-entropy loss; thus, the proposed method can extract robust fault features with high generalization performance. To demonstrate the superiority of MMDCNN, we compared the performance of our scheme with several existing deep learning models on two datasets. The results show that the proposed model successfully achieves bearing fault diagnosis with interference from noise and variable working conditions, possessing a powerful fault feature extraction capability.
engineering, mechanical, electrical & electronic
What problem does this paper attempt to address?
The problem that this paper attempts to solve is that in complex working environments and under severe interference conditions, rolling bearings are prone to failure. However, when facing environmental noise and variable - working - condition interference, the data quality of existing fault diagnosis methods is affected, resulting in a decline in diagnostic accuracy. To solve this problem, the author proposes an improved fault diagnosis method based on one - dimensional convolutional neural network (1D_CNN), named multi - scale features and mutual information - based deep convolutional neural network (MMDCNN). This method aims to enhance key fault features by introducing multi - scale convolutional layers and self - attention mechanisms, and improve the robustness and generalization ability of the model by constructing a composite loss function (combining cross - entropy loss and mutual information loss), so as to achieve more accurate fault diagnosis in uncertain interference environments.