Published: 08 September 2023

Research on fault diagnosis method of bearing based on parameter optimization VMD and improved DBN

Yingqian Sun1
Zhenzhen Jin2
1, 2Department of Marine Engineering, Guangxi Transport Vocational and Technical College, Nanning, 530023, China
Corresponding Author:
Yingqian Sun
Views 38
Reads 18
Downloads 86


Aiming at the problem that the bearing characteristics are difficult to extract accurately, and the fault diagnosis is difficult. This paper proposed a novel bearing fault diagnosis method with parameter optimization variational mode decomposition (VMD) and particle swarm optimization Deep Belief Networks (PSO-DBN). Firstly, the PSO is applied to optimize the parameter of the VMD and solve the problem of parameter setting of the VMD. Then, to effectively extract the feature information, using the optimized VMD, the original signal is decomposed into intrinsic mode components, and each component's dispersion entropy (DE) value is calculated. Finally, to further improve the accuracy of fault diagnosis, the PSO-DBN model is used to recognize the fault pattern bearing. The results of both experiments are 100 %. The results show that this method can effectively extract bearing fault features and accurately realize fault diagnosis. Compared with other methods, the accuracy of this method is increased by at least 2.08 % and the maximum is increased by 33.33 %.

Research on fault diagnosis method of bearing based on parameter optimization VMD and improved DBN


  • The PSO is applied to optimize the parameter of the VMD.
  • The combination of dispersion entropy and VMD is used to extract signal features.
  • Adaptive DBN is used to realize bearing fault classification.

1. Introduction

The bearing is widely used in large equipment such as hydro turbines, trains, cranes, etc., and is very important to ensure equipment safety. However, the bearing runs for a long time and is affected by the vibration of other components, which aggravates the occurrence of bearing faults. Effective bearing fault diagnosis can provide support for the stable operation of equipment and has a profound impact on improving equipment utilization and reducing maintenance costs.

Vibration signal analysis is widely used to diagnose bearing faults [1-2]. Traditional time-domain analysis methods cannot effectively diagnose nonlinear and non-stationary bearing vibration signal faults. At present, many modern time-frequency analysis methods have been widely applied in the field of fault diagnoses, such as wavelet transform (WT) [3], empirical mode decomposition (EMD) [4], ensemble empirical mode decomposition (EEMD) [5], empirical wavelet transform (EWT) [6] and variational mode decomposition (VMD) [7]. Zhou [8] used wavelet decomposition combined with the entropy method to diagnose bearing faults. The feature extraction effect was significant, laying the foundation for the subsequent bearing fault diagnosis. The wavelet transform method can deal with non-stationary signals and remove background noise, but the fixed basis function is unsuitable for multi-feature matching. Alexandre Trilla [9] applied the EMD method to diagnose the axle box of the bogie. Cai [10] proposed a rolling bearing failure method based on EMD decomposition and GNN-AdaBoost. Zhao [11] adopted a method based on combining EMD and kernel-related blind source extraction methods to solve the early fault diagnosis of bearings. The experiment proved that this method could extract fault features well under strong background noise. Wang [12] applied the adaptive EEMD algorithm to the denoising process of bearing vibration signals and achieved a good noise reduction effect. Li [13] proposed a combined method based on EEMD and WT analysis, which achieved good diagnostic results. Although EMD can effectively analyze and process bearing signals, it has some problems that affect signal processing accuracies, such as modal aliasing, end effects, and over-and under-envelope. EEMD can effectively suppress the modal aliasing of EMD, but the operation process is complex [14]. Cao [15] applied EWT to diagnose the signal of the bearing. The results showed that EWT performs well in detecting outer ring faults, roller faults, and composite faults of the outer ring and roller. Zhang [16] used the EWT to decompose signal and combine it with envelope analysis for fault diagnosis, which can effectively carry out fault detection. Compared with the EMD, this approach separates the modalities and provides more actionable results. However, EWT is limited by the Fourier spectrum segmentation and the selection of boundary detection methods [17]. For the problems mentioned above, Dragomiretskiy proposed the VMD algorithm. The method can simultaneously estimate the corresponding modes to perfectly balance the errors between them to obtain the separation of different components in the signal while avoiding the end effect and mode aliasing phenomenon. It has superior signal processing capability than EMD, EEMD, and EWT. Li [18] applied the VMD to extract the feature of the signal, and it turns out that VMD has a better feature extraction ability. Hang [19] used VMD and entropy theory to identify bearing faults. It is proved that VMD entropy is effective for bearing fault identification. However, the decomposition result of VMD is constrained by parameter settings. So how to determine VMD parameters is necessary to achieve efficient decomposition of VMD [20]. The research on parameter setting of VMD decomposition of bearing vibration signal has achieved certain achievements. Yan [21] estimated the number of modal components K by scale-space representation and the scale-space spectrum segmentation, optimizing the parameters of VMD. However, this approach is limited by experience. Therefore, the study of VMD parameter settings is of great significance in improving the feasibility of signal processing. The intelligent optimization algorithm provides a helpful idea for VMD parameter setting. The particle swarm optimization (PSO) algorithm has the advantages of easy implementation, high precision, and fast convergence. It is introduced into the VMD parameter setting in this paper.

