Shock and Vibration

Shock and Vibration / 2016 / Article

Research Article | Open Access

Volume 2016 |Article ID 9372691 | 10 pages |

A Fault Diagnosis Scheme for Rolling Bearing Based on Particle Swarm Optimization in Variational Mode Decomposition

Academic Editor: Athanasios Chasalevris
Received20 Jan 2016
Revised15 May 2016
Accepted23 May 2016
Published15 Jun 2016


Variational mode decomposition (VMD) is a new method of signal adaptive decomposition. In the VMD framework, the vibration signal is decomposed into multiple mode components by Wiener filtering in Fourier domain, and the center frequency of each mode component is updated as the center of gravity of the mode’s power spectrum. Therefore, each decomposed mode is compact around a center pulsation and has a limited bandwidth. In view of the situation that the penalty parameter and the number of components affect the decomposition effect in VMD algorithm, a novel method of fault feature extraction based on the combination of VMD and particle swarm optimization (PSO) algorithm is proposed. In this paper, the numerical simulation and the measured fault signals of the rolling bearing experiment system are analyzed by the proposed method. The results indicate that the proposed method is much more robust to sampling and noise. Additionally, the proposed method has an advantage over the EMD in complicated signal decomposition and can be utilized as a potential method in extracting the faint fault information of rolling bearings compared with the common method of envelope spectrum analysis.

1. Introduction

Rolling bearing is one of the most commonly used parts in rotating machinery to support rotating shafts. Due to the fact that its health state is directly related to the safety and a stable operation of the machine, the research of rolling bearing fault diagnosis has a great significance in actual application [1]. However, the early incipient fault feature is very faint and interfered by the strong background noise [2]. When the rolling bearing is in failure due to the influence of load, friction, noise, and other factors, the measured vibration signal is a multicomponent amplitude-modulated-frequency-modulated (AM-FM) signal using system’s natural frequency as carrier frequency and the fault characteristic frequency as modulation frequency, respectively. Thus, obtaining the AM-FM signal by the original signal decomposition and reducing the effect of noise are the emphasized research content in extracting the early faint fault feature of the rolling bearing.

Currently, there are many methods used to fault diagnosis for rolling bearings, but these methods have some inherent limitations. For instance, wavelet transform (WT) [3, 4] analysis is not adaptive, which is restricted by the selection of wavelet basis function and the number of decomposition levels; empirical mode decomposition (EMD) is lack of theoretical basis and there are some problems in its own algorithm such as the phenomenon of model mixing and the end effect [57], which has been used to detect the bearing fault [8]. Although the ensemble empirical mode decomposition (EEMD) [9] has some improvement in solving the problem of model mixing, the actual effect is sensitive to the strong background noise. Empirical wavelet transform (EWT) is introduced by Gilles et al. aimed at extracting a series of AM-FM signal from the original signal, which largely depends on choosing the boundaries of Fourier spectrum appropriately [10, 11]. The sparse decomposition theory [12] and manifold learning method [13] are proposed recently. However, the sparse decomposition relies on the design of the redundant atom library and decomposition algorithm, which has an obvious problem of the large amount of calculation. The manifold learning result is also restricted by the parameter selection of delay time and embedded dimension. From another aspect, the spectroscopy and ferrography of used grease are applied to condition monitoring for rolling element bearing [14], which has a limited capacity in detecting the noniron particle in lubricating oil and largely depends on the human experience.

