Improved Graph-Regularized Deep Belief Network with Sparse Features Learning for Fault Diagnosis

Yang Jie,Bao Weimin,Li Xiaoping,Liu Yanming
DOI: https://doi.org/10.1007/s00521-022-06972-5
2022-01-01
Neural Computing and Applications
Abstract:Vibration signals are widely used in fault diagnosis of rotating machinery in real-world situations. However, it is very challenging to extract effective fault features from noisy signals and construct an accurate diagnosis model. In this paper, we propose a novel Gaussian–Bernoulli deep belief network (GDBN) model for intelligent fault diagnosis, where improved graph regularization and sparse features learning are embedded in the GDBN smoothly. In particular, the improved graph regularization is added to the hidden layer of original and reconstructed data. Therefore, our model can not only transform the original data into features with improved separability, but also generate discriminant features from vibration signal. An unsupervised pre-training learning process followed by a supervised fine-tuning is implemented in proposed model to contribute the classification capabilities. The effectiveness and superiority of the proposed model have been validated by gearbox and bearing cases studies. The results illustrate that our model can learn effective discriminative features and the extracted features are more separable. Furthermore, the proposed model achieves the better diagnosis accuracy in comparison with that of other models.
What problem does this paper attempt to address?