To better extract fault feature information, information entropy is used for feature extraction. Sample entropy (SE) [22] and permutation entropy (PE) [23] have been applied to extract signal features. Qin [24] applied sample entropy to obtain feature information in bogie signals. Feng [25] obtained the eigenvectors of the bearing by PE calculation. Rostaghi [26] proposed a new method to calculate the complexity of time series data: dispersion entropy (DE). The DE can fully consider the difference between the signal amplitudes and has a faster technical speed. Compared with the other algorithms, it shows tremendous advantages in feature extraction.

The key to bearing fault diagnosis lies in the identification of fault mode. At present, support vector machine (SVM) is a widely used classification algorithm. Compared with random forest and k nearest neighbors (KNN), SVM has the advantages of accuracy and robustness, which have been proved in [27]. SVM has achieved certain results in the field of fault pattern recognition. Zhu [28] combined fuzzy entropy (FE) and SVM to identify bearing fault types. Liu [29] used SVM as the fault classifier of a train bogie braking system and obtained high classification accuracy. María Jesús Gómez [30] proved the effectiveness of SVM. However, the processing ability of SVM in high-dimensional and nonlinear data is limited. DBN is a deep learning method that can effectively reduce the interference of high-dimensional data on SVM and is widely used as a pattern recognition algorithm. He [31] effectively detects weak faults of rolling bearings using DBN. Zhong [5] used DBN as a pattern recognition algorithm in circuit intermittent fault diagnosis and achieved a high fault diagnosis rate. However, the classification effect of DBN is affected by the number of hidden layers and learning rate. Currently, most DBN algorithms determine the network structure by experience, and the classification effect is not guaranteed. Therefore, how to reasonably determine the parameters, such as the number of nodes in the hidden layer of DBN and the learning rate, is of great significance in improving the classification accuracy of DBN.

Based on the above research, when the VMD processes the bearing vibration signal, there is a problem in that the VMD parameter setting leads to a poor decomposition effect. During bearing fault classification, the recognition accuracy of DBN is affected by parameters. Therefore, combining VMD discrete entropy and PSO-DBN establishes a new method for bearing fault diagnosis. Firstly, PSO-VMD is used to improve the VMD decomposition effect. The optimized VMD is utilized to process the signal and compute the DE of the components to obtain the eigenvectors. Then, a PSO-DBN model is established to improve the accuracy of DBN recognition. Finally, use the PSO-DBN model to diagnose bearing fault types. Research shows that the method can accurately judge bearing faults and has certain advantages compared with similar existing methods. The contributions are as follows:

(1) A signal processing method of PSO-VMD is proposed to solve the VMD parameter optimization problem.

(2) Combining DE with VMD for signal feature extraction improves the effect of feature extraction.

(3) PSO-DBN is proposed to solve the problem of DBN parameter setting and improve the accuracy of fault diagnosis.

2. Methods

2.1. Variational model decomposition

VMD is a new adaptive signal decomposition method proposed by Dragomiretskiy, with strict mathematical theory support [27]. The essence of VMD is to achieve signal decomposition by constructing the variational equation and finding the optimal solution. VMD puts the original signal into the variational model and achieves the adaptive decomposition of the original signal by solving the variational model. It is an entirely non-recursive decomposition method that effectively avoids modal aliasing and the endpoint effect that occurs when EMD decomposes signals. VMD algorithm is developed based on Wiener filtering, Hilbert transform, analysis signal, and heterodyne demodulation [30]. VMD decomposes a complex signal into multiple analysis signals with their own central frequencies and sparse characteristics through Wiener filtering and Hilbert transform and calculates the marginal spectrum of each analysis signal. Then, by using the displacement property of Fourier transform, multiplying e-jωkt, the central frequencies of each mode are adjusted, and the spectrum of each mode is demodulated to the base frequency band, the broadband is estimated through the L2 paradigm. The final variational constraint model is:


