2.1. Variational Mode Extraction
Variational mode extraction (VME) is a new signal processing method, which can effectively obtain the desired mode components by presetting the penalty factor and mode center-frequency. The theoretical ideas of VME are similar to VMD, but it is faster than the VMD because it only looks for the specified frequencies. Briefly speaking, in the VME, the original time series
can be split into two parts by the following equation:
where
is the desired mode components,
is the residual signal. Specifically, mode extraction process of VME is established based on the following three conditions.
(1) The desired mode components have compactness around the center-frequency. To achieve this goal, minimization problem of the following objective function is solved to obtain the desired compact mode components.
where
denotes the center-frequency of mode components
,
represents the Dirac distribution, and the asterisk * represents the convolution operation.
(2) Spectral overlap of the residual signal
and the desired mode components
should be as small as possible. That is, in the frequency band of the desired mode components, the energy of the residual signal
should be minimized. Particularly, when the energy of the residual signal
around the center-frequency is equal to 0, a complete and accurate mode component will be obtained. To overcome these limitations, the contents of the residual signal
are firstly found out via using a proper filter, and then the energy of the residual signal
is regarded as the indicator to evaluate the spectral overlap degree of
and
. For this purpose, here a filter with frequency response of
is designed:
where
is similar to the Wiener filter at the frequencies far away from
, this because it has an infinite gain at
. Hence, the following penalty function is adopted to minimize the spectral overlap of
and
.
where
denotes the impulse response of the designed filter.
(3) The obtained mode components
should be meet the equality constraint listed in Equation (1) to guarantee complete reconstruction. That is, the extraction problem of the desired mode components can be expressed as solving the following constrained minimization problem:
where
is the penalty factor of balancing
and
. To solve the above reconstruction constrained problem, the following augmented Lagrangian function is adopted by introducing the quadratic penalty term and Lagrangian multiplier.
where
is the Lagrangian multiplier. According to the Parseval theorem, by using
instead of
and adopting the equality
, the above Equation (6) can be rewritten as follows:
To solve the minimization problem of augmented Lagrangian function, the alternate direction method of multipliers algorithm (ADMM) is introduced. In ADMM, multiple iteration suboptimizations are conducted to obtain the optimization variables (
,
, and
). Hence, in the
n + 1 iteration, the mode components
can be obtained by the following equation:
To simplify the above Equation (8), according to Equation (3) and some algebraic manipulations, the mode components
at the
n + 1 iteration can be rewritten by:
To minimize the Equation (11) with respect to
, according to some approximate calculations, in the
n + 1 iteration, the mode center-frequency
can approximately be expressed as:
Finally, the dual ascent method is used to update the Lagrangian multiplier
of ADMM, that is
where
denotes the update parameter which amounts to time-step of the dual ascent. The specific procedure of VME can be found in the original literature [
19] and the VME code is available on the Mathworks website.
2.2. Parameter Adaptive Variational Mode Extraction
When VME is used to process the collected bearing vibration signal, its two important parameters (i.e., penalty factor
and mode center-frequency
) need to be artificially selected in advance. Thus, it does not possess adaptive capability. In other words, the parameter setting of VME has a big effect on its feature extraction performance. Due to the penalty factor
controls the compactness of the obtained mode components, so the smaller penalty factor
describes the larger bandwidth of mode components. The closer the predefined mode center-frequency
is to the true center frequency of the desired mode components, the better the feature extraction ability of VME is. Therefore, a suitable method needs to be adopted to automatically select the important parameters of VME. Whale optimization algorithm (WOA) [
34] is a recently reported intelligent optimizer, which can mimic bubble-net foraging behavior of humpback whales by applying a bubble-net search mechanism. Compared with particle swarm optimization (PSO), cuckoo search algorithm (CSA), firefly algorithm (FA) and grey wolf optimizer (GWO), WOA has a faster convergence speed, higher convergence accuracy and stronger ability of extremum optimization [
35]. Hence, to avoid the problem of empirical selection of the key parameters of VME, a parameter adaptive variational mode extraction (PAVME) is proposed in this paper, where WOA is adopted to automatically determine two key parameters (i.e., penalty factor
and mode center-frequency
) of VME, which can improve fault feature extraction ability of VME.
Figure 1 shows the flowchart of using WOA to optimize the parameters of VME method. Detailed procedures of parameter optimization in the PAVME are described as follows:
(1) Initialize the population of whales and define the parameters of WOA method. Specifically, set the population size N = 50, maximum number of iterations T = 200 (i.e., epoch limits). Due to VME involves two key parameters to be optimized, so the position of each whale is expressed by a vector , where is the penalty factor of VME, denotes the initial mode center-frequency of VME and meets . The upper and lower bound of the vector respectively is set as [200, 10,000] and [, ], where is the sampling frequency of the raw bearing vibration signal.
(2) Calculate the fitness value of each whales and determine the current optimal position of whales. In this step, inspired by signal-to-noise ratio (SNR) [
36] and fault feature ratio (FFR) [
37], a new and effective sensitive index hailed as signal characteristic frequency-to-noise ratio (SCFNR) is regarded as the fitness value to guide the parameter optimization process of VME, and the SCFNR index is calculated by
where
means the
i-th fault characteristic frequency of Hilbert envelope spectrum of the extracted mode components
,
denotes the amplitude of Hilbert envelope spectrum of the original bearing vibration signal at the
i-th fault characteristic frequency,
represents the amplitude of Hilbert envelope spectrum of the original bearing vibration signal at the
j-th frequency
f,
N and
M are the number of all frequencies and fault characteristic frequencies of Hilbert envelope spectrum of the original bearing vibration signal, respectively. The larger SCFNR value represents the better feature extraction ability of VME. That is, parameter optimization process of VME can be understood as the process of maximizing the fitness value (SCFNR). Hence, the objective function of parameter optimization process of VME can be defined as follows:
where
denotes the SCFNR value of the extracted mode components under different combination parameters
,
represents the sampling frequency of the original bearing vibration signal.
(3) Before reaching the stop condition, update the parameters
a,
A,
C,
l and
p under each iteration. If
, the position updating pattern of the shrinking encircling mechanism of whales is adopted. Otherwise, the position updating pattern of the spiral model of whales is adopted. That is, the probability of selecting the shrinking encircling mechanism or the spiral model to update the position of whales is the same. Concretely, if
and
, update the position of the current whale according to Equation (14). If
, update the position of the current whale according to Equation (15). If
and
, update the position of the current whale according to the randomly prey search mechanism of Equation (16).
where
X is a position vector for all whales,
t is the time or iteration metrics,
X* is the current optimal solution,
A and
C represent the coefficient vector and they meets
and
,
a is a convergence factor that linearly decays from 2 to 0 throughout all iterations,
r is a random vector between 0 and 1,
b is a constant value that defines a logarithmic spiral shape in terms of a particular path,
l is a random value between −1 and 1,
p is a random value between 0 and 1, which can be used to switch Equations (14) and (15) when updating the position of whales.
represents the position vector for the randomly selected whales in the current iteration,
D denotes distance of the
i-th whale to the prey,
A and
C represent the coefficient vector.
(4) Calculate the fitness value of each whales and determine the global optimal position of whales. If is better than , is regarded as the global optimal position of whales. Otherwise, keep as the individual optimal position to continue to update.
(5) Check that the stop condition is met. Specifically, determine whether the largest SCFNR value or maximum iteration number is reached. If it reaches the largest SCFNR value or maximum iteration number, output the optimized results (i.e., the optimal parameters of VME). Otherwise, define t = t + 1, continue to conduct steps (3)–(4) until the stop condition is met.
(6) Use the parameter optimized VME to extract the desired mode components of the collected bearing vibration signal.
Briefly speaking, the proposed PAVME method mainly consists of two sub-blocks (i.e., parameter optimization process and mode component extraction process).
Figure 2 shows the block diagram of PAVME. Therein, the first sub-block is the parameter optimization process based on WOA method, which is aimed at obtaining the optimal combination parameters (i.e., penalty factor
and mode center-frequency
) of VME. The second sub-block is mode component extraction process based on VME containing the optimal combination parameters.
2.3. Comparison among PAVME, VME, VMD and EMD
To show the effectiveness of PAVME in extracting periodic impulse features of bearing vibration signal, according to the literature [
36], here we established one bearing fault simulation signal
x(
t), which is mainly composed of three parts (i.e.,
x1(
t),
x2(
t) and
n(
t)). The specific expression of simulation signal is as follows:
where the first part
denotes the periodic impulse series related to bearing faults,
is the bearing fault characteristic frequency and meets
= 30 Hz. The second part
represents the harmonic component with the frequency of
f2 = 20 Hz and
f3 = 30 Hz. The third part
represents the Gaussian white noise generated by MATLAB function
. The sampling frequency and sampling length of simulation signal
x(
t) are set as 8192 Hz and 4096 points, respectively.
Figure 3 shows time domain waveform of simulation signal
x(
t) and its corresponding components.
The proposed PAVME and three standard methods (VME, VMD and EMD) are adopted to process the simulation signal
x(
t). In PAVME, the penalty factor
and mode center-frequency
are automatically selected as 1680 and 2025 Hz by using WOA. In the standard VME, the combination parameters (i.e., penalty factor
and mode center-frequency
) are artificially set as 2000 and 2500 Hz. In VMD, the decomposition mode number
K and penalty factor
are also automatically selected as 4 and 2270 Hz by using WOA.
Figure 4 shows the periodic mode components extracted by different methods (i.e., PAVME, VME, VMD and EMD). Seen from
Figure 4, although three methods (PAVME, VME and VMD) can all obtain the periodic impulse features of simulation signal, but their obtained results are different. The periodic mode components extracted by EMD have a big difference with the real mode component
of the simulation signal. Hence, for a better comparison, fault feature extraction performance of the four methods (PAVME, VME, VMD and EMD) is quantitatively compared by calculating four evaluation indexes (i.e., kurtosis, correlation coefficient, root-mean-square error (RMSE) and running time).
Table 1 lists the calculation results. Seen from
Table 1, kurtosis and correlation coefficient of the proposed PAVME method is higher than that of other three methods (i.e., VME, VMD and EMD). The RMSE of the PAVME method is less than that of other three methods. This means that the proposed PAVME has better feature extraction performance. However, the running time of VMD is highest, the second is PAVME and the smallest running time is EMD. This because the PAVME and VMD are optimized by WOA, so their computational efficiency is reduced, but it is acceptable for most occasions. The above comparison shows that the PAVME method is effective in bearing fault feature extraction.