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.

Keywords

Fault diagnosis machine learning bearing vibration artificial neural network

Get full access to this article

View all access options for this article.

References

Bekar

Nyqvist

Skoogh

(2020) An intelligent approach for data pre-processing and analysis in predictive maintenance with an industrial case study. Advances in Mechanical Engineering 12(5): 1–14. DOI: 10.1177/1687814020919207.

Cui

Weng

Zhang

(2022) A feature extraction and machine learning framework for bearing fault diagnosis. Renewable Energy 191: 987–997. DOI: 10.1016/j.renene.2022.04.061.

Hakim

Omran

AAB

Ahmed

, et al. (2023) A systematic review of rolling bearing fault diagnoses based on deep learning and transfer learning: taxonomy, overview, application, open challenges, weaknesses and recommendations. Ain Shams Engineering Journal 14: 1. DOI: 10.1016/j.asej.2022.101945.

Han

Zhao

Shi

(2017) A new fault diagnosis method based on deep belief network and support vector machine with Teager–Kaiser energy operator for bearings. Advances in Mechanical Engineering 9(12): 1–11. DOI: 10.1177/1687814017743113.

Kankar

Sharma

Harsha

(2011) Fault diagnosis of ball bearings using machine learning methods. Expert Systems with Applications 38(3): 1876–1886. DOI: 10.1016/j.eswa.2010.07.119.

Khan

Ali

Farooq

, et al. (2022) Investigation of combustion performance of tannery sewage sludge using thermokinetic analysis and prediction by artificial neural network. In: Case Studies in Thermal Engineering, Vol. 40. DOI: 10.1016/j.csite.2022.102586.

Mykoniatis

(2020) A real-time condition monitoring and maintenance management system for low voltage industrial motors using internet-of-things. In: Procedia Manufacturing, pp. 450–456. Elsevier B.V. DOI: 10.1016/j.promfg.2020.02.050.

Paneiro

Rafael

(2021) Artificial neural network with a cross-validation approach to blast-induced ground vibration propagation modeling. Underground Space 6(3): 281–289. DOI: 10.1016/j.undsp.2020.03.002.

Qiu

Huang

(2019) Experimental investigation and multi-conditions identification method of centrifugal pump using Fisher discriminant ratio and support vector machine. Advances in Mechanical Engineering 11(9): 1–13. DOI: 10.1177/1687814019878041.

10.

Rastegari

Archenti

Mobin

(2017) Condition based maintenance of machine tools: vibration monitoring of spindle units. In: Proceedings - Annual Reliability and Maintainability Symposium, 29 March 2017. Institute of Electrical and Electronics Engineers Inc. DOI: 10.1109/RAM.2017.7889683.

11.

Rayjade

Bhandare

Bhagure

, et al. (2022) Fault detection and vibration analysis of flour mill system. In: International Conference on Science, Technology and Sustainability (ICSTS – 2022), 05-06 November 2022.

12.

Sharma

Amarnath

Kankar

(2016) Feature extraction and fault severity classification in ball bearings. JVC/Journal of Vibration and Control 22(1): 176–192. DOI: 10.1177/1077546314528021.

13.

Surucu

Gadsden

Yawney

(2023) Condition monitoring using machine learning: a review of theory, applications, and recent advances. In: Expert Systems with Applications. Elsevier Ltd. DOI: 10.1016/j.eswa.2023.119738.

14.

Upadhye

Borole

Sharma

(2015) Real-time wireless vibration monitoring system using LabVIEW. In: 2015 International Conference on Industrial Instrumentation and Control, ICIC 2015, pp. 925–928. DOI: 10.1109/IIC.2015.7150876.

15.

Vakharia

Gupta

Kankar

(2015) Ball Bearing Fault Diagnosis Using Supervised and Unsupervised Machine Learning Methods. International Journal of Acoustics and Vibration 204: 244–250.

16.

Wang

Huang

Gong

, et al. (2016) Improved feature extraction using structured Fisher discrimination sparse coding scheme for machinery fault diagnosis. Advances in Mechanical Engineering 8(12): 1–16. DOI: 10.1177/1687814016683085.