Abstract
This paper proposes a novel method to improve the variational mode decomposition (VMD) method and to automatically acquire the sensitive intrinsic mode function (IMF). First, since fault signals are impulsive and periodic, a weighted autocorrelative function maximum (AFM) indicator is constructed based on the Gini index and autocorrelation function to serve as the optimization objective function. The mode number K and the penalty parameter α of VMD are automatically obtained through an optimal parameter searching process underpinned by the improved particle swarm optimization (PSO) algorithm with a variety of inertia weights. This improvement solves one of the major drawbacks of the conventional VMD method, that is, the need to manually set parameters. Then, an optimal IMF automatic selecting process is performed for single-failure faults and compound faults, according to the principles of the maximum weighted AFM indicator and maximum spectrum peak ratio (SPR), respectively. The sensitive IMFs are then subjected to an envelope demodulation analysis to obtain the fault characteristic frequency. The results of simulations and experiments show that the proposed method can effectively identify fault characteristics early, especially compound faults, demonstrating great potential for real-world applications.
1. Introduction
Developing techniques for monitoring rolling bearing conditions and diagnosing faults are key to realizing the transition from traditional regular maintenance to condition-based maintenance. Owing to harsh working environments and changing conditions, vibrations of rolling bearings are typically characterized by strong background noise, unsteadiness, and coupling modulation, which make it difficult to extract and identify fault characteristics. Nonetheless, accurately extracting fault characteristic frequencies from a mix of complex, nonstationary, and nonlinear vibration signals is essential for early fault identification [1].
Time-frequency analysis methods simultaneously provide local information about nonstationary signals in both the time domain and frequency domain and can be used to extract fault characteristics from weak signals. Some commonly used time-frequency analysis methods include empirical mode decomposition (EMD) [2], ensemble empirical mode decomposition (EEMD) [3], and local mean decomposition (LMD) [4]. All of these are based on recursive decomposition and, thus, have the same disadvantages, such as endpoint effects and modal aliasing, which can impact accuracy in early detection of weak fault characteristics [5].
Dragomirestkiy and Zosso [6] recently proposed the VMD method, which is based on solid mathematical foundations and can be used for broad applications in the field of mechanical fault diagnosis. The VMD method introduces a secondary penalty factor and a Lagrange multiplier to transform the constrained variational problem into a nonconstrained variational problem. This nonrecursive decomposition approach avoids some of the problems of recursive decomposition, such as the endpoint effect and the need to determine conditions for terminating the decomposition, and, in addition, effectively remedies the deficiencies of conventional decomposition methods and achieves high computational efficiency and good noise robustness. However, the VMD method also has a number of drawbacks. Most notably, the mode number K and the penalty parameter α must be set before starting the decomposition. Moreover, problems like aliasing between mode components or excessive decomposition may occur if parameters are not set properly. Therefore, adaptive acquisition of VMD parameters is of crucial importance.
To tackle this challenge, Shi and Yang [7] proposed a step-by-step parameter optimization strategy which achieved good results in validation experiments using a wind turbine. Wang et al. [8] used the minimum average envelope spectrum entropy as the objective function and PSO to optimize the VMD parameters. The method was shown to be superior to EMD, LMD, and other envelope and wavelet transform methods. Yan et al. [9] proposed a composite fault detection method for rotating machinery based on optimal variational mode decomposition (OVMD) and a 1.5-dimensional envelope spectrum. The envelope spectrum entropy value was used as the objective function, and the optimal VMD parameters were obtained using the genetic algorithm, thereby achieving adaptive acquisition of the mode number and penalty parameter. In another approach, Zhang et al. [10] proposed a grasshopper optimization algorithm- (GOA-) based adaptive VMD parameter analysis method that works in the following way: a weighted kurtosis indicator is constructed based on the kurtosis and correlation coefficient; VMD parameters are optimized using the GOA algorithm with the weighted kurtosis indicator as the objective function; and the sensitive IMF is selected following the principle of maximum weighted kurtosis for extracting the fault characteristics. Liu et al. [11] used the principle of maximum kurtosis to obtain the optimal values of parameter pair K and α in order to achieve an accurate VMD. Lian et al. [12] proposed an adaptive VMD method that uses the principle of maximum kurtosis as an evaluation criterion to automatically determine the mode number.
Based on the above studies, it can be concluded that constructing a suitable objective function and selecting an appropriate optimization algorithm are two critical issues in the adaptive acquisition of VMD parameters. Some indicators such as kurtosis [10–14], correlation coefficient [15], autoregressive (AR) model parameters [16], and entropy [8] have proven extremely useful in vibration signal analysis and fault characteristics extraction. As a sparse indicator, kurtosis is widely used in the fault diagnosis of rotating machinery and can effectively extract impulsive components of signals. The principle of maximum kurtosis is useful for constructing the objective function for VMD parameter optimization. However, kurtosis is more suitable for measuring the most important pulses in a signal. In addition to not being robust enough, kurtosis-based techniques focus more on individual pulses than on the desired periodic pulses [17].
The Gini index is widely used to measure income inequality in countries and has recently been used in fault diagnosis of rotating machinery [18]. The index is not only very effective in identifying fault pulses but offers superb performance in preventing random pulses and disturbances. Miao et al. [19] used the Gini index to improve the kurtosis guidance map and were able to identify the resonance band, which contains the richest information. More specifically, the resonance band contains the fault pulse rather than interferences caused by fault modulation and random pulses. Kurtosis and the Gini index are both used as indicators but share the same shortcoming; that is, both measures are aimed at detecting the impact of fault pulses and ignore recurring periodic pulses.
The autocorrelation function retains periodic signal information; however, phase information of the original signal is not retained. For white noise signals, the autocorrelation function quickly decays to zero at the fastest decay rate. For an impulse signal, the autocorrelation function also acquires impulse characteristics while retaining periodic characteristics. The autocorrelation function of a noise-containing impulse signal decays rapidly, but its amplitude becomes large during the impulse. This suggests that the autocorrelation function of an impulse signal combines characteristics of the autocorrelation function of white noise and impulse signal. Chen et al. [20] proposed an AFM indicator based on this particular feature of the autocorrelation function, which was then used as a quantitative evaluation criterion of the IMF value after complementary EEMD processing and achieved good results.
Up to now, most studies in the literature have focused on single-failure faults (e.g., inner ring failure, outer ring faults, and rolling element faults of rolling bearings). In practice, most faults are compound and composed of two or more different types of faults. At present, there are many research studies on the diagnosis of compound faults. Wang et al. [21] proposed a modified VMD (MVMD) based on noise-assisted data analysis (NADA), which can extract compound fault features of gearbox under a strong noise environment. Yang et al. [22] proposed a fault diagnosis framework based on a novel dual-extreme learning machine (ELM) network that can quickly detect single and simultaneous failures in a multistage gearbox. Zhong et al. [23] proposed a fault feature extraction framework based on feature extraction and support vector machine (SVM), and the experimental results show that the method can effectively diagnose single and simultaneous faults of the gas turbine generator system (GTGS). Wang et al. [24] proposed a gearbox integrated fault diagnosis method based on ensemble LMD, defined as CELMD (Complementary Ensemble Local Mean Decomposition) which overcomes the pattern mixing phenomenon and can be used to extract composite fault features. Wang [25] proposed a method based on Multipoint Optimal Minimum Entropy Deconvolution Adjusted (MOMEDA) and EEMD, which is suitable for searching complex fault pulse signals under strong noise environment. Wang et al. [26] proposed a maximum kurtosis spectrum deconvolution (MKSED) applied to the fault diagnosis of the gearbox, and the experimental results show that the compound faults of gearbox can be successfully diagnosed by using MKSED.
In this paper, a method for improving the VMD is proposed; a new weighted AFM indicator is constructed to serve as the objective function for parameter optimization. The improved PSO with a variety of inertia weights is used for optimal parameter searching. Mutual information of the decomposed components and the original signal is used to detect overdecomposition, upon which the decision to perform the search again is made. The VMD is performed after the optimal decomposition parameters are obtained. For single-failure faults, the weighted AFM indicator is used to select the sensitive IMF and for compound faults, SPR is used. Finally, the selected IMF undergoes an envelope demodulation analysis to extract the fault characteristic frequency. The following sections are arranged as follows. The VMD method is introduced in Section 2. The construction of weighted WAFM index, IPSO algorithm, mutual information, and the method proposed in this paper are introduced in Section 3. The simulation signal and test signal are analyzed using the proposed method, respectively, in Section 4 and Section 5. Finally, the conclusions of this paper are drawn in Section 6.
2. VMD
The VMD is an adaptive signal processing technique first proposed in 2014 that can be used to decompose a signal into a series of finite-bandwidth IMFs in a nonrecursive manner, thereby creating the right conditions for effective signal separation [6]. Migrating the signal decomposition process into the variational framework, VMD accomplishes signal decomposition by seeking the optimal solution of the variational model.
2.1. Construction of the Variational Problem
With VMD, the signal is decomposed into K mode functions, each with a center frequency of . Therefore, the VMD can be expressed as the variational problem of solving K mode functions whose aggregated bandwidth is the smallest, constrained by the sum of all mode functions which must be equal to the original signal:where represents the center frequency of each mode.
The VMD can be obtained using the following procedure.
The Hilbert transform is performed on each mode function to obtain the analytical signal:
Then, a preliminary estimation is performed on the center frequency of the analytical mixed signal of each mode. The frequency of each mode component is modulated by its corresponding baseband frequency:
The square of the gradient 2-norm of the above demodulated signal is calculated and the bandwidth of each modal function is estimated.
2.2. Solution to the Variational Problem
To solve the variational problem presented in Section 2.1, the secondary penalty factor and Lagrange multiplier are introduced and used to transform the constrained variational problem into an unconstrained variational problem. The secondary penalty factor serves as a guarantee of the accuracy of signal reconstruction when Gaussian noise is present, while the Lagrange multiplier makes the constraint more stringent. The augmented Lagrange expression is as follows:
Both the IMF and its center frequency are continuously updated using the alternate direction method of multipliers (ADMM). Finally, the saddle point of the above formula is obtained, and this is the optimal solution. The IMF of all frequency domains along with their center frequencies can be obtained using the following formulae:where is the result of the current residual amount after Wiener filtering.
The above is an adaptive decomposition procedure of VMD, which is in essence a Wiener filtering process. Signal decomposition is realized by iteratively searching for the optimal solution of the variational model in the variational problem framework. The approach achieves good noise robustness and effectively avoids the endpoint effect and modal aliasing problem.
3. WAFM-IPSO-VMD
To eliminate the need to set parameters in advance, which is a drawback of the VMD, this paper proposes a method for improving the VMD by enabling adaptive parameter acquisition. The method includes three aspects: first, the weighted AFM indicator is constructed as the objective function; second, the improved PSO is used to search for the optimal parameters of the VMD algorithm and mutual information of the decomposed components, and the original signal is used to detect overdecomposition, and thus to decide whether or not to perform the search again; third, an optimal IMF automatic selecting process is performed for the single-failure fault and compound faults following the principles of the maximum weighted AFM and maximum SPR, respectively.
3.1. WAFM
When searching for the optimal parameter combination using the improved PSO, construction of the objective function is a critical step. If a fault occurs, a periodic impulse signal will appear in the mixed signal. As described before, the Gini index aims to detect the impact of fault pulses and ignores periodic pulses that repeatedly occur when a fault is present, whereas AFM can effectively extract periodic characteristics of the signal. To enable more effective fault detection, this paper proposes constructing the weighted AFM indicator based on characteristics of the Gini index and the AFM.
3.1.1. Gini Index
The Gini index is an inequality index that is often used in economics to reflect the overall income gap [27]. In mechanical fault diagnosis, the sequence elements must be positive in order to calculate the Gini index; therefore, the envelope spectrum Gini index has been used to extract the bearing fault signal [19] and achieved good results. The envelope spectrum Gini index can be defined as follows:where the vector is the envelope spectrum of the signal; is a sequence of points in the envelope spectrum arranged in ascending order of amplitude; and are the new index numbers after sorting.
. In economics, the closer the Gini index is to zero, the more equal the income distribution is; the closer the Gini index is to 1, the more uneven the income distribution is. Thus, extracting a bearing fault signal from the vibration signal of a faulty machine by applying the principle of the maximum Gini index of the envelope spectrum is feasible.
3.1.2. AFM
The autocorrelation function is extremely effective in ascertaining the periodicity of a signal and can determine whether the signal is noisy [20].
The autocorrelation function describes how similar a signal is to itself at different times. For a random process , a signal that occurs at a certain moment in time is denoted and the same signal becomes after a certain period k. Self-covariance of the signal can be defined as
The autocorrelation coefficient of and is
For a stationary process, . So, the autocorrelation coefficient of and can be defined as
The autocorrelation coefficient sequence with the time interval of k serving as a variable is called the autocorrelation function and can be defined as
The autocorrelation function is an even function. When performing the analysis, it is enough to analyze one-half of the autocorrelation function corresponding to the positive half of the axis. In signal processing, the autocorrelation function is defined as
As a sequence, the autocorrelation function cannot describe the characteristics of the signal. To convert the autocorrelation function into a quantitative indicator, the root mean square value of the maximum value sequence (denoted as AFM [20]) of the autocorrelation function can be used as an index for evaluating the IMF. The AFM can be expressed by the following formula:where is the maximum point sequence of the autocorrelation function of the signal envelope and N is the sequence length.
3.1.3. WAFM
In view of the impulsive and periodic nature of fault signals, this paper uses the product of the Gini index and the AFM as the objective function for VMD parameter optimization. Since , multiplying the AFM by the Gini index is equivalent to multiplying the AFM by a weight. Hence, the new indicator is defined as the weighted AFM:
3.2. Improved PSO
The PSO algorithm is a population-based random search method proposed by Kennedy and Eberhart [28]. The algorithm can be detailed as follows:
Suppose that the population space dimensions are , where N represents the number of populations, D represents the dimensionality of particles; the D-dimensional vector represents the position of the ith particle and its velocity can be expressed as ; represents the individual extremum; and represents the global extremum. After the individual and global extrema are obtained, the particle velocity and position are updated using the following formulae:where and are learning factors, and are random numbers uniformly distributed within [0, 1], and is the inertia weight.
The inertia weight affects local convergence and global exploration ability of the algorithm. Global exploration is performed when the weight is large and a local search is carried out when the weight is small. The PSO algorithm with a nonlinear decrease of the inertia weight improves upon on the inertia weight [29]. The improved PSO algorithm is more effective since the solution space is analyzed during the initial search period such that the algorithm quickly approaches the optimal solution. In the late stage of search, higher accuracy can be achieved through iterative fine-tuning. The formula for improving the weight is as follows:where is the inertia weight after initialization, is the final inertia weight, is the total number of iterations, is the current number of iterations, and is the nonlinear adjustment factor.
3.3. Automatic Acquisition of IMF
For single-failure faults, the sensitive IMF is automatically obtained using the principle of the maximum weighted AFM. For compound faults, the sensitive IMF is automatically obtained using the maximum SPR principle.
The SPR can effectively extract the periodic impulse components of the signal and is immune to interference from abnormal spikes in time-domain signals [30]. A larger value of SPR indicates a larger periodic impulse component of the fault signal:
Here, denotes the amplitude of each point on the envelope spectrum, denotes the number of envelope lines; denote the amplitude peaks corresponding to the fault characteristic frequencies of the estimated bearing inner ring’s frequencies—that is, the bearing outer ring, roller, and cage, respectively. Parameters denote the number of harmonic waves (for the purpose of this paper, h = 3).
3.4. Mutual Information
Mutual information (MI) [31] can quantitatively represent the correlation between two random variables, and it is more accurate than the correlation coefficient method. Mutual information can effectively measure the coupling degree between IMF components. The stronger the coupling of variables is, the greater the mutual information is. It is defined as follows:where S is the domain of X and Y, is the joint distribution of X and Y, and and are the marginal distributions of X and Y, respectively.
3.5. Proposed Method
A flowchart of the method proposed in this paper is shown in Figure 1. The procedure is as follows:(1)Input vibration signal ; set the range of the VMD parameter to be optimized and initialize the IPSO parameters.(2)Randomly initialize the particles’ positions, calculate the objective function of each particle, and save the position corresponding to the maximum WAFM. That is, the parameter combination .(3)Construct a weighted AFM indicator as the objective function for optimization. Use the IPSO algorithm to update the position of the particle swarm and calculate the objective function of the particle swarm. Determine whether overdecomposition occurs using the minimum value of mutual information from each decomposed component and the original signal. If the minimum value of mutual information is less than 0.1, the current optimal solution is not updated; otherwise, the optimal solution is updated.(4)Judge whether the stop condition is met: if so, exit to with optimal solution; if not, repeat step 3.(5)Perform VMD on the original signal using the determined optimal parameter set . For a single-failure fault, the sensitive IMF is automatically obtained using the principle of the maximum weighted AFM. For compound faults, the sensitive IMF is automatically obtained using the maximum SPR principle.(6)Envelope demodulation analysis is performed on the IMF to obtain the fault characteristics.