Lately, Dragomiretskiy and Zosso proposed a new variational mode decomposition (VMD) [15] method, which was a new advanced multiresolution technique. A series of iterative updating process was used to minimize the constrained variational model; therefore, the vibration signal was decomposed into various modes or intrinsic mode functions (IMFs) using calculus of variation. It overcame the disadvantage of lacking theoretical basis and noise sensitivity of EMD. Moreover, the VMD method could adaptively determine the relevant frequency band and estimated the corresponding model. Based on the above-mentioned advantage, it was applied to the early fault diagnosis of rolling bearing [16, 17] and economics field [18]. From the theory of VMD algorithm, it can be known that the decomposed result of VMD is restricted by the penalty parameter selection and the number of components. In order to extract the faint fault information of the bearing vibration signal effectively, the parameters’ selection needs to be optimized. Taking into account the characteristics of fast convergence speed, small setting parameters and easy to implement particle swarm optimization (PSO) algorithm [19, 20], the paper applies it into the parameters optimization of VMD. The optimized two parameters are obtained by the PSO algorithm, which is used to the process of feature extraction for faint fault signal. After the original signal is decomposed by improved VMD method, the optimal component can be identified by the principle of maximum correlated kurtosis [21]. The index of correlated kurtosis has integrated the characteristics of the kurtosis and correlation function, which can represent the changes of shock signal and avoid the problem of overfitting. The merits of the proposed method can be summarized that the parameter selection and decomposition result of VMD algorithm are less affected by human experience. Using this method to analyze the simulated signal and the measured vibration signal from rolling bearing experiment system, the results indicate that the proposed method can accurately extract the early characteristic frequency of faint fault signal of the rolling bearing. The flowchart about the proposed method is shown in Figure 1.

The rest of the paper is organized as follows. In Section 2, the basic ideas of VMD and PSO are introduced. The simulation signal analysis is described in Section 3. The measured signal in rolling bearing experiment system was analyzed in Section 4. The final conclusions are given in Section 5.

2. Theories

2.1. Variational Mode Decomposition (VMD) Algorithm

Variational mode decomposition (VMD) is a new method of signal decomposition based on Wiener filtering, one-dimensional Hilbert transform, and heterodyne demodulation proposed lately by Konstantin Dragomiretskiy. Different from EMD, it defines the mode component as amplitude-modulated-frequency-modulated (AM-FM) signals as follows:where is a nondecreasing function; thus ; the envelope is nonnegative . Additionally, the change of envelope and instantaneous frequency are much slower than . Therefore, the mode component can be regarded as a pure harmonic signal with amplitude and instantaneous frequency .

The input signal is decomposed by VMD method into the mode component of a specified scale according to the subjective setting scale assuming that each mode is a finite bandwidth signal with a pulse as the center. In order to evaluate the bandwidth of each mode and construct the constraint model of the variational problem, VMD firstly uses the Hilbert transform to obtain the single spectrum of each mode and then transfers them to the fundamental frequency by exponential correction. The bandwidth of each mode is obtained through Gauss smooth demodulation signal finally, which is called as norm squared of the gradient. Thereby, the constrained variational problem is given as follows: where is the original signal, is the Dirac distribution, is number of modes, denotes each mode function, indicates each center frequency, represents the sum of all mode function, and denotes convolution. Due to the difficulty of solving the constrained problem, the penalty parameter and the Lagrange multiplication operator are introduced to convert the above constrained problem to the nonconstraint problem. Thereby obtaining a new solution expression:

Therefore, the Lagrange multipliers, modal functions, and their corresponding central frequency are iteratively updated by using alternating direction multiplier method (ADMM) to obtain the saddle point in the expression. Specific algorithm of classical VMD is given as the flowing expression.

Step 1. Initialize .

Step 2. The value of , , and is updated according to the following formula:

Step 3. Repeat the iterative process of (2) until the function converges, which is to satisfy the condition of , where is a given accuracy requirement.

2.2. Particle Swarm Optimization (PSO) Algorithm

Particle swarm optimization (PSO) is an intelligent algorithm to imitate birds’ foraging behavior proposed by Kennedy, through referring to the characteristics of all the individual bird in the process of feeding, which is widely used in solving nonlinear problems.

Each particle in the PSO algorithm is used as the solution of the optimization problem, which has a position and the corresponding speed determined by the optimization function. The algorithm evaluates the pros and cons of all particles by setting the appropriate fitness function. In each iteration, the particles constantly update their speed and position according to the fitness value of individual and the group. The updated particles continue to search the optimal value in the search space.

Specific configuration steps of PSO are as follows.