where, f is the input signal, tis the time, δ is the Dirac distribution, uk=u1,u2,u3,,uk represents the K natural modal components IMF, and wk=w1,w2,,wk is the frequency center of each IMF.

In order to facilitate the solution and ensure the absolute integrability of each component, the quadratic penalty factor α and Lagrange multiplier factor λ(t) are introduced, and the extended Lagrange Eq. (2) is obtained. The variational constraint problem is transformed into a non-constrained problem:


where, α is the secondary penalty factor, and λ is the Lagrange operator. To obtain the saddle point of the Lagrange function through the alternating direction multiplier algorithm, that is, the optimal solution of the constrained variational model, the modal component uk and the center frequency wk.

VMD is a feasible signal processing method that can avoid end effects and modal aliasing. It is widely used in signal processing. For details, please refer to [18].

2.2. VMD parameter optimization

PSO is to find the optimal solution through cooperation and information sharing among individuals, which has the characteristics of easy realization, high precision, fast convergence, and global solid optimization ability. The VMD algorithm is used to decompose the signal of bearing. If the component is sensitive, the signal of the component presents a regular pulse, and the FE of the envelope signal is minor. If the decomposition effect is poor, the periodicity of the pulse is not prominent, and the FE of the envelope signal is also large. The envelope entropy is used as the fitness function, and PSO is used to optimize VMD parameters in this paper. When a set of parameters k is obtained, multiple envelope entropy values are obtained after VMD decomposition. The minimum is called the local minimum envelope value, which contains rich fault feature information. In this paper, the local minimum entropy is used as the objective function value of the optimization stage, and the minimum local minimum entropy is used as the optimization goal:


where, i=1,2,3,,N, Ee is the envelopment entropy, and αi is the envelopment signal obtained by Hilbert demodulation of signal xi.

2.3. PSO optimization DBN

DBN composes multiple restricted Boltzmann machines (RBM), and each layer of the RBM network is composed of multiple neurons, such as in Fig. 1. The layers are connected by weight. The super parameter setting of DBN is usually set according to experience, which restricts the classification accuracy of the algorithm. This paper takes the highest classification accuracy of the training set as the goal and uses PSO to optimize the algorithm to obtain DBN. The detailed steps are as follows.

Fig. 1DBN structure

DBN structure

Step 1. Initialize particle swarm. Initialize the position Xik=0 and velocity of particles Vik=0.

Step 2. Fitness value calculation. Calculate the fitness value of each particle to find the best particle in this round Xgbestk and the best particle in search history Xpbestk.

Step 3. Update the velocity and position of the particles. The velocity and position of the particles are updated using formulas XIk+1=Xik+Vik+1 and Vik+1=wVik+c1r1Xipbestk-Xik+c2r2Xigbestk-Xik. Where w is the inertia weight, the range is [0, 1], c1 and c2 are acceleration parameters, r1 and r2 are two random values that change in the range of [0, 1].

Step 4. If the misclassification rate of training samples meets the set conditions or reaches the set number of iterations, the PSO cycle ends. Otherwise, go to step 2, K=K+1, and repeat steps 3 and 4 until the conditions are met.

Step 5. Using the trained DBN network training test data, output bearing classification results.

2.4. Fault diagnosis method based on VMD dispersion entropy and PSO-DBN

A bearing fault diagnosis method based on VMD-DE and PSO-DBN is proposed to effectively judge the type of bearing failure, which has good fault diagnosis ability. The main steps are shown in Fig. 2:

Step 1: Build bearing fault simulation test bench, collect bearing vibration signal.

Step 2: PSO is used to optimize VMD parameters to obtain VMD parameters K,α.

Step 3: The original vibration fault data is decomposed into a series of IMFS by VMD.

Step 4: Calculate the dispersion entropy of IMFS, construct the feature matrix, and divide the test set and training set.

Step 5: The PSO is utilized to optimize the parameters of DBN to obtain the optimal penalty coefficient and kernel function. The PSO-DBN model is used for the fault diagnosis of bearing.

