Improved Detection and Diagnosis of Faults in Deep Neural Networks Using Hierarchical and Explainable Classification

TL;DR

This paper introduces DEFault, a hierarchical and explainable AI-based method for comprehensive fault detection and diagnosis in deep neural networks, improving reliability by capturing dynamic and static features.

Contribution

It presents a novel technique combining dynamic features, hierarchical classification, static features, and explainable AI to detect and diagnose a wide range of faults in DNNs.

Findings

Achieves ~94% recall in fault detection

Attains ~63% recall in fault diagnosis

Outperforms state-of-the-art techniques by 3.92%-11.54%

Abstract

Deep Neural Networks (DNN) have found numerous applications in various domains, including fraud detection, medical diagnosis, facial recognition, and autonomous driving. However, DNN-based systems often suffer from reliability issues due to their inherent complexity and the stochastic nature of their underlying models. Unfortunately, existing techniques to detect faults in DNN programs are either limited by the types of faults (e.g., hyperparameter or layer) they support or the kind of information (e.g., dynamic or static) they use. As a result, they might fall short of comprehensively detecting and diagnosing the faults. In this paper, we present DEFault (Detect and Explain Fault) -- a novel technique to detect and diagnose faults in DNN programs. It first captures dynamic (i.e., runtime) features during model training and leverages a hierarchical classification approach to detect all major fault categories from the literature. Then, it captures static features (e.g., layer types) from DNN programs and leverages explainable AI methods (e.g., SHAP) to narrow down the root cause of the fault. We train and evaluate DEFault on a large, diverse dataset of ~14.5K DNN programs and further validate our technique using a benchmark dataset of 52 real-life faulty DNN programs. Our approach achieves ~94% recall in detecting real-world faulty DNN programs and ~63% recall in diagnosing the root causes of the faults, demonstrating 3.92% - 11.54% higher performance than that of state-of-the-art techniques. Thus, DEFault has the potential to significantly improve the reliability of DNN programs by effectively detecting and diagnosing the faults.