journal article·2026-07-28

Artificial Intelligence-Driven Quality Control in Mechanical Manufacturing: Vibration-Based Multiclass Gear Fault Detection Using LightGBM

Peter Malega, Juraj Kováč, Róbert Munkáči, Jozef Svetlík

Applied Sciences· Volume 16· Issue 15
Publisher page

Abstract

Artificial intelligence is increasingly used to improve industrial quality control, but its practical value depends on whether models remain accurate under different operating conditions and fault classes. This study evaluates an artificial-intelligence-based workflow for gear quality control using vibration signals measured on a real two-stage reduction gearbox. Two orthogonal vibration channels were analyzed for six health states, three shaft speeds, and two load levels. Because the time-series data were only partly stationary, the dataset was divided chronologically into training and test segments. A 54-feature representation was built from rolling-window statistics and operating variables, and six classifiers were compared: Light Gradient Boosting Machine (LGBM), Extreme Gradient Boosting (XGBM), random forest, decision tree, multilayer perceptron (MLP), and logistic regression. LGBM achieved the best overall accuracy (0.9728) while maintaining substantially lower training time than several competing nonlinear models. Class-wise precision, recall, and F1-score ranged from 0.95 to 1.00, and the nominal response time for most operating-condition transitions was approximately 0.0998 s. The results show that vibration-based machine learning can support robust, near-real-time fault identification in mechanical manufacturing environments. The study also highlights the importance of chronological validation, feature engineering over multiple time windows, and the trade-off between predictive performance and deployment efficiency. Because the validation dataset originates from one gearbox platform, the results should be interpreted as promising internal evidence rather than as proof of universal industrial robustness.
Source: crossref · Quality: baseline

Affiliations

Department of Industrial and Digital Engineering, Faculty of Mechanical Engineering, Technical University of Košice, Park Komenského 9, 042 00 Košice, Slovakia

Department of Manufacturing Technology and Robotics, Faculty of Mechanical Engineering, Technical University of Košice, Letná 9, 042 00 Košice, Slovakia

Funding

KEGA

References (26)

Show source-provided references
1. Kotecha, K., Piuri, V., Shah, H., and Patel, R. (2021). Artificial Intelligence: Prospect in Mechanical Engineering Field—A Review. Data Science and Intelligent Applications, Springer.10.1007/978-981-15-4474-3 ↗
2. Applications of Artificial Intelligence in Mechanical Engineering
3. A Review on Machinery Diagnostics and Prognostics Implementing Condition-Based Maintenance10.1016/j.ymssp.2005.09.012 ↗
4. Applications of Machine Learning to Machine Fault Diagnosis: A Review and Roadmap10.1016/j.ymssp.2019.106587 ↗
5. Dutta, N., Kaliannan, P., and Subramaniam, U. (2022, January 1–3). Bearing Fault Detection for Water Pumping System Using Artificial Neural Network. Proceedings of the 2022 Second International Conference on Power, Control and Computing Technologies (ICPC2T), Raipur, India.10.1109/ICPC2T53885.2022.9776994 ↗
6. Recent Advances in Vibration Signal Processing Techniques for Gear Fault Detection—A Review10.4028/www.scientific.net/AMM.430.78 ↗
7. Detection and Advancement Monitoring of Distributed Pitting Failure in Gears10.1007/s10921-010-0066-4 ↗
8. Diagnosis of Localized Faults in Multistage Gearboxes: A Vibrational Approach by Means of Automatic EMD-Based Algorithm
9. Xie, F., Wang, G., Shang, J., Sun, E., and Xie, S. (2023). Gearbox Fault Diagnosis Based on Multi-Sensor Deep Spatiotemporal Feature Representation. Mathematics, 11.10.3390/math11122679 ↗
10. Hu, Q., Zhou, H., Wang, C., Zhu, C., Shen, J., and He, P. (2023). Time-Frequency Fusion Features-Based GSWOA-KELM Model for Gear Fault Diagnosis. Lubricants, 12.10.3390/lubricants12010010 ↗
11. Robust Deep Learning-Based Fault Detection of Planetary Gearbox Using Enhanced Health Data Map under Domain Shift Problem
12. Robust Vibration-Based Gearbox Fault Diagnosis: A Review of Transfer Learning, Domain Adaptation, Continual Learning, and Self-Supervised Methods10.1007/s42417-026-02458-z ↗
13. Nguyen, V.-T., and Diep, Q.B. (2025). Vibration-Based Gearbox Fault Diagnosis Using a Multi-Scale Convolutional Neural Network with Depth-Wise Feature Concatenation. PLoS ONE, 20.10.1371/journal.pone.0324905 ↗
14. Gear Fault Identification Using Sensor Fusion and Ensemble Machine Learning: Development and Validation of an Industry Ready Pre-Trained Model10.1007/s40032-025-01280-y ↗
15. Diebolt, C., and Kyrtsou, C. (2005). Non-Stationarity Tests in Macroeconomic Time Series. New Trends in Macroeconomics, Springer.10.1007/3-540-28556-3 ↗
16. Dama, F., and Sinoquet, C. (2021). Analysis and Modeling to Forecast in Time Series: A Systematic Review. arXiv.
17. Vabalas, A., Gowen, E., Poliakoff, E., and Casson, A.J. (2019). Machine Learning Algorithm Validation with a Limited Sample Size. PLoS ONE, 14.10.1371/journal.pone.0224365 ↗
18. Géron, A. (2019). Hands-On Machine Learning with Scikit-Learn, Keras, and TensorFlow, O’Reilly Media. [2nd ed.].
19. Evaluating the Impact of Data Transformation Techniques on the Performance and Interpretability of Software Defect Prediction Models10.1049/2023/6293074 ↗
20. Basic Tenets of Classification Algorithms K-Nearest-Neighbor, Support Vector Machine, Random Forest and Neural Network: A Review
21. Implementation of Machine-Learning Classification in Remote Sensing: An Applied Review10.1080/01431161.2018.1433343 ↗
22. Li, H., Song, J., Xue, M., Zhang, H., Ye, J., Cheng, L., and Song, M. (2022). A Survey of Neural Trees. arXiv.
23. Ke, G., Meng, Q., Finley, T., Wang, T., Chen, W., Ma, W., Ye, Q., and Liu, T.-Y. (2017). LightGBM: A Highly Efficient Gradient Boosting Decision Tree. Advances in Neural Information Processing Systems 30, Curran Associates, Inc.
24. Chen, T., and Guestrin, C. (2016, January 13–17). XGBoost: A Scalable Tree Boosting System. Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, San Francisco, CA, USA.10.1145/2939672.2939785 ↗
25. Random Forests10.1023/A:1010933404324 ↗
26. A Unified Approach to Interpreting Model Predictions