4. Simulation Signal
In order to verify the effectiveness of the method proposed in this paper, the following model is used to simulate the vibration signal of the failure of the bearing outer ring, and add random noise with an amplitude of 2. The simulation signal is as follows:where denotes the impact signal of a bearing fault, an intrinsic frequency , ; a displacement constant , a damping coefficient , an impact fault frequency of 100 Hz, and a sampling frequency , with a sampled point count of . The waveform of the simulation signal is shown in Figure 2.

It can be seen from Figure 2 that, under the interference of random noise, the periodic pulse in the simulation signal has been buried, and it is difficult to find the periodic impulse signal in the time-domain waveform. The envelope demodulation analysis of the simulation signal is conducted, and the resulting envelope spectrum is shown in Figure 3, in which failure characteristic components cannot be observed.

4.1. Optimization of the VMD Parameters
The simulation signal is being decomposed by the parametric adaptive VMD method proposed in this paper (search range of K: [2, 12]; search range of α: [100, 5000]). The setting of PSO parameters in this paper is obtained by adjusting according to experience and previous studies on PSO. In this paper, two parameters need to be optimized, so the dimension of the particle is 2. According to the literature [32], PSO is not sensitive to population size. At present, the study of learning factors and is limited to the application of the problem. Considering the computational complexity and reliability, the population size was set as 10, as 1.7, and as 1.5 according to the experience after several tests and adjustments. The authors of literature [33] found that when the inertia weight belongs to [0.4, 0.95], the performance of particle swarm optimization algorithm will be greatly improved. So, this paper sets maximum inertia weight and minimum inertia weight . Initialize the IPSO, and set the population size N = 10, particle dimension D = 2, maximum evolutionary generation number t = 20, acceleration coefficients , , maximum inertia weight , minimum inertia weight , and nonlinear modulation index n = 1.2 [29]. Figure 4 shows the iteration curve for the optimization search, from which approximate a global maximum max (WAFM) = 0.0782 and the optimal parameter combination of [10, 3146] can be obtained.