Step 1. Establish the appropriate fitness function according to the actual problem. The iteration calculation is carried out by setting the number of iterations, population number, the initial position, and velocity of the particles.

Step 2. Calculate the optimal values and , respectively, compared with the optimal solution of the current population and retain the better results, where represents local extremum of individual particles and indicates global extremum of group particles.

Step 3. Update the position and speed of all particles in the population according to the formula , , where is the inertia weight, , are the learning factor, is the random number between 0~1, and , are the speed and position of the particle in the th iteration of the dimension.

Step 4. Repeat Steps and until meeting the maximum number of iterations.

2.3. Variational Mode Decomposition Based on Particle Swarm Optimization Algorithm

In the traditional algorithm of variational mode decomposition, the user needs to set the penalty parameter and the number of the components before processing the signal because of the theory limitation. From the theoretical study of VMD, it can be known that the larger penalty parameter indicates the smaller bandwidth of each component decomposed by source signal and vice versa. Similarly, inappropriate setting number of components will also result in some unacceptable mode compositions. Therefore, selecting the appropriate parameter group of the component number and the penalty parameter is the key to accurately extract the fault information.

PSO algorithm is a widely used intelligent optimization algorithm compared with other optimization algorithms such as genetic algorithm and artificial fish algorithm. It is suitable for the optimal selection of parameters in consideration of its simple principle and mechanism, fast convergence speed, and, meanwhile, the good performance of global search. The key part of the PSO algorithm based on the variable mode decomposition is the selection of fitness function. Because of the incorrect settings of the penalty parameter and the number of components, some artifact components will generate, which are independent with the source signal. It is acceptable that the artifact components have less similarity with the source signal. Therefore, the cross-correlation coefficient between the decomposed mode component and the original signal is regarded as an evaluation index, which is defined in the following formula:where , represent the original signal and the mode component, respectively; is the data length; represents the cross-correlation coefficient.

From the above analysis, it can be seen that the cross-correlation coefficient may fluctuate under the condition of different parameters selection. The largest mean value of cross-correlation coefficient does not imply the best result of mode decomposition. The globally optimal value is achieved by considering the mean value and the variance of cross-correlation coefficient. The smaller value of variance indicates the less deviation from the mean value. Therefore, the optimal penalty parameters and the number of components can be well obtained by regarding the maximum ratio between the mean value of and the variance as fitness function. The detailed fitness function is expressed in the following formula:

On the basis of the above theory analysis, VMD based on PSO algorithm is applied to the analysis of simulated signal and the fault feature extraction of rolling bearing experiment system to verify the validity of the method in fault diagnosis.

3. The Analysis of Simulated Signal

The measured rolling bearing vibration signal is always consisted of the amplitude-modulated-frequency-modulated (AM-FM) signals, harmonic signal, and noisy signal in actual application. In order to verify the validity of the VMD based on PSO, the fault signal model is built by the following simulated signal:and then where the frequencies of , , , and are chosen as 90 Hz, 150 Hz, 500 Hz, and 270 Hz, respectively. The synthetic signal is composed by sinusoidal signal , cosine signal , and frequency-modulated (FM) signal . The sampling frequency is set as 1000 Hz and the sampling point is 1000. The time-domain graph of simulated signal is shown in Figure 2.

The VMD based on particle swarm optimization algorithm is applied to decompose the above simulated signal. The number of iterations and the particles is 20, the inertia weight linear decrease in the iterative process of the initial value is 0.9, and the final value is 0.4. The penalty parameter and the number of components optimized and selected by particle swarm optimization (PSO) search algorithm are a collection of (2064, 3), and the results of the decomposition are shown in Figure 3.

In the three decomposed components shown in Figures 3(b), 3(c), and 3(d), the blue line is on behalf of the original signal and the red line represents the signal after the decomposition. In order to more directly express the result of decomposition, the similarity analysis is carried out between the decomposed component and the original signal by the index of cross-correlation coefficient. Table 1 demonstrates that the proposed method has a perfect ability of complicated signal decomposition.

Sinusoidal signalCosine signalFM signal

Cross-correlation coefficient0.99620.99850.9974

