A Multi-Fault Diagnosis Method for Cylindrical Roller Bearings Based on RSNGO-Optimized VMD and CNN-BiLSTM-SAT
Abstract
1. Introduction
2. Basic Theory
2.1. Northern Goshawk Algorithm Combining Refractive Reverse Learning and Sine–Cosine
2.2. Variational Mode Decomposition
2.3. Convolutional Neural Network
2.4. Bidirectional Long Short-Term Memory
2.5. Self-Attention
3. The RSNGO Algorithm and the Multi-Fault Identification Model for Bearings
3.1. Comparative Performance Analysis of Different Metaheuristic Optimization Algorithms
3.2. Structural Design of a Multi-Fault Diagnosis Model
3.3. Model Diagnostic Process
- (1)
- The collected bearing vibration signals were first preprocessed to construct the sample dataset. For raw signals under different health conditions, variational mode decomposition (VMD) was employed for signal decomposition. The RSNGO algorithm was utilized to adaptively optimize the VMD parameters, including the decomposition level K and the penalty factor α. The parameter search ranges, fitness function, and constraint conditions were kept consistent with those described in the previous section. Subsequently, the optimal intrinsic mode function (IMF) components were selected based on the minimum envelope entropy criterion, and ten time-domain statistical features were extracted to establish the feature sample set.
- (2)
- A CNN–BiLSTM–SAT network model was developed for bearing fault classification. The RSNGO algorithm was further employed to optimize key hyperparameters of the network, including the kernel size and the number of filters in the first CNN layer, the number of hidden neurons in the BiLSTM layer, and the initial learning rate. The network weights and biases were initialized, and corresponding parameter search ranges were defined. Classification accuracy was adopted as the fitness function for the optimization process. Through iterative training and parameter optimization, the optimal CNN–BiLSTM–SAT model was obtained.
- (3)
- The constructed sample dataset was divided into training and testing subsets. Based on the optimized model configuration, the training set was used for model learning, while the testing set was employed to evaluate the diagnostic performance. Finally, the identification results of multiple bearing fault modes were obtained and analyzed.
4. Experiment Design and Model Performance Analysis
4.1. Design of Experiments
4.2. Data Processing and Dataset Construction
4.3. Optimal Model Parameter Configuration
4.4. Analysis of Model Diagnosis Results
4.5. Comparison of Model Experiment Analysis
4.6. Model Diagnosis Results Under Different Optimization Algorithms
4.7. Ablation Analysis of Optimization Strategies
4.8. Robustness Test of the Model
4.9. Adaptability Verification of the Diagnostic Model
- (1)
- Self-acquired test analysis under other operating conditions
- (2)
- CWRU bearing fault dataset analysis
5. Conclusions
- (1)
- Compared with the NGO, PSO, WOA, and DBO algorithms, the proposed RSNGO algorithm achieves a faster convergence rate and stronger global search capability. More importantly, this optimization advantage effectively translates into enhanced feature discrimination and improved diagnostic robustness. RSNGO-VMD is employed to decompose the original vibration signals and select the optimal IMF components, which effectively suppresses environmental noise interference. Furthermore, RSNGO is utilized to optimize the hyperparameters of the CNN-BiLSTM-SAT model, enabling the construction of an optimal network architecture and effectively overcoming the limitations of traditional convolutional neural networks.
- (2)
- Compared with other benchmark models, the proposed method achieves more than 99% in key evaluation metrics, including macro-precision, macro-recall, and macro-F1 score, and exhibits significantly higher fault recognition accuracy. In addition, compared with other optimization algorithms, the RSNGO-optimized network model demonstrates superior recognition performance. The t-SNE visualization results further reveal that different fault categories exhibit clear clustering characteristics in the feature space, confirming the effectiveness of the proposed model in multi-fault pattern recognition.
- (3)
- Under noisy conditions, the proposed model exhibits strong robustness. Even at an SNR of −3 dB, the average recognition accuracy remains above 95%, demonstrating excellent noise resistance. Under different self-acquired experimental conditions, the model achieves an average recognition accuracy exceeding 99%. Furthermore, on the CWRU bearing fault dataset, the proposed model maintains high identification accuracy under all four rotational speeds, further verifying its strong adaptability to different operating conditions.
Author Contributions
Funding
Data Availability Statement
Conflicts of Interest
References
- Chen, X.; Yang, R.; Xue, Y.; Huang, M.; Ferrero, R.; Wang, Z. Deep transfer learning for bearing fault diagnosis: A systematic review since 2016. IEEE Trans. Instrum. Meas. 2023, 72, 1–21. [Google Scholar]
- Barai, V.; Ramteke, S.M.; Dhanalkotwar, V.; Nagmote, Y.; Shende, S.; Deshmukh, D.S. Bearing Fault Diagnosis Using Signal Processing and Machine Learning Techniques: A Review. In IOP Conference Series: Materials Science and Engineering; IOP Publishing: Bristol, UK, 2022; Volume 1259, p. 012034. [Google Scholar]
- Dragomiretskiy, K.; Zosso, D. Variational mode decomposition. IEEE Trans. Signal Process. 2013, 62, 531–544. [Google Scholar]
- Zhou, Z.; Chen, W.; Yang, C. Adaptive range selection for parameter optimization of VMD algorithm in rolling bearing fault diagnosis under strong background noise. J. Mech. Sci. Technol. 2023, 37, 5759–5773. [Google Scholar] [CrossRef]
- Li, M.; Yan, C.; Liu, W.; Liu, X.; Zhang, M.; Xue, J. Fault diagnosis model of rolling bearing based on parameter adaptive AVMD algorithm. Appl. Intell. 2022, 53, 3150–3165. [Google Scholar] [CrossRef]
- Zhou, J.B.; Xiao, M.H.; Niu, Y.; Ji, G.J. Rolling Bearing Fault Diagnosis Based on WGWOA-VMD-SVM. Sensors 2022, 22, 6281. [Google Scholar] [CrossRef] [PubMed]
- Dehghani, M.; Hubálovský, Š.; Trojovský, P. Northern goshawk optimization: A new swarm-based algorithm for solving optimization problems. IEEE Access 2021, 9, 162059–162080. [Google Scholar] [CrossRef]
- Liang, Y.; Hu, X.; Hu, G.; Dou, W. An enhanced northern goshawk optimization algorithm and its application in practical optimization problems. Mathematics 2022, 10, 4383. [Google Scholar] [CrossRef]
- Huang, Y.J.; Liao, A.H.; Hu, D.Y.; Shi, W.; Zheng, S.B. Multi-scale convolutional network with channel attention mechanism for rolling bearing fault diagnosis. Measurement 2022, 203, 111935. [Google Scholar] [CrossRef]
- Song, B.; Liu, Y.; Fang, J.; Liu, W.; Zhong, M.; Liu, X. An optimized CNN-BiLSTM network for bearing fault diagnosis under multiple working conditions with limited training samples. Neurocomputing 2024, 574, 127284. [Google Scholar] [CrossRef]
- Huang, D.; Su, X.; Yang, J.; Du, S.; Wang, D.; Ran, Q. An improved dual-channel CNN-BILSTM fusion attention model for fault diagnosis of aero-engine bearings. Measurement 2025, 253, 117761. [Google Scholar] [CrossRef]
- Siddique, M.F.; Saleem, F.; Umar, M.; Kim, C.H.; Kim, J.-M. A hybrid deep learning approach for bearing fault diagnosis using continuous wavelet transform and attention-enhanced spatiotemporal feature extraction. Sensors 2025, 25, 2712. [Google Scholar] [CrossRef] [PubMed]
- Fu, G.; Wei, Q.; Yang, Y. Bearing fault diagnosis with parallel CNN and LSTM. Math. Biosci. Eng. 2024, 21, 2385. [Google Scholar] [CrossRef] [PubMed]
- Han, K.; Wang, W.; Guo, J. Research on a bearing fault diagnosis method based on a CNN-LSTM-GRU model. Machines 2024, 12, 927. [Google Scholar] [CrossRef]
- Deng, X.; Wang, Y.; Meng, Q. A CNN-BiLSTM Bearing Fault Diagnosis Model Based on an Improved Grey Wolf Optimization Algorithm. Eng. Lett. 2026, 34, 1874. [Google Scholar]
- Dong, Z.; Zhao, D.; Cui, L. An intelligent bearing fault diagnosis framework: One-dimensional improved self-attention-enhanced CNN and empirical wavelet transform. Nonlinear Dyn. 2024, 112, 6439–6459. [Google Scholar] [CrossRef]
- Zhao, D.; Tian, C.; Fu, Z.; Zhong, Y.; Hou, J.; He, W. Multi scale convolutional neural network combining BiLSTM and attention mechanism for bearing fault diagnosis under multiple working conditions. Sci. Rep. 2025, 15, 13035. [Google Scholar] [CrossRef] [PubMed]
- Zhang, Y.; Lin, L.; Wang, J.; Zhang, W.; Gao, S.; Zhang, Z. Attention activation network for bearing fault diagnosis under various noise environments. Sci. Rep. 2025, 15, 977. [Google Scholar] [CrossRef] [PubMed]
- Wang, Y.; Li, D.; Li, L.; Sun, R.; Wang, S. A novel deep learning framework for rolling bearing fault diagnosis enhancement using VAE-augmented CNN model. Heliyon 2024, 10, e35407. [Google Scholar] [CrossRef] [PubMed]
- Zhang, Y.; Li, S.; Zhang, A.; Li, C.; Qiu, L. A novel bearing fault diagnosis method based on few-shot transfer learning across different datasets. Entropy 2022, 24, 1295. [Google Scholar] [CrossRef] [PubMed]
- Wang, H.; Wu, Z.; Rahnamayan, S.; Liu, Y.; Ventresca, M. Enhancing particle swarm optimization using generalized opposition-based learning. Inf. Sci. 2011, 181, 4699–4714. [Google Scholar] [CrossRef]
- Mirjalili, S. SCA: A sine cosine algorithm for solving optimization problems. Knowl.-Based Syst. 2016, 96, 120–133. [Google Scholar] [CrossRef]
- Jiang, K.; Yang, Z.; Jin, T.; Chen, C.; Liu, Z.; Zhang, B. CNN-based rolling bearing fault diagnosis method with quantifiable interpretability. IEEE Trans. Instrum. Meas. 2025, 74, 1–12. [Google Scholar] [CrossRef]
- Shao, L.; Zhao, B.; Kang, X. Rolling Bearing Fault Diagnosis Based on VMD-DWT and HADS-CNN-BiLSTM Hybrid Model. Machines 2025, 13, 423. [Google Scholar] [CrossRef]
- Shen, J.; Zhou, H.; Jin, M.; Jin, Z.; Wang, Q.; Mu, Y.; Hong, Z. Rul prediction of rolling bearings based on fruit fly optimization algorithm optimized CNN-LSTM neural network. Lubricants 2025, 13, 81. [Google Scholar] [CrossRef]
- Vaswani, A.; Shazeer, N.; Parmar, N.; Uszkoreit, J.; Jones, L.; Gomez, A.N.; Kaiser, L.; Polosukhin, I. Attention is all you need. Adv. Neural Inf. Process. Syst. 2017, 30. [Google Scholar]
- Wang, Z.; Peng, Q.; Rao, W.; Li, D. An improved sparrow search algorithm with multi-strategy integration. Sci. Rep. 2025, 15, 3314. [Google Scholar] [CrossRef] [PubMed]
- Shen, J.; Wang, Z.; Wang, Y.; Zhu, H.; Zhang, L.; Tang, Y. AGWO-PSO-VMD-TEFCG-AlexNet bearing fault diagnosis method under strong noise. Measurement 2025, 242, 116259. [Google Scholar] [CrossRef]
- Wang, S.; Wang, C.; Lian, Y.; Luo, B. Research on Bearing Fault Diagnosis Based on VMD-RCMWPE Feature Extraction and WOA-SVM-Optimized Multidataset Fusion. Sensors 2025, 25, 5139. [Google Scholar] [CrossRef] [PubMed]
- Yuan, Y.; Wei, W. Fault diagnosis of wind turbine bearings based on improved dung beetle optimizer optimized LSTM. Eng. Res. Express 2025, 7, 035221. [Google Scholar] [CrossRef]
- Wang, J.; Yuan, Y.; Zheng, W.; Wan, X.; Zhang, J. Intelligent Diagnosis of Rolling Bearings Based on Parameter-Optimized Variational Mode Decomposition and BiLSTM. In 2025 International Conference on Advanced Machine Learning and Data Science (AMLDS); IEEE: New York, NY, USA, 2025; pp. 111–116. [Google Scholar]
- Ma, J.; Yu, S.; Cheng, W. Composite fault diagnosis of rolling bearing based on chaotic honey badger algorithm optimizing VMD and ELM. Machines 2022, 10, 469. [Google Scholar] [CrossRef]
- Bhende, A.R. A robust XGBoost approach for bearing fault prediction from vibration signals: Feature analysis and model evaluation. Noise Vib. Worldw. 2026, 57, 120–129. [Google Scholar] [CrossRef]
- Xiao, Z.; Cao, X.; Hao, H.; Liang, S.; Liu, J.; Li, D. A Spatio-Temporal Joint Diagnosis Framework for Bearing Faults via Graph Convolution and Attention-Enhanced Bidirectional Gated Networks. Sensors 2025, 25, 3908. [Google Scholar] [CrossRef] [PubMed]
- Bensaoucha, S.; Gharib, G.M.; Al Soudi, M.; Teta, A.; Benzita, S.; Bertal, A.S.; Guiatni, A.; Rabehi, A.; Benghanem, M. Robust Bearing Fault Detection and Classification Using Deep Neural Networks: A Comprehensive Study on the CWRU Dataset. J. Eur. Syst. Autom. 2025, 58, 2435–2443. [Google Scholar] [CrossRef]





















