Skip to content

Machine learning-based feature-driven model generation and evaluation for multi-fault bearing diagnosis using XGBoost and time-domain statistical features of vibration data

Aug 2026 · Insight - Non-Destructive Testing and Condition Monitoring · 0 citations · 3 references

TL;DR

The findings affirm the efficacy of XGBoost in bearing fault classification and emphasise the diagnostic value of carefully selected time-domain features, as well as suggesting strong potential for deploying such models in real-time condition monitoring and predictive maintenance systems.

Abstract

Accurate and timely detection of bearing faults is critical for ensuring the reliability and safety of rotating machinery in industrial environments. This study investigates the application of machine learning (ML) techniques for classifying ten different types of bearing fault using features extracted from vibration signals. Among the various classifiers evaluated, the extreme gradient boosting (XGBoost) algorithm demonstrates superior performance. Through a combination of standard signal preprocessing, hyperparameter optimisation via GridSearchCV and comprehensive model evaluation using accuracy, F1 score and confusion matrix analysis, the optimised XGBoost model achieves a high classification accuracy of 95.48%. Notably, fault types such as IR_007_1, IR_014_1 and OR_007_6_1 are classified with perfect precision and recall. While minor misclassifications occur, primarily between ball and outer race (OR) faults due to overlapping signal characteristics, the overall confusion matrix underscores the strong generalisation and class discrimination capabilities of the model. Additionally, feature importance analysis highlights kurtosis, standard deviation (SD) and mean as the most influential statistical descriptors, with kurtosis emerging as the dominant factor for distinguishing impulsive fault characteristics. These findings affirm the efficacy of XGBoost in bearing fault classification and emphasise the diagnostic value of carefully selected time-domain features. The results suggest strong potential for deploying such models in real-time condition monitoring and predictive maintenance systems. Future research may explore integrating deep learning or hybrid ensemble methods to further enhance diagnostic accuracy in more complex or noisy fault scenarios.

View source

Similar papers

Aug 2026

Multi-Class Fault Detection and Diagnosis of Rolling Bearings: a Machine Learning Approach

Results show that using statistical vibration features with ensemble classifiers is a good way to diagnose multi-class bearing faults and establishes a comprehensive benchmark for ML- and DL-based rolling bearing FDD.

M. I. Quamar, Abdulrazaq Nafiu Abubakar, Ali Nasir · 0 citations
Open access Jul 2026

Intelligent Fault Diagnosis of Rolling Element Bearings Using Statistical Feature Extraction and Random Forest Classifier

Accurate detection of faults within rotary machinery components is vital for assuring the structural reliability of equipment in manufacturing facilities and power generation systems. Automatic fault detection is an increasingly common approach whereby data acquired from sensors are analyzed by machine learning algorithms to distinguish between normal and faulty component states. Vibration signatures recorded from healthy and faulty ball bearings, taken from the Case Western Reserve University (CWRU) 12k Drive End database, were utilized in the present study. These vibration signatures were segmented into windows of 0.1 s duration. Standard statistical time-domain features (root mean square, kurtosis, peak value and crest factor) were extracted from each sample. A random forest (RF) classifier was trained on vibration data from healthy bearings and several types of faulty bearings, partitioned into training (70% of data) and testing (30%) sets. The RF classifier distinguished healthy from faulty bearings with 98.8% accuracy across ball, inner race, outer race and normal categories. Feature importance analysis indicated that RMS (50% importance) and peak value (28% importance) were the most influential contributors, together comprising about 78% of the decision making. The RF model further recorded a recall of 1.00 for the normal baseline category, indicating that healthy bearings were identified with zero false negatives. The results demonstrate that ensemble machine learning, combined with statistically significant time-domain features, provides a highly accurate and physically interpretable solution for industrial condition monitoring, serving as a foundation for more advanced rotor-dynamic fault detection.

Manoj Adhikari, Bikki Chhantyal, Mahesh Chandra Luintel · 0 citations
Jul 2026

Machine learning-based fault detection in low-speed bearings using a multi-environmental dataset

The near-perfect linear separability indicates that the dataset’s binary, controlled-laboratory labelling rather than intrinsic bearing-degradation physics drives the clean classification, and validation on 500–1000 or more samples with progressive-degradation labelling is essential before any operational claim can be supported.

P. Pugazhendi, Vinoth Vishwanathan, Aadil Arshad Ferhath et al. · 0 citations
Conference Jul 2026

Machine learning-based bearing failure research

It is confirmed that traditional machine learning models, optimized through manual feature engineering, can provide a ‘high-precision, low-risk’ solution for bearing fault diagnosis and offers significant reference value for the intelligent operation of aero-engines and other industrial equipment.

Qianxi Ye, Pengfang Gao · 0 citations
Open access Jul 2026

Mode entropy knowledge machine: a fully automated bearing fault diagnosis model for complex operating conditions

Accurate diagnosis of rolling bearing faults is critical to the reliability of industrial equipment. However, rolling bearings often operate under complex operating conditions, and with data imbalances and noise interference, fault diagnosis of them remains extremely challenging. To address these issues, a novel mode entropy knowledge machine (MEKM) framework for robust bearing fault diagnosis is proposed in this study. For MEKM, the mode entropy space is firstly constructed to decompose the vibration signal into intrinsic mode components, and the noise-resistant feature extraction and dimensionality reduction are realized by principal component analysis. Secondly, a fast classifier based on extreme learning machines is introduced, and its parameters are automatically adjusted through a particle swarm optimization to establish an adaptive extreme learning machine diagnosis model, ensuring optimal generalization under different load and speed levels. Then, a collaborative optimization paradigm is developed to coordinate mode entropy features and classifier parameters through fully automated learning, in which entropy-driven feature characterization guides the iterative refinement of decision boundaries, while classifier feedback dynamically improves the selectivity of entropy features. Finally, validation is performed on bearings with multiple operating conditions, and the results indicated that the MEKM outperformed conventional deep learning methods in terms of diagnostic accuracy and generalization ability. The work provides a theoretical basis and an industrially feasible solution for health monitoring of mechanical equipment.

Hongchuang Tan, Yiheng Su, Jiang Ding et al. · 0 citations
Open access Aug 2026

Data-Driven Fault Diagnosis of Rolling Bearings Using Neural Networks Optimized via Bayesian, Particle Swarm, and Genetic Algorithm Methods with Time–Frequency Features

This study presents an applied comparative evaluation of automated rolling bearing fault identification using Artificial Neural Networks optimized through Bayesian Optimization, Particle Swarm Optimization, and Genetic Algorithm, highlighting optimized ANNs as competitive and computationally efficient solutions for bearing fault diagnosis.

Khoualdia Kaaïs, Khoualdia Tarek, M. Nahal · 0 citations