Fig. 2Fault diagnosis process based on VMD-DE and PSO-DBN

Fault diagnosis process based on VMD-DE and PSO-DBN

3. Experimental analysis

3.1. Case study 1

3.1.1. Data collection

We use the Windows 10 system, and the processor is Intel(R) Core (TM) i5-8250U CPU @ 1.60 GHz 1.80 GHz, running on the Matlab2020. The bearing fault diagnosis platform collects the bearing vibration signals. The experiment uses the bearing fault diagnosis platform shown in Fig. 3. Fig. 3 is a photo taken by experimenters in the laboratory, including the motor, coupling, bearings, gearbox, and sensors. The speed is set to 1200 r/min, and the sampling frequency is set to Fs= 10240 Hz. The bearing parameters are shown in Table 1. Moreover, the characteristic frequency of various fault types can be calculated by bearing parameters. Collect the signals of the four states of the bearing, each vibration type includes 24 samples, and each sample has 10240 sampling points. The vibration signals of the bearing collected under four test conditions are shown in Fig. 4. The signal amplitudes of the four states are small and have no high-frequency periodic characteristics. Therefore, it is impossible to discern bearing fault type through time-domain waveform diagram intuitively.

Fig. 3Bearing fault diagnosis platform

Bearing fault diagnosis platform

Fig. 4Time domain waveform

Time domain waveform

Table 1Bearing parameters

Bearing pitch diameter / mm
Roller diameter / mm
Number of rolling elements
Contact angle / (°)

3.1.2. Fault diagnosis

According to the fault diagnosis process shown in Fig. 2, the VMD parameters should be optimized first by using the PSO. The quantity of populations is set as 30, the quantity of iteration steps is 15, and the PSO is. Taking a set of vibration signals, for example, the iterative curve is shown in Fig. 5. The best result is obtained on the third time, indicating that the algorithm has fast convergence speed and strong global optimization ability, which is suitable for VMD parameter optimization. After the third iteration, the fitness value of 0.393 is the smallest and reaches optimal. It shows that the optimization algorithm has good performance, suited to the optimization VMD parameter, and the corresponding parameter is 8,2382.

Fig. 5PSO optimization of VMD parameter convergence curve

PSO optimization of VMD parameter convergence curve

The modal components with rich fault information are selected for envelope spectrum demodulation analysis to characterize the fault characteristics better. Peak index is a statistical index used to detect whether there is an impact in the signal; Kurtosis is an index to evaluate fault impact and is sensitive to impact signal. If the kurtosis index and kurtosis value are large, it is proved that the signal contains rich fault information. The components with a large kurtosis index and kurtosis value were selected as the best components in this paper. The parameter-optimized VMD is utilized to process the vibration signal, and each component’s peak index and kurtosis value are calculated. Table 2 shows the calculation results. The peak index and kurtosis value of IMF4 are the largest, so the IMF4 component is determined as the best component. Draw the envelope spectrum of the IMF4 component, as shown in Fig. 6. Table 1 shows the bearing roller fault characteristic frequency f= 41.38Hz. A clear peak line at fault characteristic frequency f1, the double f2, and the triple frequency f3 are in Fig. 5. Therefore, it can be concluded that there is local damage in the bearing roller, and the analysis results are consistent with the actual situation.

Table 2Peak index and kurtosis value of IMF component.

Peak index
Kurtosis value

To verify the feasibility of PSO- VMD, the traditional VMD method and EMD method are used to process the signal of the rolling element. Then the optimal component is selected to draw the envelope diagram for observation and comparison. The traditional parameter setting method of VMD is the center frequency method, as shown in Table 3. when K= 9, the central frequencies of IMF8 and IMF9 components are relatively close, which may lead to modal mixing. Therefore, K is selected as 9, and the penalty factor is set as 2000. Then decompose the signal and select the best component to draw the envelope diagram, as shown in Fig. 7. The envelope diagram of the optimal component after EMD decomposition is shown in Fig. 8. There is a clear peak spectrum at the characteristic frequency f1 of the rolling element fault. However, modal aliasing is at the double f2 and triple frequency f3. Through the above analysis, it can be known that the method of optimizing the VMD parameters using the PSO algorithm makes the eigenfrequencies of the optimal component envelope spectrum significant.

Table 3The center frequency of IMFS /Hz


