Improved Mobile-Vit model and its application in rolling bearing fault diagnosis
Huina Mu, Xiaoyun Zeng, Wei Liu, Jialin Sun, Yeshu Zhang · 2024
In order to improve the efficiency and accuracy of rolling bearing fault diagnosis, FDMT fault diagnosis model was proposed. Firstly, based on the Mobile-Vit model, a FDMT model for fault diagnosis with one-dimensional rolling bearing vibration signal as direct input was designed, and the structure of the FDMT model was given. Secondly, the FDMT model was trained and tested using the CWRU data set to verify the effectiveness of the model. Then, the influence of hyperparameter adjustment on the performance of FDMT model was discussed. Finally, the FDMT model was compared with other deep learning models. The results show that the improved Mobile-Vit model has higher fault diagnosis accuracy.