An enhanced selective ensemble deep learning method for rolling bearing fault diagnosis with beetle antennae search algorithm

Rolling bearing fault diagnosis is a meaningful yet challengeable task. To improve the performance of rolling bearing fault diagnosis, this paper proposes an enhanced selective ensemble deep learning method with beetle antennae search (BAS) algorithm. Firstly, multiple deep base models are constructed to automatically capture sensitive features from raw vibration signals. Secondly, to ensure the diversity of the base models, sparse autoencoder, denoising autoencoder and linear decoder are used to construct different deep autoencoders, respectively, and also Bootstrap is used to design distinctive training data subsets for each base model. Thirdly, an enhanced weighted voting (EWV) combination strategy with class-specific thresholds is proposed to implement selective ensemble learning. Finally, BAS algorithm is used to adaptively select the optimal class-specific thresholds. Experimental bearing data are used to verify the effectiveness of the proposed method. The results suggest that the proposed method can more accurately and robustly recognize different kind of faults than both the individual base models and other existing ensemble learning methods.