4.2. VMD and Sensitive IMF Selection
On this basis, the optimal parameter combination [10, 3146] is used to perform the VMD of the original signal. The waveform of each IMF component after decomposition is shown in Figure 5(a). The histogram of IMF-WAFM is shown in Figure 5(b), in which it can be seen that the WAFM value of IMF4 is the highest, so it is selected for further envelope spectrum analysis, and the effect is shown in Figure 5(c).

(a)

(b)

(c)

(d)

(e)
Envelope spectrum analysis was also performed on the other IMF components. Due to limited space, only envelope spectrums of IMF3 and IMF5 were given, as shown in Figures 5(d) and 5(e). In the envelope spectrum of IMF5 in Figure 5(e), failure characteristic components cannot be observed. Although prominent spectral line can be observed at the fault characteristic frequency in Figure 5(d), there is a lot of background noise. In contrast, in the envelope spectrum of sensitive IMF in Figure 5(c), the spectral lines at the fault characteristic frequency and double frequency are more prominent. In Figure 5(c), the rotation frequency can also be observed significantly. This suggests that the improved VMD method proposed in this paper can effectively decompose the signal component containing fault characteristic components from the original signal buried by noise, which proves the effectiveness of the method.
4.3. Comparison with Other Methods
In order to verify the advantages of WAFM index proposed in this paper, the same optimal combination [10, 3146] is used for VMD, and sensitive IMF is selected through the maximum kurtosis principle. It is found that IMF6 is a sensitive IMF and that no fault characteristic components can be observed from its envelope spectrum, as shown in Figure 6.

