Title
An Explainable Ai-Based Fault Diagnosis Model For Bearings
Abstract
In this paper, an explainable AI-based fault diagnosis model for bearings is proposed with five stages, i.e., (1) a data preprocessing method based on the Stockwell Transformation Coefficient (STC) is proposed to analyze the vibration signals for variable speed and load conditions, (2) a statistical feature extraction method is introduced to capture the significance from the invariant pattern of the analyzed data by STC, (3) an explainable feature selection process is proposed by introducing a wrapper-based feature selector-Boruta, (4) a feature filtration method is considered on the top of the feature selector to avoid the multicollinearity problem, and finally, (5) an additive Shapley explanation followed by k-NN is proposed to diagnose and to explain the individual decision of the k-NN classifier for debugging the performance of the diagnosis model. Thus, the idea of explainability is introduced for the first time in the field of bearing fault diagnosis in two steps: (a) incorporating explainability to the feature selection process, and (b) interpretation of the classifier performance with respect to the selected features. The effectiveness of the proposed model is demonstrated on two different datasets obtained from separate bearing testbeds. Lastly, an assessment of several state-of-the-art fault diagnosis algorithms in rotating machinery is included.
Year
DOI
Venue
2021
10.3390/s21124070
SENSORS
Keywords
DocType
Volume
bearing, Boruta, condition-based monitoring, explainable AI, fault diagnosis, model interpretability, SHAP, Stockwell transform
Journal
21
Issue
ISSN
Citations 
12
1424-8220
0
PageRank 
References 
Authors
0.34
0
3
Name
Order
Citations
PageRank
Md Junayed Hasan101.69
Muhammad Sohaib200.68
Jong Myon Kim314432.36