Fig. 6Optimal component envelope diagram after VMD optimization

Optimal component envelope diagram after VMD optimization

Fig. 7Traditional VMD optimal component envelope graph

Traditional VMD optimal component envelope graph

In order to further judge the bearing failure, firstly, the parameters of the inner ring fault, outer ring fault, and normal signal are obtained by using PSO to optimize VMD. Then, the optimized VMD is applied to process the signal to obtain the IMF component and compute the DE of the components to construct the eigenvectors. The PSO is applied to optimize the DBN parameters with the goal of the minimum diagnostic error rate of the train set, the population size is set to 30, and the maximum quantity of iterations is 30. The optimal range of DBN parameters is [1, 100], and the obtained DBN parameters are [19 15 8]. A total of 96 groups of sample data, 48 groups were selected as training data, and 48 groups were selected as test data. Fig. 9 is the diagnosis results of the PSO-VMD-IDBN method under four different fault states. The accuracy of the method for bearing fault diagnosis reaches 100 %. Therefore, it can be proved that the method has high reliability and strong feasibility in judging bearing faults.

Fig. 8Envelope spectral value of EMD optimal component

Envelope spectral value of EMD optimal component

Fig. 9Fault diagnosis results

Fault diagnosis results

Eleven fault diagnosis methods are selected for comparison to prove the superiority of the method. Fig. 10 shows the fault identification accuracy. The fault diagnosis methods are divided into four groups. In the first group of fault diagnosis methods, EMD is used to process signals, and FE is utilized to extract fault feature information. SVM, PSO-SVM, and PSO-DBN proposed in this paper are used as the diagnosis method for comparison. The recognition rate of PSO-DBN is higher than PSO-SVM and SVM, but only 83.33 %. The second group of fault diagnosis methods takes the DE as the feature vector, and the vibration signal decomposition algorithm and diagnosis method are the same as the first group. The highest recognition rate of the EMD-DE-IDBN method is 89.58 %, which proves the effectiveness of the PSO -DBN. Compared with the first group of fault diagnosis results, the vibration signal decomposition method and fault diagnosis method are the same. The feature vector represented by the DE is better than that represented by the FE. In the third group of fault diagnosis methods, the optimized VMD method is applied as the signal decomposition method. The method for characterizing the feature vector and the pattern recognition method are identical to those in the first group. The highest diagnostic rate of the VMD-FE-IDBN method is 95.83 %, which proves the superiority of the PSO to optimize DBN. By comparing with the first group of fault diagnosis results, it is proved again that the PSO algorithm optimized VMD vibration signal decomposition method is better than the EMD, which is the same as the above analysis result. In the fourth group of fault diagnosis methods, the PSO algorithm is used to optimize VMD as the vibration signal decomposition algorithm. The dispersion entropy is used to characterize the fault characteristics. SVM, PSO-SVM, and PSO-DBN are applied for fault diagnosis. The VMD-DE-IDBN method proposed has the best diagnostic effect, and the diagnostic rate reaches 100 %. By comparing with the results of the third group of fault diagnoses, it is proved again that using the DE as the feature vector is superior to using the FE to characterize the feature vector. Compared with the results of the second group of fault diagnosis, the PSO-optimized VMD method proposed in this paper is proven effective in decomposing bearing vibration signals. From the above discussion, the parameter optimization VMD DE and PSO-DBN method for bearing fault diagnosis have significant advantages. Compared with other methods, they have obvious advantages in the single vibration signal decomposition method, feature vector representation method, pattern recognition method, and overall fault diagnosis method, which can reliably determine bearing faults. Therefore, the method proposed is suitable for bearing diagnosis.

Fig. 10Comparison of different diagnostic methods

Comparison of different diagnostic methods

3.2. Case study 2

In order to verify the generalization and robustness of the method, the bearing data collected by Case Western Reserve University are used for verification [32]. The experimental platform is shown in Fig. 11, including motor, torque sensor, and power tester. The drive end bearing used here, the model is SKF6205, the damage of the bearing is single point damage by EDM, and the damage diameter is 0.1778 mm. The state of the bearing is divided into normal, outer ring fault, inner ring fault, and rolling element fault. The sampling frequency is 12 KHz. Select 2048 sampling points as a sample, and each type has 48 samples.

