An Intelligent Fault Diagnosis Model for Rolling Bearings Based on IGTO-Optimized VMD and LSTM Networks

To address the issue of rolling bearing fault diagnosis, this paper proposes a novel model combining the Improved Gorilla Troop Optimization (IGTO) algorithm, Variational Mode Decomposition (VMD), Permutation Entropy (PE), and Long Short-Term Memory (LSTM) networks. The IGTO algorithm is used to optimize the parameters of VMD and LSTM, enhancing signal decomposition and feature extraction. The proposed model achieves fault classification accuracies of 96.67% and 98.96% in the testing and training phases, respectively, on the Case Western Reserve University dataset, with minimal accuracy fluctuations. Furthermore, on the Jiangnan University dataset, the model reaches an average testing accuracy of 98.85%, with the highest accuracy reaching 99.48%. The results also demonstrate high stability, as indicated by low standard deviations (1.2148 and 1.3217) and narrow 95% confidence intervals ([95.75%, 97.58%] and [96.73%, 97.49%]). Despite a longer average runtime of 13.88 s per sample, the model’s superior accuracy justifies the computational cost. These results demonstrate the model’s excellent diagnostic performance, adaptability to different datasets, and practical applicability for rolling bearing fault diagnosis. This approach provides a valuable reference for predictive maintenance and fault detection systems in industrial applications.