Optimal Source Selection for Distributed Bearing Fault Classification Using Wavelet Transform and Machine Learning Algorithms

Abstract

Early and accurate detection of distributed bearing faults is essential to prevent equipment failures and reduce downtime in industrial environments. This study explores the optimal selection of input signal sources for high-accuracy distributed fault classification, employing wavelet transform and machine learning algorithms. The primary contribution of this work is to demonstrate that robust distributed bearing fault diagnosis can be achieved through optimal sensor fusion and wavelet-based feature engineering, without the need for deep learning or high-dimensional inputs. This approach provides interpretable, computationally efficient, and generalizable fault classification, setting it apart from most existing studies that rely on larger models or more extensive data. All experiments were conducted in a controlled laboratory environment across multiple loads and speeds. A comprehensive dataset, including three-axis vibration, stray magnetic flux, and two-phase current signals, was used to diagnose six distinct bearing fault conditions. The wavelet transform is applied to extract frequency-domain features, capturing intricate fault signatures. To identify the most effective input signal combinations, we systematically evaluated Random Forest, XGBoost, and Support Vector Machine (SVM) models. The analysis reveals that specific signal pairs significantly enhance classification accuracy. Notably, combining vibration signals with stray magnetic flux consistently achieved the highest performance across models, with Random Forest reaching perfect test accuracy (100%) and SVM showing robust results. These findings underscore the importance of optimal source selection and wavelet-transformed features for improving machine learning model performance in bearing fault classification tasks. While the results are promising, validation in real-world industrial settings is needed to fully assess the method's practical reliability and impact on predictive maintenance systems.