13292 FACTA UNIVERSITATIS Series: Mechanical Engineering Vol. 23, No 4, 2025, pp. 881 - 907 https://doi.org/10.22190/FUME241130008L © 2025 by University of Niš, Serbia | Creative Commons License: CC BY-NC-ND Original scientific paper A ROLLING BEARING FAULT DIAGNOSIS METHOD BASED ON EXTREME LEARNING MACHINE OPTIMIZED BY IMPROVED WHALE OPTIMIZATION ALGORITHM Xin Li1, Shiliang Guo2, Dejie Sun1, Lijun Cao1, Cong Li1, Shuyao Tian3, Peng Liu2, Yadong Qi2 1School of Mathematics and Information Science & Technology, Hebei Normal University of Science & Technology, Qinhuangdao, China 2School of Electrical Engineering, Yanshan University, Qinhuangdao, China 3College of Electronics and Control Engineering, North China Institute of Aerospace Engineering, Langfang, China ORCID iDs: Xin Li https://orcid.org/0009-0009-3792-4083 Shiliang Guo https://orcid.org/0000-0003-2033-6807 Dejie Sun https://orcid.org/0009-0006-4882-1053 Lijun Cao https://orcid.org/0009-0004-5759-9031 Cong Li https://orcid.org/0009-0005-8217-3785 Shuyao Tian https://orcid.org/0009-0006-1853-1151 Peng Liu https://orcid.org/0009-0000-9345-4260 Yadong Qi https://orcid.org/0009-0003-8268-3369 Abstract. Rolling bearing is one of the most commonly used components in rotating machinery, and researching fault diagnosis techniques for it has important practical significance. In this paper, a fault diagnosis method based on extreme learning machine optimized by improved whale optimization algorithm (IWOA-ELM) is proposed for rolling bearing vibration signals. Firstly, Variational Mode Decomposition (VMD) is used to decompose the vibration signal of the bearing, and the energy entropy is calculated to form the eigenvector. Secondly, based on the original whale optimization algorithm, a hybrid initialization population strategy is adopted to generate an initial population with a certain quality. Selecting convergence factors based on reinforcement learning to improve global search capability, and using adaptive weights and random jumps to update individual positions. In this process, the t-distribution-levy flight variation strategy is introduced to avoid being attracted by local extremum. Then, the improved whale optimization algorithm is used to optimize the input weights and hidden layer thresholds of the Extreme Learning Machine (ELM). Finally, the feature set is input into an improved ELM model for training and testing. Experiments on fault diagnosis of rolling bearings of different types and degrees have shown that the model proposed in this paper can effectively improve the accuracy of fault classification. Key words: Fault diagnosis, Variational mode decomposition, Whale optimization algorithm, Reinforcement learning Received: November 30, 2024 / Accepted February 10, 2025 Corresponding author: Shiliang Guo School of Electrical Engineering, Yanshan University, Street Address No.438, Hebei Avenue, Haigang District, Qinhuangdao 066004, China E-mail: guosl0112@ysu.edu.cn 882 X. LI, S. GUO, D. SUN, L. CAO, C. LI, S. TIAN, P. LIU, Y. QI 1. INTRODUCTION Rotating machinery plays an important role in modern society, and rolling bearings are critical components that impact its performance. However, the working environment of most rolling bearings is relatively harsh, which makes them prone to failure, resulting in significant economic losses and even catastrophic events. The fault vibration signal of rolling bearings has nonlinear and non-stationary characteristics. How to extract fault information from vibration signals and accurately locate bearing faults are essential for fault diagnosis and normal machinery operation [1-4] The application of machine learning methods in fault diagnosis can minimize manual involvement and greatly improve the accuracy of fault diagnosis, such as support vector machines (SVM) [5-7], backpropagation neural networks (BPNN) [8-10], and convolutional neural networks (CNN) [11-13]. However, these models face limitations: SVM needs to store large datasets, making it difficult to solve multi classification problems; BPNN and CNN have slow convergence speed and high computational cost, particularly with deeper layers. Extreme Learning Machine (ELM) has the characteristics of fast training speed, few learning parameters, strong generalization ability, and is widely used in the field of fault diagnosis [14-16]. Wang et al. [17] proposed a bearing fault diagnosis method based on ELM for adaptive parameter optimization, and introduced spectral cross-correlation to help select the optimal penalty factor, improving classification accuracy. In order to further enhance the generalization ability of ELM, Zhang et al. [18] used the Multitask Beetle Antennae Swarm Algorithm (MBAS) to optimize the input weights and biases of ELM, which can simultaneously reduce the number of conditions and regression errors. From the above description, it can be seen that optimizing the parameters of ELM based on optimization algorithms can further improve the performance of the model. Whale Optimization Algorithm (WOA) is a meta- heuristic algorithm developed in recent years, which is used to handle various optimization problems in different fields due to its advantage of fewer adjustable parameters [19-21]. However, as the dimensionality of the search space increases, WOA faces problems such as easily falling into local optima and slow convergence speed. Based on the above analysis, we propose a rolling bearing fault diagnosis method based on extreme learning machine optimized by improved whale optimization algorithm (IWOA-ELM), as shown in Fig. 1. Firstly, the variational mode decomposition (VMD) method is used to extract the fault signal features of rolling bearings, which are then divided into a training set and a testing set. Then, based on the WOA, a hybrid initialization population strategy, a convergence factor selection based on reinforcement learning, an adaptive weight and random jump update strategy, and a t-distribution-levy flight variation strategy are introduced to solve the problem of the algorithm falling into local optimum. Finally, the IWOA was used to optimize the parameter settings of the extreme learning machine, thereby improving the speed and accuracy of bearing fault diagnosis. The IWOA- ELM algorithm was compared with WOA-ELM, PSO-ELM, GWO-ELM, and the experimental results verified the effectiveness of the improved rolling bearing fault diagnosis method proposed in this paper. A Rolling Bearing Fault Diagnosis Method based on Extreme Learning Machine... 883 Fig. 1 Methodological framework of this study 2. SIGNAL DECOMPOSITION AND FEATURE EXTRACTION 2.1. VMD Theory VMD is an innovative and adaptive signal processing method [22] that can solve the common endpoint effects and mode component mixing problems in Empirical Mode Decomposition (EMD). VMD decomposes the initial signal into k discrete sub-signals, ensuring that each decomposition represents a finite bandwidth modal component centered at a specific frequency. At the same time, the sum of estimated bandwidths for each modality is minimized, and the sum of all modes is equal to the original signal as a constraint. The constraint variational expression can be written as:       2 j 2, 1 min ( ( ) j / )* ( ) e s.t. k k k t t k u K kk k t t u t u f    − =     +         =   (1) where, K is the number of decomposed modes. The symbols {uk} and {ωk} refer to the kth mode component and center frequency after decomposing, respectively. (t) is the Dirac function where t represents time. * is the convolution operator. Furthermore, we introduce the Lagrange multiplication operator  and the quadratic penalty factor  to find the optimal solution for the constrained variational problem, and obtain the augmented function: 884 X. LI, S. GUO, D. SUN, L. CAO, C. LI, S. TIAN, P. LIU, Y. QI    ( )   2 j 2 2 2 , , ( ( ) j / )* ( ) e ( ) ( ) ( ), ( ) ( ) k t k k t k k k k k k L u t t u t f t u t t f t u t      − =  +  + − + −    (2) where the mentioned quadratic penalty factor α is used to reduce the interference of Gaussian noise. The alternate direction method of multipliers (ADMM) iterative algorithm combined with Parseval/Plancherel and Fourier isometry transform is used to optimize each eigenmode component and center frequency, and to search for saddle points of the extended Lagrangian function. The expressions for uk, ωk, and λ change after alternating optimization iteration is as follows: 1 2 ˆ ˆˆ ˆ ( ) ( ) ( ) / 2 ( ) 1 2 ( ) in i k k k f u u         +  − +  + −  (3) 2 1 1 0 2 1 0 ˆ ( ) d ) dˆ ( n kn k n k u u        + +  +    (4) 1 1ˆˆ ˆ ˆ( ) ( ) ( ) ( )n n n k k f u      + +   + −     (5) where, γ is the noise margin, 1ˆ ( )n ku + , ˆ ( )iu  , ˆ( )f  and ˆ( )  correspond to the Fourier transforms of uk n+1(t), ui(t), f(t) and λ(t) respectively. The main iterative VMD solution process is as follows: (1) Initialize 1 1 1ˆˆ , ,k ku   and set the maximum number of iterations N, n←0; (2) Using Eqs. (3) and (4), update ûk and ωk; (3) Update ̂ using Eq. (5); (4) Accuracy convergence criterion ε>0, if n