Deep Morphological Shrinkage Convolutional Autoencoder-Based Feature Learning of Vibration Signals for Gearbox Fault Diagnosis

Zhuang Ye,Shang Yue,Pu Yang,Ruixu Zhou,Jianbo Yu
DOI: https://doi.org/10.1109/tim.2024.3366570
IF: 5.6
2024-03-02
IEEE Transactions on Instrumentation and Measurement
Abstract:Fault diagnosis is significant to guarantee the safety and reliability of the machinery. Local faults will make the collected vibration signals deviate from the normal signals. To extract fault-related features from vibration signals, many deep-learning-based methods have been proposed for machinery fault diagnosis recently. However, due to the extreme conditions (e.g., high background noise and limited labeled training samples), it is a challenging task to implement feature extraction from the collected signals with nonlinear and nonstationary characteristics. To implement feature extraction and noise reduction of vibration signals, this article proposes a novel network, that is, deep morphological shrinkage convolutional autoencoder (DMSCAE) for gearbox fault diagnosis considering the insufficient labeled training samples. First, a morphological convolutional autoencoder is proposed for noise filtering and feature extraction. Second, a multibranch structure with different structural elements (SEs) is used in the morphological layer to extract impulsive components. Finally, a soft thresholding-based shrinkage is employed to filter ineffective features, where an adaptative method is developed to adjust the threshold automatically in the backpropagation procedure. The experiments on two gearbox fault diagnosis tests are conducted to verify the performance of DMSCAE. The results indicate that DMSCAE obtains a better performance for fault diagnosis than other DNNs, for example, efficient channel attention network (ECANet) and self-calibrated convolutional network (SCNet).
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?
The paper aims to address several key issues in gearbox fault diagnosis: 1. **Nonlinear and Non-stationary Signal Feature Extraction**: Extracting fault-related features from collected vibration signals becomes highly challenging due to the operation of mechanical equipment under extreme conditions (such as high background noise and limited labeled training samples). These vibration signals typically exhibit nonlinear and non-stationary characteristics. 2. **Feature Learning with Limited Labeled Samples**: In practical industrial applications, faulty components are replaced promptly once defects are detected, making it difficult to collect sufficient labeled samples. This results in poor generalization performance of existing deep learning models. 3. **Multi-scale Information Extraction**: Due to the dynamic nature of vibration signals, a single network branch cannot effectively extract multi-scale information. To address the above challenges, the paper proposes a novel deep learning network—Deep Morphological Shrinkage Convolutional Autoencoder (DMSCAE), with the main contributions including: 1. **Unsupervised Learning Network**: A morphological convolutional autoencoder is proposed for simultaneous denoising and feature learning of vibration signals without the need for labeled samples. This enables effective feature learning even with insufficient labeled samples. 2. **Multi-branch Morphological Layer**: A multi-branch morphological layer is constructed in DMSCAE, using different structural elements (SEs) to achieve multi-scale feature learning and denoising of non-stationary signals. 3. **Soft-thresholding-based Shrinkage Layer**: A soft-thresholding-based shrinkage method is introduced to highlight impulsive information. The soft threshold can be automatically updated according to the characteristics of the vibration signals. Experimental validation shows that DMSCAE performs excellently in gearbox fault diagnosis tasks, outperforming other deep neural networks (such as ECANet and SCNet).