To address the problems of low recognition accuracy, slow convergence speed and weak generalization ability of traditional LeNet-5 network used in rolling-element bearing fault diagnosis, a rolling-element bearing fault diagnosis method using improved 2D LeNet-5 network is put forward. The following improvements to the traditional LeNet-5 network are made: the convolution and pooling layers are reasonably designed and the size and number of convolution kernels are carefully adjusted to improve fault classification capability; the batch normalization (BN) is adopted after each convolution layer to improve convergence speed; the dropout operation is performed after each full-connection layer except the last layer to enhance generalization ability. To further improve the efficiency and effectiveness of fault diagnosis, on the basis of improved 2D LeNet-5 network, an end-to-end rolling-element bearing fault diagnosis method based on the improved 1D LeNet-5 network is proposed, which can directly perform 1D convolution and pooling operations on raw vibration signals without any preprocessing. The results show that the improved 2D LeNet-5 network and improved 1D LeNet-5 network achieve a significant performance improvement than traditional LeNet-5 network, the improved 1D LeNet-5 network provides a higher fault diagnosis accuracy with a less training time in most cases, and the improved 2D LeNet-5 network performs better than improved 1D LeNet-5 network under small training samples and strong noise environment.Sensors 2020, 20, 1693 2 of 23 clustering [10], back propagation neural network (BPNN) [11], etc. The traditional rolling-element bearing fault diagnosis methods have been widely used, but with the increasing complexity of vibration signals, these methods have a certain limitation; however, the deep learning methods have a greater advantage in analyzing complicated and non-stationary vibration signals.The deep learning methods can automatically extract fault features from vibration signals [12], recently there are many researches are conducted on rolling-element bearing fault diagnosis using deep learning. Yin et al. [13] extracted the original features of vibration signals through time-domain analysis, frequency-domain analysis and wavelet transform, and obtained the low-dimensional features from 38 original features using the nonlinear global algorithm, and the low-dimensional features array is input into the deep belief network (DBN) to evaluate the performance status of rolling-element bearing. Liu et al. [14] obtained the spectrogram of vibration signals through STFT, used the stacked sparse auto-encoder (SAE) to automatically extract fault features, and employed the softmax regression to identify the fault type of rolling-element bearing. Liu et al. [15] used the recurrent neural network (RNN) to classify the faults of rolling-element bearing, and adopted the gated recurrent unit based denoising auto-encoder to enhance fault classification accuracy. Among different deep learning methods, compared with DBN, ...
To effectively predict the rolling bearing fault under different working conditions, a rolling bearing fault prediction method based on quantum particle swarm optimization (QPSO) backpropagation (BP) neural network and Dempster–Shafer evidence theory is proposed. First, the original vibration signals of rolling bearing are decomposed by three-layer wavelet packet, and the eigenvectors of different states of rolling bearing are constructed as input data of BP neural network. Second, the optimal number of hidden-layer nodes of BP neural network is automatically found by the dichotomy method to improve the efficiency of selecting the number of hidden-layer nodes. Third, the initial weights and thresholds of BP neural network are optimized by QPSO algorithm, which can improve the convergence speed and classification accuracy of BP neural network. Finally, the fault classification results of multiple QPSO-BP neural networks are fused by Dempster–Shafer evidence theory, and the final rolling bearing fault prediction model is obtained. The experiments demonstrate that different types of rolling bearing fault can be effectively and efficiently predicted under various working conditions.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.