| Function Name | Formula | Search Area | Dimension |
|---|---|---|---|
| Sphere Function | [−100, 100] | 30 | |
| Schwefel 1.2 | [−100, 100] | 30 | |
| Quartic Function | [−1.28, 1.28] | 30 | |
| Ackley’s Function | [−32, 32] | 30 |
| Faulty Bearing Type | NF205EM | NJ205EM |
|---|---|---|
| Bearing outside diameter/mm | 52 | 52 |
| Bearing bore diameter/mm | 25 | 25 |
| Width/mm | 15 | 15 |
| Rolling element diameter/mm | 7.49 | 7.49 |
| Number of rolling elements | 13 | 13 |
| Contact angle β/(°) | 0 | 0 |
| Manufacturer | HRB | HRB |
| Bearing clearance | Normal radial internal | Normal radial internal clearance |
| Label | Bearing Health Condition | Rotate Speed/(r/min) | Torque/(N·m) | Fault Degree/(Width × Depth/mm) | Number of Samples Under Different Bearing Conditions |
|---|---|---|---|---|---|
| 1 | Normal | 1511 | 11 | — | 126 |
| 2 | Inner ring fault | 1511 | 11 | 0.5 × 0.4 | 126 |
| 3 | Rolling element fault | 1511 | 11 | 0.5 × 0.4 | 126 |
| 4 | Outer ring fault | 1511 | 11 | 0.5 × 0.4 | 126 |
| 5 | Combination faults | 1511 | 11 | 0.5 × 0.4 | 126 |
| Bearing Health Condition | Selected IMF Index | Decomposition Levels/K | Penalty Factor/α | Minimum Envelope Entropy |
|---|---|---|---|---|
| Normal | 3 | 5 | 1433 | 7.29 |
| Inner ring fault | 4 | 10 | 137 | 7.18 |
| Rolling element fault | 2 | 4 | 655 | 7.32 |
| Outer ring fault | 10 | 10 | 899 | 7.23 |
| Combination faults | 5 | 10 | 2500 | 7.3 |
| Parameter | Value | Parameter | Value |
|---|---|---|---|
| 1CNN__kernel_size | 6 | Learning rate | 0.008 |
| 1CNN__kernel_number | 16 | Epoch | 120 |
| 2CNN__kernel_size | 2 | Batch size | 64 |
| 2CNN__kernel_number | 8 | Loss function | Cross Entropy Loss |
| BiLSTM_hidden_size | 32 | Activation function | Leaky ReLU |
| SAT_head | 1 | Optimizer | Adam |
| SAT_key | 50 | SAT_value_channel | 50 |
| Model | VMD Parameter Optimization | Network Parameter Optimization |
|---|---|---|
| M1 | Fixed VMD | Fixed CNN-BiLSTM-SAT |
| M2 | RSNGO-VMD | Fixed CNN-BiLSTM-SAT |
| M3 | Fixed VMD | RSNGO-CNN-BiLSTM-SAT |
| M4 | RSNGO-VMD | RSNGO-CNN-BiLSTM-SAT |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2026 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license.
Share and Cite
Chen, L.; Li, Z.; Tan, A.; Li, Y.; Jia, D.; Yang, F.; Zhong, Z. A Multi-Fault Diagnosis Method for Cylindrical Roller Bearings Based on RSNGO-Optimized VMD and CNN-BiLSTM-SAT. Machines 2026, 14, 901. https://doi.org/10.3390/machines14080901
Chen L, Li Z, Tan A, Li Y, Jia D, Yang F, Zhong Z. A Multi-Fault Diagnosis Method for Cylindrical Roller Bearings Based on RSNGO-Optimized VMD and CNN-BiLSTM-SAT. Machines. 2026; 14(8):901. https://doi.org/10.3390/machines14080901
Chicago/Turabian StyleChen, Lihai, Zhenshui Li, Ao Tan, Yican Li, Dong Jia, Fang Yang, and Zhidan Zhong. 2026. "A Multi-Fault Diagnosis Method for Cylindrical Roller Bearings Based on RSNGO-Optimized VMD and CNN-BiLSTM-SAT" Machines 14, no. 8: 901. https://doi.org/10.3390/machines14080901
APA StyleChen, L., Li, Z., Tan, A., Li, Y., Jia, D., Yang, F., & Zhong, Z. (2026). A Multi-Fault Diagnosis Method for Cylindrical Roller Bearings Based on RSNGO-Optimized VMD and CNN-BiLSTM-SAT. Machines, 14(8), 901. https://doi.org/10.3390/machines14080901