PSO is applied to optimize the parameters of VMD, such as the rolling element fault. Fig. 12 shows the optimization curve. The optimal fitness value is 0.6562 at the 7th optimization, and the output VMD parameter is 2175,7. The algorithm has fast optimization speed and accuracy and can realize VMD parameter optimization. The conclusion is consistent with the previous conclusion, which shows high reliability. Optimized VMD for processing vibration signals, then the DE extracts the feature. A total of 96 groups of feature vectors are extracted. Forty-eight groups of them are selected as the training set, and the rest are selected as the test set. The PSO-DBN is applied as the diagnosis algorithm for fault pattern recognition. The same parameters are set as above, and the diagnosis results are obtained in Fig. 13. The fault diagnosis accuracy is 100 %, consistent with case 1. It shows that the proposed method has excellent generalization and robustness, which can provide theoretical guidance for bearing fault pattern recognition.

Fig. 11Case Western Reserve large bearing fault diagnosis test bench

Case Western Reserve large bearing fault diagnosis test bench

Fig. 12Iterative optimization graph is a figure

Iterative optimization graph is a figure

The data collected in the laboratory are often pure signals, and the actual operation of the signal usually contains a lot of noise, affecting the accuracy of fault diagnosis. In order to further verify the robustness of the method, this paper chooses to add Gaussian white noise with signal-to-noise ratio (SNR) of –2, –3, and –4 to the original signal, respectively. Taking normal data as an example, the time-domain waveform after adding noise is shown in Fig. 14. It can be seen from Fig. 14 that the signal without noise has the characteristics of periodic impact. With the increase of the intensity of Gaussian white noise, the periodic impact is gradually submerged. That is, the characteristics of the signal are gradually submerged, which brings great challenges to fault diagnosis. In order to verify the robustness of the proposed method in fault diagnosis under strong noise background, the characteristics of signals under different Gaussian white noises are extracted according to the process of fault diagnosis in Section 2.4, and the feature vectors are constructed. Finally, the optimized DBN is used for classification. The results are shown in Fig. 15. It can be seen from Fig. 15 that as the intensity of Gaussian white noise increases, the accuracy of fault diagnosis gradually decreases, indicating that Gaussian white noise has a great influence on fault characteristics. In fact, even when SNR = −4, the proposed method can still maintain a high accuracy of 87.5 %, indicating that the proposed method has good robustness.

Fig. 13Case Western Reserve University bearing fault diagnosis results

Case Western Reserve University bearing fault diagnosis results

Fig. 14Time domain waveforms with different SNR

Time domain waveforms with different SNR

Fig. 15Fault diagnosis results under different SNR

Fault diagnosis results under different SNR

4. Conclusions

Given the influence of the vibration of other parts and environmental noise on the bearing, the vibration signal is nonlinear and non-stationary, which makes it hard to obtain the fault characteristics. A parameter optimization VMD dispersion entropy and PSO-DBN-based bearing diagnosis method are presented. Firstly, with FE as the fitness value, the PSO algorithm is used to optimize the VMD parameters to obtain the parameter K,α. The optimized VMD is used for signal decomposition to obtain K modal components. Then, the dispersion entropy of K modal components is calculated as the feature vector; Finally, DBN is optimized by the PSO and used as a pattern recognition method for pattern recognition. The method can accurately identify the faults of different bearing states, and the recognition rate reaches 100 %. The contribution mainly includes the following aspects:

(1) A signal processing method of PSO-VMD is proposed, which solves the problem of the limited decomposition effect of VMD. Compared with the traditional VMD method and EMD method, this method can effectively overcome the problem of modal aliasing, and the signal processing effect is better.

(2) A feature extraction method based on DE is proposed, which improves the effect of feature extraction and has more significant advantages than FE and SE.

(3) A fault pattern recognition algorithm of PSO-DBN is proposed, which effectively improves the accuracy of bearing fault diagnosis.

(4) The proposed parameter optimization VMD discrete entropy and PSO-DBN bearing fault diagnosis method have high fault diagnosis rates. It can provide theoretical references for fault diagnosis of bearings and other rotating machinery.

