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

Fig. 1.

The developed ball bearing test station.
The developed ball bearing test station.

Fig. 2.

Confusion matrix for the cosine KNN algorithm.
Confusion matrix for the cosine KNN algorithm.

Fig. 3.

ROC curve for the cosine KNN algorithm.
ROC curve for the cosine KNN algorithm.

Fig. 4.

Confusion matrix for the quadratic SVM algorithm.
Confusion matrix for the quadratic SVM algorithm.

Fig. 5.

ROC curve for the quadratic SVM algorithm.
ROC curve for the quadratic SVM algorithm.

Fig. 6.

The optimization flow for the SVM algorithm.
The optimization flow for the SVM algorithm.

Fig. 7.

The optimization flow for the KNN algorithm.
The optimization flow for the KNN algorithm.

The optimized hyperparameters of the KNN algorithm_

HyperparameterRange
Number of neighbors1
Distance metricCorrelation
Distance weightInverse
Standardize datatrue
Accuracy100 %

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].

The optimized hyperparameters of the SVM algorithm_

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

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

SVM hyperparameter search range_

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

SVM classification_

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

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 %
Language: English
Page range: 22 - 29
Submitted on: Apr 12, 2024
|
Accepted on: Mar 7, 2025
|
Published on: Apr 12, 2025
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.