Empirical mode decomposition (EMD) is used to decompose the above simulated signals, and the results of EMD decomposition are shown in Figure 4. There are eight IMFs decomposed by EMD.

Similarly, the decomposition result also carried out similarity analysis with the original component. Since the similarity of the three signal components (sinusoidal signal, cosine signal, and FM signal) compared with mode components after IMF3 is basically close to zero, so there is only a list of the previous three components shown in Table 2.

Sinusoidal signalCosine signalFM signal


From the decomposition results of EMD, it can be known that sine, cosine, and FM signal are partly mixed in IMF1 and IMF2, which cannot be separated well. Compared with the proposed method in this paper, the decomposition effect of VMD is obviously better than EMD, which can well separate the components from the original signal.

The above experimental results show that the proposed method can almost completely separate the components from the simulation signal free of noisy signal. However, the actual measured bearing vibration signal in the runtime is often affected by the strong noise background. Thus, the fault feature information is usually submerged in the noise environment. In order to prove the validity of the method, we discuss the feature extraction effect of this method under the different noise level. In the simulation signal, the Gauss white noise with standard variance being 0.1, 0.2, 0.4, 0.3, 0.5, 0.7, and 0.9 is added in turn. The cross-correlation coefficient between the decomposed mode components and the original components in different noise conditions is calculated. Results of signal reconstruction by the proposed method are shown in Figure 5.

The noise of different intensity is added in the simulation signal, and then the signal is processed by using the proposed method. From Figure 5, it is obvious that the VMD algorithm based on PSO has a good performance of denoising.

4. Analysis of Measured Signal in Rolling Bearing Experiment System

In order to verify that the proposed method is effective in the experiment, the vibration data of rolling bearing experiment system is used to be analyzed. The experimental system is shown in Figure 6. The whole experimental device is driven by a 550 W (220 V~50 Hz) AC motor. The yellow arrow points the position of the replaceable bearing in Figure 5. In this experiment, the electric spark machining method is used to carry out pitting treatment on the outer ring of replaceable bearing to simulate the faults of the outer ring of bearing. The acceleration signal of the experiment is collected in the vertical direction of the bearing on the right side of the experimental platform using the CSI2130 data analyzer of America. The parameter failure frequencies of fault simulation test-bed are shown in Table 3. It is worth mentioning that the rotating frequency and outer fault frequency are 24.17 Hz and 87.01 Hz, respectively.

Rotating speed /minRotating frequency/HzSampling frequency/HzSampling time/sOuter fault frequency/Hz


The time-domain graph of measured bearing fault signal is shown in Figure 7(a). The result of envelope spectrum analysis demonstrates that it is difficult to extract the fault feature due to the interference of noisy signal, which is presented in Figure 7(b).

From Figure 7(b), it is a fact that the 1 to 3 multiplications’ frequency of bearing outer fault is interfered by other irrelevant signals such as noisy signal. Particularly, the rotating frequency can hard be identification by the means of envelope spectrum analysis. The proposed method in this paper is introduced to optimize and analyze the bearing fault data. In the PSO algorithm, the number of iterations is set as 50 and the population size is selected as 20. The fitness value during the iteration process is shown in Figure 8. The optimized parameters of the penalty parameter and the number of the components about VMD method are selected as a collection (1800, 10). By using the optimized parameters, the bearing fault data are decomposed into ten-mode component and the selection of optimal mode component is carried out. Subsequently, the best representative decomposition component of eighth decomposed component was employed to verify the validity of the method, which was determined by the criterion of maximum correlated kurtosis. The correlated kurtosis of different component was shown in Figure 9 and we can draw a conclusion that the eighth decomposed component has a lager correlated kurtosis value. Figure 10 indicates the result of envelope spectrum analysis for the eighth decomposed component.