Although the proposed method can effectively extract the characteristic frequency of the signal, when the background noise is enhanced, it may lead to feature submergence. Therefore, the study of signal noise reduction significantly improves fault diagnosis accuracy.


  • D. He, C. Liu, Z. Jin, R. Ma, Y. Chen, and S. Shan, “Fault diagnosis of flywheel bearing based on parameter optimization variational mode decomposition energy entropy and deep learning,” Energy, Vol. 239, p. 122108, Jan. 2022,
  • Z. Jin, D. He, R. Ma, X. Zou, Y. Chen, and S. Shan, “Fault diagnosis of train rotating parts based on multi-objective VMD optimization and ensemble learning,” Digital Signal Processing, Vol. 121, p. 103312, Mar. 2022,
  • M. Ge, J. Wang, Y. Xu, F. Zhang, K. Bai, and X. Ren, “Rolling bearing fault diagnosis based on EWT Sub-modal Hypothesis test and ambiguity correlation classification,” Symmetry, Vol. 10, No. 12, p. 730, Dec. 2018,
  • Z. Liu, K. Ding, H. Lin, G. He, C. Du, and Z. Chen, “A novel impact feature extraction method based on EMD and sparse decomposition for gear local fault diagnosis,” Machines, Vol. 10, No. 4, p. 242, Mar. 2022,
  • T. Zhong, J. Qu, X. Fang, H. Li, and Z. Wang, “The intermittent fault diagnosis of analog circuits based on EEMD-DBN,” Neurocomputing, Vol. 436, pp. 74–91, May 2021,
  • Y. Lu, R. Xie, and S. Y. Liang, “Detection of weak fault using sparse empirical wavelet transform for cyclic fault,” The International Journal of Advanced Manufacturing Technology, Vol. 99, No. 5-8, pp. 1195–1201, Nov. 2018,
  • Z. Jin, D. He, Z. Lao, Z. Wei, X. Yin, and W. Yang, “Early intelligent fault diagnosis of rotating machinery based on IWOA-VMD and DMKELM,” Nonlinear Dynamics, Vol. 111, No. 6, pp. 5287-5306, Mar. 2023,
  • Zhou P. T., San Q., and Ye Y. G., “Application of wavelet packet entropy and multi kernel learning in fault diagnosis of train bogie bearing,” Journal of Yanshan University, Vol. 41, pp. 401–406, 2017.
  • A. Trilla and P. Gratacòs, “Maintenance of bogie components through vibration inspection with intelligent wireless sensors: A case study on axle-boxes and wheel-sets using the empirical mode decomposition technique,” Proceedings of the Institution of Mechanical Engineers, Part F: Journal of Rail and Rapid Transit, Vol. 230, No. 5, pp. 1408–1414, 2016.
  • G. Cai, C. Yang, Y. Pan, and J. Lv, “EMD and GNN-AdaBoost fault diagnosis for urban rail train rolling bearings,” Discrete and Continuous Dynamical Systems – S, Vol. 12, No. 4-5, pp. 1471–1487, 2019,
  • X. Zhao, Y. Qin, C. He, and L. Jia, “Underdetermined blind source extraction of early vehicle bearing faults based on EMD and kernelized correlation maximization,” Journal of Intelligent Manufacturing, Vol. 33, No. 1, pp. 185–201, Jan. 2022,
  • C.-S. Wang, C.-Y. Sha, M. Su, and Y.-K. Hu, “An algorithm to remove noise from locomotive bearing vibration signal based on self-adaptive EEMD filter,” Journal of Central South University, Vol. 24, No. 2, pp. 478–488, Feb. 2017,
  • Li G. et al., “The fault diagnosis method of locomotive axle box bearing based on EEMD and wavelet packet,” Journal of Lanzhou Jiaotong University, Vol. 36, No. 4, pp. 1–5, 2017.
  • Wang Y. Q. et al., “Full-oscillatory components decomposition from noisy machining vibration signals by minimizing the Q-factor variation,” Transactions of the Institute of Measurement and Control, Vol. 39, pp. 1313–1328, 2017.
  • H. Cao, F. Fan, K. Zhou, and Z. He, “Wheel-bearing fault diagnosis of trains using empirical wavelet transform,” Measurement, Vol. 82, pp. 439–449, Mar. 2016,
  • Q. Zhang, J. Ding, and W. Zhao, “An adaptive boundary determination method for empirical wavelet transform and its application in wheelset-bearing fault detection in high-speed trains,” Measurement, Vol. 171, p. 108746, Feb. 2021,
  • Z. Jin, D. He, and Z. Wei, “Intelligent fault diagnosis of train axle box bearing based on parameter optimization VMD and improved DBN,” Engineering Applications of Artificial Intelligence, Vol. 110, p. 104713, Apr. 2022,
  • Z. Li, J. Chen, Y. Zi, and J. Pan, “Independence-oriented VMD to identify fault feature for wheel set bearing fault diagnosis of high speed locomotive,” Mechanical Systems and Signal Processing, Vol. 85, pp. 512–529, Feb. 2017,
  • Jin H., Lin J., and Chen X., “VMD entropy method and its application in early fault diagnosis of bearing,” in Proceedings of the 2018 International Conference on Signal Processing and Machine Learning, pp. 128–134, 2018.
  • T. Liu, Z. Luo, J. Huang, and S. Yan, “A comparative study of four kinds of adaptive decomposition algorithms and their applications,” Sensors, Vol. 18, No. 7, p. 2120, Jul. 2018,
  • Y. Huang, J. Lin, Z. Liu, and W. Wu, “A modified scale-space guiding variational mode decomposition for high-speed railway bearing fault diagnosis,” Journal of Sound and Vibration, Vol. 444, pp. 216–234, Mar. 2019,
  • Tzu-Kang Lin and Yu-Ching Chen, “Integration of refined composite multiscale cross-sample entropy and backpropagation neural networks for structural health monitoring,” Applied Sciences, Vol. 10, No. 3, p. 839, 2020.
  • X. Yan, M. Jia, and Z. Zhao, “A novel intelligent detection method for rolling bearing based on IVMD and instantaneous energy distribution-permutation entropy,” Measurement, Vol. 130, pp. 435–447, Dec. 2018,
  • Qinna et al., “Fault feature extraction of high-speed train bogie based on EEMD sample entropy,” Journal of Southwest Jiaotong University, Vol. 49, No. 1, pp. 27–32, 2015.
  • Feng Bo, “Rolling bearing fault diagnosis model of high-speed train based on SVD-PE,” Modular machine tool and automatic processing technology, Vol. 7, p. 108, 2018.
  • M. Rostaghi and H. Azami, “Dispersion entropy: A measure for time-series analysis,” IEEE Signal Processing Letters, Vol. 23, No. 5, pp. 610–614, 2016.
  • Y. Ni, Chen Liu, Erli Gao, and Wen-Shuai Song, “Performance evaluation method of spherical bearing based on correlation and sensitivity analysis and SVM,” Journal of Vibroengineering, Vol. 22, No. 8, pp. 1834–1846, 2020.
  • K. Zhu, L. Chen, and X. Hu, “Rolling element bearing fault diagnosis based on multi-scale global fuzzy entropy, multiple class feature selection and support vector machine,” Transactions of the Institute of Measurement and Control, Vol. 41, No. 14, pp. 4013–4022, 2019.
  • J. Liu, Y.-F. Li, and E. Zio, “A SVM framework for fault detection of the braking system in a high speed train,” Mechanical Systems and Signal Processing, Vol. 87, pp. 401–409, Mar. 2017,
  • M. J. Gómez, C. Castejón, E. Corral, and J. C. García-Prada, “Railway axle condition monitoring technique based on wavelet packet transform features and support vector machines,” Sensors, Vol. 20, No. 12, p. 3575, Jun. 2020,
  • X. He and J. Ma, “Weak fault diagnosis of rolling bearing based on FRFT and DBN,” Systems Science and Control Engineering, Vol. 8, No. 1, pp. 57–66, Jan. 2020,
  • Z.-F. Liu, L.-L. Li, M.-L. Tseng, R. R. Tan, and K. B. Aviso, “improving the reliability of photovoltaic and wind power storage systems using least squares support vector machine optimized by improved chicken swarm algorithm,” Applied Sciences, Vol. 9, No. 18, p. 3788, Sep. 2019,

About this article

17 June 2022
31 March 2023
08 September 2023
Fault diagnosis based on vibration signal analysis
feature extraction
dispersion entropy
deep belief networks

This research was funded Project of Improving the Basic Ability of Scientific Research of Young and Middle-aged Teachers in Guangxi Universities [Grant number 2022KY1134], Natural Science Foundation of Guangxi Transport Vocational and Technical College [Grant number JZY2020KAZ16] and Innovation Project of Guangxi Graduate Education [Grant number YCBZ2022043].

Data Availability

The datasets generated during and/or analyzed during the current study are available from the corresponding author on reasonable request.

Conflict of interest

The authors declare that they have no conflict of interest.