{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,4,21]],"date-time":"2026-04-21T12:29:44Z","timestamp":1776774584596,"version":"3.51.2"},"reference-count":33,"publisher":"MDPI AG","issue":"9","license":[{"start":{"date-parts":[[2025,9,12]],"date-time":"2025-09-12T00:00:00Z","timestamp":1757635200000},"content-version":"vor","delay-in-days":0,"URL":"https:\/\/creativecommons.org\/licenses\/by\/4.0\/"}],"funder":[{"name":"Guangzhou Railway Polytechnic","award":["GTXYR2431"],"award-info":[{"award-number":["GTXYR2431"]}]}],"content-domain":{"domain":[],"crossmark-restriction":false},"short-container-title":["Algorithms"],"abstract":"<jats:p>Rolling bearing failures cause significant production downtime and economic losses. Traditional diagnostic methods suffer from low efficiency, suboptimal accuracy, and susceptibility to human subjectivity. To address these limitations, this paper proposes a novel bearing fault diagnosis (BFD) approach leveraging a Gated Recurrent Unit (GRU) network. Key contributions include: (1) Employing Bayesian optimization to automate the search for the optimal GRU architecture (layers, hidden units) and hyperparameters (learning rate, batch size, epochs), significantly enhancing diagnostic performance (achieving 97.9% accuracy). (2) Integrating a self-attention mechanism to further improve the GRU\u2019s feature extraction capability from vibration signals, boosting accuracy to 99.6%. (3) Demonstrating the robustness of the optimized GRU with self-attention across varying motor speeds (1772 rpm, 1750 rpm, 1730 rpm), consistently maintaining diagnostic accuracy above 97%. Comparative studies with Bayesian-optimized Long Short-Term Memory (LSTM) and Convolutional Neural Network (CNN) models confirm the superior accuracy (97.9% vs. 95.1% and 90.0%) and faster inference speed (0.27 s) of the proposed GRU-based method. The results validate that the combination of Bayesian optimization, GRU, and self-attention provides an efficient, accurate, and robust intelligent solution for automated BFD.<\/jats:p>","DOI":"10.3390\/a18090576","type":"journal-article","created":{"date-parts":[[2025,9,12]],"date-time":"2025-09-12T07:52:28Z","timestamp":1757663548000},"page":"576","update-policy":"https:\/\/doi.org\/10.3390\/mdpi_crossmark_policy","source":"Crossref","is-referenced-by-count":1,"title":["Optimized GRU with Self-Attention for Bearing Fault Diagnosis Using Bayesian Hyperparameter Tuning"],"prefix":"10.3390","volume":"18","author":[{"ORCID":"https:\/\/orcid.org\/0000-0002-7331-7987","authenticated-orcid":false,"given":"Zongchao","family":"Liu","sequence":"first","affiliation":[{"name":"College of Finance and Commerce, Guangzhou Railway Polytechnic, Guangzhou 511300, China"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"ORCID":"https:\/\/orcid.org\/0000-0003-1703-3362","authenticated-orcid":false,"given":"Shuai","family":"Teng","sequence":"additional","affiliation":[{"name":"School of Intelligent Construction and Civil Engineering, Zhongyuan University of Technology, Zhengzhou 450007, China"}],"role":[{"role":"author","vocabulary":"crossref"}]},{"given":"Shaodi","family":"Wang","sequence":"additional","affiliation":[{"name":"Earthquake Engineering Research & Test Center, Guangzhou University, Guangzhou 510006, China"}],"role":[{"role":"author","vocabulary":"crossref"}]}],"member":"1968","published-online":{"date-parts":[[2025,9,12]]},"reference":[{"key":"ref_1","doi-asserted-by":"crossref","first-page":"103872","DOI":"10.1016\/j.micpro.2021.103872","article-title":"Derivative and enhanced discrete analytic wavelet algorithm for rolling bearing fault diagnosis","volume":"82","author":"Li","year":"2021","journal-title":"Microprocess. Microsyst."},{"key":"ref_2","doi-asserted-by":"crossref","first-page":"045102","DOI":"10.1088\/1361-6501\/ad1872","article-title":"Bearing fault feature extraction method: Stochastic resonance-based negative entropy of square envelope spectrum","volume":"35","author":"Zhao","year":"2024","journal-title":"Meas. Sci. Technol."},{"key":"ref_3","doi-asserted-by":"crossref","first-page":"015039","DOI":"10.1088\/2631-8695\/ad0f79","article-title":"Bearing fault diagnosis based on the feature enhancement of improved mean differential SDP images","volume":"6","author":"Wang","year":"2024","journal-title":"Eng. Res. Express"},{"key":"ref_4","first-page":"9412787","article-title":"Rolling Bearing Diagnosing Method Based on Time Domain Analysis and Adaptive Fuzzy SV","volume":"2016","author":"Fu","year":"2016","journal-title":"Shock Vib."},{"key":"ref_5","first-page":"30","article-title":"Doubly-fed Induction Generator Stator Fault Diagnosis Based on Rotor Instantaneous Power Spectrum","volume":"38","author":"Ma","year":"2014","journal-title":"Dianli Xitong Zidonghua\/Autom. Electr. Power Syst."},{"key":"ref_6","first-page":"13","article-title":"Gear Fault Diagnosis Method based on Feature-enhanced Cepstrum Analysis","volume":"43","author":"Jiang","year":"2019","journal-title":"J. Mech. Transm."},{"key":"ref_7","doi-asserted-by":"crossref","first-page":"395","DOI":"10.1016\/j.jsv.2016.07.013","article-title":"Amplitude and frequency demodulation analysis for fault diagnosis of planet bearings","volume":"382","author":"Feng","year":"2016","journal-title":"J. Sound Vib."},{"key":"ref_8","first-page":"1688","article-title":"Fault Diagnosis of Planetary Gearboxes Based on NLSTFT Order Tracking under Variable Speed Conditions","volume":"29","author":"Wang","year":"2018","journal-title":"China Mech. Eng."},{"key":"ref_9","doi-asserted-by":"crossref","first-page":"761","DOI":"10.1007\/s42417-019-00189-y","article-title":"Vibration Analysis of a Rotating Flywheel\/Flexible Coupling System with Angular Misalignment and Rubbing Using Smoothed Pseudo Wigner\u2013Ville Distributions","volume":"8","author":"Liu","year":"2020","journal-title":"J. Vib. Eng. Technol."},{"key":"ref_10","first-page":"2193","article-title":"Fault diagnosis of planetary gearbox based on improved empirical wavelet transform","volume":"37","author":"Zhu","year":"2016","journal-title":"Chin. J. Sci. Instrum."},{"key":"ref_11","doi-asserted-by":"crossref","first-page":"4869","DOI":"10.1007\/s12206-014-1107-1","article-title":"Combined rotor fault diagnosis in rotating machinery using empirical mode decomposition","volume":"28","author":"Singh","year":"2014","journal-title":"J. Mech. Sci. Technol."},{"key":"ref_12","doi-asserted-by":"crossref","unstructured":"Shen, Z., Kong, X., Cheng, L., Wang, R., and Zhu, Y. (2024). Fault Diagnosis of the Rolling Bearing by a Multi-Task Deep Learning Method Based on a Classifier Generative Adversarial Network. Sensors, 24.","DOI":"10.3390\/s24041290"},{"key":"ref_13","doi-asserted-by":"crossref","first-page":"056204","DOI":"10.1088\/1361-6501\/ad2828","article-title":"Adversarial training of multi-scale channel attention network for enhanced robustness in bearing fault diagnosis","volume":"35","author":"Peng","year":"2024","journal-title":"Meas. Sci. Technol."},{"key":"ref_14","first-page":"21","article-title":"Fault Diagnosis of Rolling Bearing Based on KNN-Naive Bayesian Algorithm","volume":"26","author":"Lu","year":"2018","journal-title":"Comput. Meas. Control"},{"key":"ref_15","first-page":"31","article-title":"Bearing fault diagnosis based on the deep learning feature extractionand WOA SVM state recognition","volume":"38","author":"Zhao","year":"2019","journal-title":"J. Vib. Shock"},{"key":"ref_16","first-page":"883","article-title":"Rolling bearing fault identification based on convolution neural network and discrete wavelet transform","volume":"31","author":"Chen","year":"2018","journal-title":"Zhendong Gongcheng Xuebao\/J. Vib. Eng."},{"key":"ref_17","doi-asserted-by":"crossref","first-page":"41","DOI":"10.4236\/jcc.2018.67005","article-title":"Application of Improved Deep Auto-Encoder Network in Rolling Bearing Fault Diagnosis","volume":"6","author":"Di","year":"2018","journal-title":"J. Comput. Commun."},{"key":"ref_18","doi-asserted-by":"crossref","first-page":"436","DOI":"10.1038\/nature14539","article-title":"Deep learning","volume":"521","author":"Lecun","year":"2015","journal-title":"Nature"},{"key":"ref_19","doi-asserted-by":"crossref","first-page":"504","DOI":"10.1126\/science.1127647","article-title":"Reducing the Dimensionality of Data with Neural Networks","volume":"313","author":"Hinton","year":"2006","journal-title":"Science"},{"key":"ref_20","doi-asserted-by":"crossref","first-page":"7322","DOI":"10.1177\/09544062231222806","article-title":"Bearing fault diagnosis based on variational autoencoder and non-local block wide kernel convolutional neural network","volume":"238","author":"Jiang","year":"2024","journal-title":"Proc. Inst. Mech. Eng. Part C J. Mech. Eng. Sci."},{"key":"ref_21","doi-asserted-by":"crossref","first-page":"439","DOI":"10.1016\/j.cja.2019.07.011","article-title":"A new bearing fault diagnosis method based on modified convolutional neural networks","volume":"33","author":"Zhang","year":"2020","journal-title":"Chin. J. Aeronaut."},{"key":"ref_22","first-page":"8837958","article-title":"Development of Deep Convolutional Neural Network with Adaptive Batch Normalization Algorithm for Bearing Fault Diagnosis","volume":"2020","author":"Fu","year":"2020","journal-title":"Shock Vib."},{"key":"ref_23","doi-asserted-by":"crossref","first-page":"4041","DOI":"10.1007\/s12652-021-03177-x","article-title":"A bearing fault diagnosis model based on CNN with wide convolution kernels","volume":"13","author":"Song","year":"2022","journal-title":"J. Ambient. Intell. Humaniz. Comput."},{"key":"ref_24","doi-asserted-by":"crossref","first-page":"6687331","DOI":"10.1155\/2021\/6687331","article-title":"Deep Transfer Learning Method Based on 1D-CNN for Bearing Fault Diagnosis","volume":"2021","author":"He","year":"2021","journal-title":"Shock Vib."},{"key":"ref_25","doi-asserted-by":"crossref","first-page":"046006","DOI":"10.1088\/1361-6501\/ad2052","article-title":"Transfer learning rolling bearing fault diagnosis model based on deep feature decomposition and class-level alignment","volume":"35","author":"Dong","year":"2024","journal-title":"Meas. Sci. Technol."},{"key":"ref_26","unstructured":"Li, X., Guo, S., Sun, D., Cao, L., Li, C., Tian, S., Liu, P., and Qi, Y. (2025). A rolling bearing fault diagnosis method based on extreme learning machine optimized by improved whale optimization algorithm. Facta Univ. Ser. Mech. Eng."},{"key":"ref_27","doi-asserted-by":"crossref","first-page":"38023","DOI":"10.1109\/ACCESS.2020.2975257","article-title":"Research on a Mine Gas Concentration Forecasting Model Based on a GRU Network","volume":"8","author":"Jia","year":"2020","journal-title":"IEEE Access"},{"key":"ref_28","doi-asserted-by":"crossref","first-page":"122960","DOI":"10.1016\/j.energy.2021.122960","article-title":"Usage of correlation analysis and hypothesis test in optimizing the gated recurrent unit network for wind speed forecasting","volume":"242","author":"Wu","year":"2022","journal-title":"Energy"},{"key":"ref_29","doi-asserted-by":"crossref","first-page":"16225","DOI":"10.3390\/s150716225","article-title":"Bearing Fault Diagnosis Based on Statistical Locally Linear Embedding","volume":"15","author":"Wang","year":"2015","journal-title":"Sensors"},{"key":"ref_30","first-page":"11","article-title":"One-Dimensional Convolutional Neural Network Based Bearing Fault Diagnosis","volume":"9","author":"Chen","year":"2022","journal-title":"Open Access Libr. J."},{"key":"ref_31","doi-asserted-by":"crossref","unstructured":"Cho, K., Merrienboer, B.V., Bahdanau, D., and Bengio, Y. (2014, January 25). On the Properties of Neural Machine Translation: Encoder-Decoder Approaches. Proceedings of the SSST-8, Eighth Workshop on Syntax, Semantics and Structure in Statistical Translation, Doha, Qatar.","DOI":"10.3115\/v1\/W14-4012"},{"key":"ref_32","doi-asserted-by":"crossref","unstructured":"Wang, Y., Liao, W., and Chang, Y. (2018). Gated Recurrent Unit Network-Based Short-Term Photovoltaic Forecasting. Energies, 11.","DOI":"10.3390\/en11082163"},{"key":"ref_33","unstructured":"Jozefowicz, R., Zaremba, W., and Sutskever, I. (2015, January 7\u20139). An Empirical Exploration of Recurrent Network Architectures. Proceedings of the 32nd International Conference on Machine Learning, Lille, France."}],"container-title":["Algorithms"],"original-title":[],"language":"en","link":[{"URL":"https:\/\/www.mdpi.com\/1999-4893\/18\/9\/576\/pdf","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2025,10,9]],"date-time":"2025-10-09T18:44:19Z","timestamp":1760035459000},"score":1,"resource":{"primary":{"URL":"https:\/\/www.mdpi.com\/1999-4893\/18\/9\/576"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2025,9,12]]},"references-count":33,"journal-issue":{"issue":"9","published-online":{"date-parts":[[2025,9]]}},"alternative-id":["a18090576"],"URL":"https:\/\/doi.org\/10.3390\/a18090576","relation":{},"ISSN":["1999-4893"],"issn-type":[{"value":"1999-4893","type":"electronic"}],"subject":[],"published":{"date-parts":[[2025,9,12]]}}}