To verify the advantages of the optimization method introduced in this paper, the combinations of parameters [10, 400] and [5, 3000] are artificially selected for VMD. The sensitive IMF envelope spectrum selected with WAFM can be seen in Figure 7. According to our findings, fault characteristic components cannot be observed in Figure 7(b). The fault characteristic frequency can be observed on the envelope spectrum of sensitive IMF in Figure 7(b), but it has been buried by noise. This reveals the fact that there is a gap between the method of artificial parameter selection and the method proposed in this paper.

(a)

(b)
For the purpose of further validating the advantage of the method proposed in this paper, EMD and EEMD algorithms were adopted to decompose the simulation signal. The two parameters of EEMD, ensemble number and amplitude coefficient, were set as 100 and 0.01 in [34]. After decomposition, maximum kurtosis principle is adopted to select the acquired IMFs. The time-domain waveforms of the sensitive IMFs are shown in Figures 8 and 9 [35, 36], in which fault characteristics cannot be observed. It is clear that the VMD method proposed in this paper is superior to both EMD and EMMD.


5. Experimental Signal Analysis
In order to verify the effectiveness of this method, it is applied to the actual data for analysis. The test rig is shown in Figure 10. The bearing type is HRB6304, the sampling frequency is 10 kHz, and 4096 data points are sampled every 1 minute. Failure of bearing inner ring of bearing and compound failure of inner and outer ring were selected for analysis, respectively, and the failure diagrams are shown in Figure 11. Table 1 shows the multiple of fault frequency of each bearing component.