It can be seen from the decomposition results of optimized VMD algorithm that the identification accuracy of fault feature frequency is improved compared with the traditional envelope spectrum analysis shown in Figure 7(b). Only 3 multiplications’ frequency of the outer ring fault frequency can be found in the envelope spectrum of the original signal, while after optimization, it can find more than 10 multiplications of the fault frequency and detect the rotating frequency . Additionally, the characteristic spectral line was obvious and less inferred by other spectral lines, which confirms the validity of the proposed method to the fault feature extraction of rolling bearing. The contrastive analysis of the EMD and the proposed method is also carried out, which indicates that the optimized VMD algorithm has a better ability in fault feature identification shown in Figure 11.

5. Conclusions

A novel method of particle swarm optimization in variational mode decomposition method was introduced in faint fault feature extraction of rolling bearing. The main conclusions of this paper include the following. (1) The particle swarm optimization algorithm was applied to the parameter selection of the optimal penalty parameter and the number of components, which largely depends on the suitable fitness function determined by the maximum ratio between the mean value and the variance of cross-correlation coefficient. Moreover, the maximum correlated kurtosis is used to select the optimal component. It is significant that the proposed method can avoid the interference of human experience and the diagnostic results are more reasonable. (2) Simulated signal and measured fault bearing signal measured from the rolling bearing experiment system were used to verify the validity of the method. The result demonstrated that the proposed method has an advantage over the traditional EMD method and envelope spectrum analysis in faint fault signal processing for rolling bearings, which make it possible for the proposed method to be a powerful tool in solving the problem of signal channel bind source separation.

Competing Interests

The authors declare that there are no competing interests regarding the publication of this paper.


