Bearing Fault Classification of Induction Motors Using Discrete Wavelet Transform and Ensemble Machine Learning Algorithms

Bearing fault diagnosis at early stage is very significant to ensure seamless operation of induction motors in industrial environment. The identification and classification of faults helps to undertook maintenance operation in an efficient manner. This paper presents an ensemble machine learning-based fault classification scheme for induction motors (IMs) utilizing the motor current signal that uses the discrete wavelet transform (DWT) for feature extraction. Three wavelets (db4, sym4, and Haar) are used to decompose the current signal, and several features are extracted from the decomposed coefficients. In the pre-processing stage, notch filtering is used to remove the line frequency component to improve classification performance. Finally, the two ensemble machine learning (ML) classifiers random forest (RF) and extreme gradient boosting (XGBoost) are trained and tested using the extracted feature set to classify the bearing fault condition. Both classifier models demonstrate very promising results in terms of accuracy and other accepted performance indicators. Our proposed method achieves an accuracy slightly greater than 99%, which is better than other models examined for the same dataset.