Trustworthy Fault Diagnosis with Uncertainty Estimation through Evidential Convolutional Neural Networks

Zhou, Hanting and Chen, Wenhe and Cheng, Longsheng and Liu, Jing and Xia, Min (2023) Trustworthy Fault Diagnosis with Uncertainty Estimation through Evidential Convolutional Neural Networks. IEEE Transactions on Industrial Informatics, 19 (11): 11. pp. 10842-10852. ISSN 1551-3203

[thumbnail of rev-TII submit 1.1]
Text (rev-TII submit 1.1)
rev_TII_submit_1.1.pdf - Accepted Version
Available under License Creative Commons Attribution.

Download (1MB)

Abstract

Deep neural networks (DNNs) have been widely used for intelligent fault diagnosis under the closed world assumption that any testing data is within classes of the training data. However, in reality, out-of-distribution (OOD) cases such as new fault conditions can happen after the original trained model is deployed. Most of the current DNNs are deterministic which can misclassify with high confidence in the open-world scenario. This overconfident behavior would not guarantee the reliability and robustness of fault diagnosis results in practice. Therefore, trustworthy intelligent fault diagnosis with uncertainty estimation is crucial for real applications. In this paper, we develop a novel convolutional neural network integrating evidence theory to achieve fault classifications with prediction uncertainty estimation. The estimated prediction uncertainty can identify potential OOD samples. This approach allows a minimal modification of the state-of-the-art DNN model by using a risk-calibrated evidential loss function and Dirichlet distribution that replaces the classification probabilities. The experimental results show that the proposed approach can not only achieve accurate classification of known classes but also detect unknown classes effectively. The proposed method shows significant potential in detecting OOD patterns and provides trustworthy fault diagnosis in open and non-stationary environments.

Item Type:
Journal Article
Journal or Publication Title:
IEEE Transactions on Industrial Informatics
Additional Information:
©2023 IEEE. Personal use of this material is permitted. However, permission to reprint/republish this material for advertising or promotional purposes or for creating new collective works for resale or redistribution to servers or lists, or to reuse any copyrighted component of this work in other works must be obtained from the IEEE.
Uncontrolled Keywords:
/dk/atira/pure/subjectarea/asjc/1700/1710
Subjects:
?? estimationevidence theoryfault diagnosisneural networkstask analysistrainingtrustworthy aiuncertaintyevidential convolutional neural networksfault diagnosisopen-set recognition (osr)uncertainty estimationinformation systemselectrical and electronic engine ??
ID Code:
187603
Deposited By:
Deposited On:
28 Feb 2023 14:25
Refereed?:
Yes
Published?:
Published
Last Modified:
14 Jun 2024 01:31