Does Your Model Think Like an Engineer? Explainable AI for Bearing Fault Detection with Deep Learning
arXiv:2310.12967 · doi:10.1109/ICASSP49357.2023.10096396
Abstract
Deep Learning has already been successfully applied to analyze industrial sensor data in a variety of relevant use cases. However, the opaque nature of many well-performing methods poses a major obstacle for real-world deployment. Explainable AI (XAI) and especially feature attribution techniques promise to enable insights about how such models form their decision. But the plain application of such methods often fails to provide truly informative and problem-specific insights to domain experts. In this work, we focus on the specific task of detecting faults in rolling element bearings from vibration signals. We propose a novel and domain-specific feature attribution framework that allows us to evaluate how well the underlying logic of a model corresponds with expert reasoning. Utilizing the framework we are able to validate the trustworthiness and to successfully anticipate the generalization ability of different well-performing deep learning models. Our methodology demonstrates how signal processing tools can effectively be used to enhance Explainable AI techniques and acts as a template for similar problems.
arXiv comment2023 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)
References in corpus (8)
- Towards A Rigorous Science of Interpretable Machine Learning
- Axiomatic Attribution for Deep Networks
- Machine Learning and Deep Learning Algorithms for Bearing Fault Diagnostics -- A Comprehensive Review
- Captum: A unified and generic model interpretability library for PyTorch
- Explaining by Removing: A Unified Framework for Model Explanation
- An Evaluation of the Human-Interpretability of Explanation
- Transformation Importance with Applications to Cosmology
- Human-interpretable model explainability on high-dimensional data