(a)

(b)
5.1. Single Fault Data Analysis
This section analyses the data of inner race fault, and the 111th minute signal of the experiment is extracted for decomposition. The velocity of the signal is at 3,000 rpm, and the bearing inner race fault characteristic frequency can be calculated according to Table 1. The time-domain waveform and its envelope spectrum of this signal are shown in Figure 12. It can be observed that no periodic impact is visible on the time-domain waveform, and there is no prominent spectral line near the fault characteristic frequency in Figure 12(b), so it is impossible to judge the state of bearing accurately.

(a)

(b)
By using the method proposed in this paper, the signal is decomposed, and the approximate global maximum max (WAFM) = 0.1356 and the optimal parameter combination [K, α] = [10, 2416] can be obtained. On this basis, the optimal parameter combination [10, 2416] is used to perform the VMD of the original signal. The waveform of each IMF component after decomposition is shown in Figure 13(a). The histogram of IMF-WAFM is shown in Figure 13(b), in which it can be seen that the WAFM value of IMF8 is the highest, so it is selected for further envelope spectrum analysis, and the effect is shown in Figure 13(c).

(a)

(b)

(c)

(d)

(e)
Envelope spectrum analysis was also performed on the other IMF components. Due to limited space, only envelope spectrums of IMF7 and IMF9 were given, as shown in Figures 13(d) and 13(e). In the envelope spectrum of IMF9 in Figure 13(e), failure characteristic components cannot be observed. The envelope spectrum of IMF9 component in Figure 13(d) shows that prominent spectral lines cannot be observed obviously at the inner race fault characteristic frequency . In contrast, in the envelope spectrum of sensitive IMF in Figure 13(c), the spectral lines at the fault characteristic frequency are more prominent. On this basis, the following conclusion can be drawn: the bearing inner ring has serious local defects.
Similarly, in order to verify the advantages of WAFM index proposed in this paper, sensitive IMF is selected through the maximum kurtosis principle. It is found that IMF10 is a sensitive IMF, and its envelope spectrum is shown in Figure 14. Although the envelope line amplitudes at the fault characteristic frequency and double frequency can be observed, there is a lot of background noise, so it shows that the effect is not as good as that of the method proposed in this paper.

