A fault diagnosis method for rolling bearing based on gram matrix and multiscale convolutional neural network

Sci Rep. 2024 Dec 30;14(1):31902. doi: 10.1038/s41598-024-83315-5.

Abstract

The safety and reliability of rotating machinery hinge significantly on the proper functioning of rolling bearings. In the last few years, there have been significant advances in the algorithms for intelligent fault diagnosis of bearings. However, the vibration signals collected by machines are inevitably affected by irrelevant noise because of the complex working environments of bearings. So, an end-to-end bearing fault diagnosis method: GMSCNN, a bearing fault diagnosis method based on Gram Matrix (GM) and Multi scale Convolutional Neural Network (MSCNN), is proposed in this paper. In this method, first, GM is used to reduce the noise of the collected vibration signals; Secondly, MSCNN is used for feature extraction, and the characteristics of vibration signals at different frequencies and time scales can be captured by the convolutional kernels of different scales; thirdly, two feature enhancement branches are added, utilizing the undenoised vibration signal as input, to enrich and diversify features while enhancing the model's expressive and generalization capabilities; Finally, the experimental analysis was conducted on two bearing datasets to indicates that the noise robustness of GMSCNN is strong.

Keywords: Denoising; Fault diagnosis; Gram matrix; Multiscale convolutional neural network; Rolling bearing.