Performance evaluation of machine learning algorithms and impact of activation functions in artificial neural network classifier for bearing fault diagnosis

Abstract

Machine learning algorithms are used to identify the bearing condition. In this work, different machine learning techniques such as decision tree, logistic regression, support vector machine (SVM), and artificial neural network (ANN) are used and compared to find healthy and faulty conditions of the bearing. The identification of the condition of the bearing is based on vibrations recorded using Fast Fourier Transform Analyzer. The vibration data recorded for the bearings have been used to categorize the condition of the bearing as healthy or faulty by applying machine learning techniques. The dataset of healthy and faulty bearings is collected using a four-channel Fast Fourier Transform Analyzer (FFT) analyzer. However, the statistical feature extraction technique has been used to evaluate the accuracy and performance of artificial neural network, support vector machine, logistic regression, and decision tree algorithms based on their classification accuracy and total costs. The result of the work reveals that the performance of the activation function based artificial neural network (ANN-AF) and SVM algorithm is better than logistic regression and decision tree models. However, it is observed that the use of appropriate activation functions within the ANN-AF technique improves the accuracy of the machine learning model.