EMD and EEMD are used for signal decomposition, and kurtosis was used to select sensitive IMF. As shown by their time-domain waveform and envelope spectrum, although fault characteristic frequency can be observed in Figure 15, there are more noise interference patterns in the envelope spectrum obtained in this way, so the effect is less obvious than that in Figure 13(c). Failure characteristic components cannot be observed in Figure 16. It is, thus, clear that the VMD-based decomposition method proposed in this paper is superior to both EMD and EEMD.


5.2. Analysis of Compound Faults Data
A dataset of inner-outer ring coupling faults was used as a sample for analysis. The rotational speed indicated by the data was 2020 r/min. According to Table 1, the characteristic frequency of the inner ring fault was and the characteristic frequency of the outer ring fault was . The time domain waveform diagram and envelope spectrum of the group of signals are shown in Figure 17. No salient periodic impact can be observed in Figure 17(a). In the envelope spectrum as shown in Figure 17(b), the characteristic of the outer ring fault can be clearly observed; however, the characteristic of the inner ring fault cannot, making it difficult to determine the bearing condition.

(a)

(b)
By using the method proposed in this paper, the signal is decomposed, and the approximate global maximum max (WAFM) = 0.2203 and the optimal parameter combination [K, α] = [6, 4605] can be obtained. On this basis, the optimal parameter combination is used to perform the VMD of the original signal. The waveform of each IMF component after decomposition is shown in Figure 18(a). The histograms of IMF-SPRI and IMF-SPRO are shown in Figures 18(b) and 18(c), in which it can be seen that the SPRI value of IMF1 is the highest and the SPRO value of IMF6 is the highest, so the IMFs are selected for further envelope spectrum analysis, and the effects are shown in Figures 18(d) and 18(e).

