Mechanical Fault Classification And Root Cause Analysis Based on Machine Learning and Explainable AI  
Author Ya-Ting Lee


Co-Author(s) Hung-Kai Wang


Abstract This study utilizes machine learning and explainable AI (XAI) to construct accurate prediction models and root cause analysis for Prognostic and Health Management (PHM). The study proposes a novel research framework consisting of two stages: the first stage involves fault type classification, while the second stage involves fault severity classification and root cause analysis. In both stages, multiple machine learning classifiers are employed to identify faults in rotating machinery, utilizing vibration features associated with the rotation frequency and its harmonics for classification. XAI techniques are incorporated for feature extraction of important features and root cause analysis. The proposed approach is evaluated using the Machinery Fault Database (MAFAULDA), and experimental results show that the first stage achieves an average accuracy of 98.7%, better than results reported in the literature. The second stage achieves an average accuracy of 86.7%. The results indicate that XAI feature importance improves classification accuracy and reduces computational costs and time. Additionally, XAI-based root cause analysis provides a faster way to identify the type and severity of faults through important features and setting thresholds determining the root cause of process variations.


Keywords Machine Learning, Explainable AI, PHM, Root cause analysis, Feature importance, fault diagnosis
    Article #:  RQD28-211

Proceedings of 28th ISSAT International Conference on Reliability & Quality in Design
August 3-5, 2023