This work was supported by the National Natural Science Foundation of China (nos. 51475339 and 51405353) and the Key Laboratory of Metallurgical Equipment and Control of Education Ministry, Wuhan University of Science and Technology (2015B11).


  1. M. Zeng, Y. Yang, J. Zheng, and J. Cheng, “Maximum margin classification based on flexible convex hulls for fault diagnosis of roller bearings,” Mechanical Systems and Signal Processing, vol. 66-67, pp. 533–545, 2016. View at: Publisher Site | Google Scholar
  2. H. Wang, J. Chen, and G. Dong, “Feature extraction of rolling bearing's early weak fault based on EEMD and tunable Q-factor wavelet transform,” Mechanical Systems and Signal Processing, vol. 48, no. 1-2, pp. 103–119, 2014. View at: Publisher Site | Google Scholar
  3. X. An, D. Jiang, C. Liu, and M. Zhao, “Wind farm power prediction based on wavelet decomposition and chaotic time series,” Expert Systems with Applications, vol. 38, no. 9, pp. 11280–11285, 2011. View at: Publisher Site | Google Scholar
  4. H. Li and X. Wang, “Detection of electrocardiogram characteristic points using lifting wavelet transform and Hilbert transform,” Transactions of the Institute of Measurement and Control, vol. 35, no. 5, pp. 574–582, 2013. View at: Publisher Site | Google Scholar
  5. N. E. Huang, Z. Shen, S. R. Long et al., “The empirical mode decomposition and the Hilbert spectrum for nonlinear and non-stationary time series analysis,” Proceedings of The Royal Society A: Mathematical, Physical and Engineering Sciences, vol. 454, no. 1971, pp. 903–995, 1998. View at: Publisher Site | Google Scholar | MathSciNet
  6. C.-F. Lin, “Chaotic visual cryptosystem using empirical mode decomposition algorithm for clinical EEG signals,” Journal of Medical Systems, vol. 40, no. 3, pp. 1–10, 2016. View at: Publisher Site | Google Scholar
  7. Y. Lei, J. Lin, Z. He, and M. J. Zuo, “A review on empirical mode decomposition in fault diagnosis of rotating machinery,” Mechanical Systems and Signal Processing, vol. 35, no. 1-2, pp. 108–126, 2013. View at: Publisher Site | Google Scholar
  8. P. Nikolakopoulos and A. Zavos, “Slew bearings damage detection using hilbert huang transformation and acoustic methods,” Tribology in Industry, vol. 37, no. 2, pp. 170–175, 2015. View at: Google Scholar
  9. H. Aied, A. González, and D. Cantero, “Identification of sudden stiffness changes in the acceleration response of a bridge to moving loads using ensemble empirical mode decomposition,” Mechanical Systems and Signal Processing, vol. 66-67, pp. 314–338, 2016. View at: Publisher Site | Google Scholar
  10. J. Gilles, “Empirical wavelet transform,” IEEE Transactions on Signal Processing, vol. 61, no. 16, pp. 3999–4010, 2013. View at: Publisher Site | Google Scholar | MathSciNet
  11. Y. Jiang, H. Zhu, and Z. Li, “A new compound faults detection method for A new compound faults detection method for rolling bearings based on empirical wavelet transform and chaotic oscillatorrolling bearings based on empirical wavelet transform and chaotic oscillator,” Chaos, Solitons & Fractals, 2015. View at: Publisher Site | Google Scholar
  12. X. Chen, Z. Du, J. Li, X. Li, and H. Zhang, “Compressed sensing based on dictionary learning for extracting impulse components,” Signal Processing, vol. 96, pp. 94–109, 2014. View at: Publisher Site | Google Scholar
  13. Q. He and X. Wang, “Time-frequency manifold correlation matching for periodic fault identification in rotating machines,” Journal of Sound and Vibration, vol. 332, no. 10, pp. 2611–2626, 2013. View at: Publisher Site | Google Scholar
  14. R. Liu, “Condition monitoring of low-speed and heavily loaded rolling element bearing,” Industrial Lubrication and Tribology, vol. 59, no. 6, pp. 297–300, 2007. View at: Publisher Site | Google Scholar
  15. K. Dragomiretskiy and D. Zosso, “Variational mode decomposition,” IEEE Transactions on Signal Processing, vol. 62, no. 3, pp. 531–544, 2014. View at: Publisher Site | Google Scholar | MathSciNet
  16. C. Aneesh, S. Kumar, P. Hisham et al., “Performance comparison of variational mode decomposition over empirical wavelet transform for the classification of power quality disturbances using support vector machine,” Procedia Computer Science, vol. 46, pp. 372–380, 2015. View at: Google Scholar
  17. Y. Wang, R. Markert, J. Xiang, and W. Zheng, “Research on variational mode decomposition and its application in detecting rub-impact fault of the rotor system,” Mechanical Systems and Signal Processing, vol. 60, pp. 243–251, 2015. View at: Publisher Site | Google Scholar
  18. S. Lahmiri, “Long memory in international financial markets trends and short movements during 2008 financial crisis based on variational mode decomposition and detrended fluctuation analysis,” Physica A: Statistical Mechanics and its Applications, vol. 437, pp. 130–138, 2015. View at: Publisher Site | Google Scholar
  19. J. Kennedy, “Particle swarm optimization,” in Encyclopedia of Machine Learning, pp. 760–766, Springer US, 2010. View at: Google Scholar
  20. M. Yuwono, Y. Qin, J. Zhou, Y. Guo, B. G. Celler, and S. W. Su, “Automatic bearing fault diagnosis using particle swarm clustering and Hidden Markov Model,” Engineering Applications of Artificial Intelligence, vol. 47, pp. 88–100, 2016. View at: Publisher Site | Google Scholar
  21. G. L. McDonald, Q. Zhao, and M. J. Zuo, “Maximum correlated Kurtosis deconvolution and application on gear tooth chip fault detection,” Mechanical Systems and Signal Processing, vol. 33, no. 1, pp. 237–255, 2012. View at: Publisher Site | Google Scholar

Copyright © 2016 Cancan Yi et al. This is an open access article distributed under the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.

More related articles

1668 Views | 976 Downloads | 43 Citations
 PDF  Download Citation  Citation
 Download other formatsMore
 Order printed copiesOrder

Related articles

We are committed to sharing findings related to COVID-19 as quickly and safely as possible. Any author submitting a COVID-19 paper should notify us at to ensure their research is fast-tracked and made available on a preprint server as soon as possible. We will be providing unlimited waivers of publication charges for accepted articles related to COVID-19. Sign up here as a reviewer to help fast-track new submissions.