(a)

(b)

(c)

(d)

(e)
As shown in Figure 18(d), the fault characteristic frequency of the inner ring can be observed in the envelope spectrum of the IMF1 component. Salient lines can be observed near the characteristic frequency of the outer ring fault and its double frequency in the envelope spectrum of the IMF6 component in Figure 18(e). This indicates serious defects on both the inner and outer rings.
Signal decomposition was performed using EMD and EEMD, and the sensitive IMF was selected using SPR. After EMD processing, the sensitive IMF component obtained through SPRI-based selection or via SPRO-based selection were the same, as shown in Figure 19. The fault characteristic frequency is clearly buried in noise to the extent that no prominent spectral lines can be observed at the location of the fault characteristic frequency.

After EEMD processing, the sensitive IMF of the inner ring fault obtained via SPRI-based selection is IMF3 and the sensitive IMF of the outer ring fault obtained through SPRO-based selection is IMF5. The envelope spectrum is shown in Figure 20. No fault characteristic frequency can be observed in the sensitive IMFs of the inner ring and outer ring, indicating that the VMD-based decomposition method proposed in this paper is superior to EMD and EMMD.

(a)

(b)
In order to verify the advantage of SPR index, this paper uses WAFM maximum principle, AFM maximum principle, and kurtosis maximum principle to select sensitive IMFs.
The two sensitive IMFs with the maximum WAFM value selected by WAFM maximum principle and AFM maximum principle are IMF2 (Figure 21) and IMF6 (Figure 18(e)), respectively. IMF6 can observe the fault characteristic frequency of the outer ring, which is the same as the IMF selected by SPR, while IMF2 cannot observe the fault characteristic frequency.

The two sensitive IMFs with kurtosis maximum principle are IMF2 (Figure 21) and IMF4 (Figure 22), respectively. IMF2 cannot observe the fault characteristic frequency, and the fault frequency of inner and outer ring can be found in the envelope spectrum of IMF4, but there are many interferences, and the effect is not as good as that of the method in this paper.

6. Conclusions
(1)This paper proposes a method for improving VMD and automatically acquiring the sensitive IMF. The proposed strategy solves the problem of setting parameter combination for the conventional VMD method.(2)The weighted AFM indicator constructed in this paper takes into account the impulsive and periodic nature of fault signals and makes it possible to search for the optimal VMD parameter set using the IPSO algorithm. Moreover, the proposed method has the benefit of adaptive parameter acquisition and the decomposition result was found to be superior compared to that of other methods such as EMD and EEMD.(3)In view of the differences between single-failure faults and compound faults, this paper proposes using two different principles (principle of maximum weighted AFM and principle of maximum SPR) for selecting the sensitive IMF, thereby solving the difficult problem of accurately detecting compound faults. The proposed method could potentially provide technical support for engineering applications.
Data Availability
The data for this article are available from the public dataset website: http://ides.nuaa.edu.cn.
Conflicts of Interest
The authors declare that there are no conflicts of interest regarding the publication of this paper.
Acknowledgments
This research was supported by the Joint Funds of the National Natural Science Foundation of China (U1733201 and U1933202) and the Fundamental Research Funds for the Central Universities (NJ2019014).