Skip to main content
Have a personal or library account? Click to login
Implementation and Evaluation of Machine Learning Algorithms in Ball Bearing Fault Detection Cover

Implementation and Evaluation of Machine Learning Algorithms in Ball Bearing Fault Detection

Open Access
|Apr 2025

Figures & Tables

Table 1.

Hybrid ML models.

Applied ML modelResearch description
SVM with GAA study applied SVM combined with GA to develop optimal classifiers for distinguishing healthy and faulty bearings in ASD systems, achieving 97.5 % accuracy [21].
SVM and ANN with CWTThis study explored the use of SVM and ANN alongside CWT to analyze frame vibrations during motor start-up, achieving 96.67 % accuracy with SVM and 90 % with ANN [22].
PCA and SVDDPCA and SVDD were used to predict bearing failures, achieving 93.45 % accuracy [23].
GA-based SVMA GA-based kernel discriminative feature analysis was combined with one-against-all multicategory SVMs (OAA MCSVMs) for fault diagnosis in low-speed bearings, achieving the highest reported accuracy of 98.66 % [24].
FEM and WPT with SVMA hybrid approach integrating FEM, WPT, and SVM was proposed for fault classification, achieving 81 % accuracy for inner race faults and 79 % for rolling body faults [25].
FFT-based feature extraction with SVMThe frequency domain features derived from FFT were used to train an SVM model for bearing fault classification, achieving 87.35 % accuracy [26].

[i] Note:

  • SVM - support vector machines

  • GA - genetic algorithm

  • ASD - adjustable speed drive

  • ANN - artificial neural networks

  • CWT - continuous wavelet transform

  • PCA - principal component analysis

  • SVDD - support vector data description

  • FEM - finite element method

  • OAA - one-against-all

  • MCSVM - multicategory SVM

  • WPT - wavelet packet transform

  • FFT - fast fourier transform

Fig. 1.

The developed ball bearing test station.

Table 2.

The k-nearest neighbor (KNN) classification.

Model NoModel nameDistance metricDistance weightNumber of neighborsClassification success rate
1Cosine KNNCosineEqual1098.5 %
2Coarse KNNEuclideanEqual10074 %
3Fine KNNEuclideanEqual197.4 %
4Weighted KNNEuclideanSquared inverse1097.4 %
5Medium KNNEuclideanEqual198 %
6Cubic KNNMinkowskiEqual1098.2 %
Table 3.

SVM classification.

Model NoKernel functionClassification success rate
1Linear93.9 %
2Polynomial (ρ = 2)99.5 %
3RBF99 %

[i] Note: RBF - radial basis function

Fig. 2.

Confusion matrix for the cosine KNN algorithm.

Fig. 3.

ROC curve for the cosine KNN algorithm.

Fig. 4.

Confusion matrix for the quadratic SVM algorithm.

Fig. 5.

ROC curve for the quadratic SVM algorithm.

Table 4.

SVM hyperparameter search range.

HyperparameterRange
Box constraint level0.001-1000
Kernel scale0.001-1000
Kernel functionGaussian, Linear, Quadratic, Cubic
Standardize datatrue, false
Fig. 6.

The optimization flow for the SVM algorithm.

Table 5.

The optimized hyperparameters of the SVM algorithm.

HyperparameterValue
Box constraint level977.88
Kernel scale1
Kernel functionQuadratic
Standardize datatrue
Accuracy100 %
Table 6.

KNN hyperparameter search range.

HyperparameterRange
Number of neighbors1–98
Distance metricEuclidean, Cosine, Euclidean, Correlation, Chebyshev, Hamming, Minakowski, Spearman, Jaccard, City block, Mahalanobis
Distance weightEqual, Inverse, Squared, Inverse
Standardize datatrue, false
Fig. 7.

The optimization flow for the KNN algorithm.

Table 7.

The optimized hyperparameters of the KNN algorithm.

HyperparameterRange
Number of neighbors1
Distance metricCorrelation
Distance weightInverse
Standardize datatrue
Accuracy100 %
Language: English
Page range: 22 - 29
Submitted on: Apr 12, 2024
Accepted on: Mar 7, 2025
Published on: Apr 12, 2025
Published by: Slovak Academy of Sciences, Institute of Measurement Science
In partnership with: Paradigm Publishing Services
Publication frequency: Volume open

© 2025 Pavle Stepanić, Nedeljko Dučić, Jelena Vidaković, Jelena Baralić, Marko Popović, published by Slovak Academy of Sciences, Institute of Measurement Science
This work is licensed under the Creative Commons Attribution-NonCommercial-NoDerivatives 4.0 License.