RFR-GA-BLS: A Feature Selection and Parameter Optimization Method for Fault Diagnosis of Rolling Bearing Using Infrared Images

Abstract

To overcome the problems of low machine learning fault diagnosis rate and long consumption time of deep learning in rolling bearing fault diagnosis, an RFR-GA-BLS model is proposed. The model is validated by infrared images of rolling bearings to find the most representative features, the most suitable parameters and the best diagnostic rate. Based on the pre-processed infrared thermal images of the faulty bearing, 72 second-order statistical features were obtained as information for fault diagnosis. RFR considered the robustness of the features, and new sequences were obtained. BLS was optimized by GA for fault diagnosis. New sequence features were added to the model sequentially, one at a time. After satisfying the model conditions, the most appropriate number of features was selected as the first 20. The search results for the number of feature nodes, the number of feature node windows and the number of enhancement nodes for the BLS were 24, 19 and 544, respectively, and the fault diagnosis rate of 98.8889% was achieved. According to a comparison with CFR-GA-BLS, BLS, PSO-BLS and Grdy-BLS, our proposed model is more advantageous in the search for the best performance. The fault diagnosis accuracy is higher compared to SVM and RF. The speed of our proposed model is 207 times faster than 1DCNN and 10,147 times faster than 2DCNN.