Rolling Bearing Fault Diagnosis and Prediction Based on VMD-CWT and MobileNet

Jing Zhu,Aidong Deng,Shun Zhang,Xue Ding,Shuo Xue
DOI: https://doi.org/10.5121/csit.2021.110305
2021-01-01
Abstract:When deep learning is used for rolling bearing fault diagnosis, there are problems of high model complexity, time-consuming, and large memory. In order to solve this problem. This paper presents an intelligent diagnosis method of rolling bearings based on VMD-CWT feature extraction and MobileNet, VMD is used to extract the signal features, and then wavelet transform is used to extract the timefrequency features. After the image is enhanced, the MobileNet network is trained. In order to accelerate the convergence speed, this paper adds transfer learning in the network training process, and migrates the weights of the first several layers pretrained to the corresponding network. Experimental results based on bearing fault data sets show that after adopting VMD-CWT, the accuracy of mobilenet increased from 68.7% to 94%, and its network parameters were reduced by an order of magnitude compared with CNN.
What problem does this paper attempt to address?