Reliable and Intelligent Fault Diagnosis with Evidential VGG Neural Networks

Zhou, H. and Chen, W. and Cheng, L. and Williams, D. and De Silva, C.W. and Xia, M. (2023) Reliable and Intelligent Fault Diagnosis with Evidential VGG Neural Networks. IEEE Transactions on Instrumentation and Measurement, 72: 3508612. pp. 1-12. ISSN 0018-9456

[thumbnail of Final-TIM-22-05986]
Text (Final-TIM-22-05986)
Final_TIM_22_05986.pdf - Accepted Version
Available under License Creative Commons Attribution.

Download (1MB)

Abstract

With the emergence of Internet-of-Things (IoT) and big data technologies, data-driven fault diagnosis approaches, notably deep learning (DL)-based methods, have shown promising capabilities in achieving high accuracy through end-to-end learning. However, these deterministic neural networks cannot incorporate the prediction uncertainty, which is critical in practical applications with possible out-of-distribution (OOD) data. This present article develops a reliable and intelligent fault diagnosis (IFD) framework based on evidence theory and improved visual geometry group (VGG) neural networks, which can achieve accurate and reliable diagnosis results by incorporating additional estimation of the prediction uncertainty. Specifically, this article treats the predictions of the VGG as subjective opinions by placing a Dirichlet distribution on the category probabilities and collecting the evidence from data during the training process. A specific loss function assisted by evidence theory is adopted for the VGG to obtain improved uncertainty estimations. The proposed method, which incorporates evidential VGG (EVGG) neural networks, as termed here, is verified by a case study of the fault diagnosis of rolling bearings, in the presence of sensing noise and sensor failure. The experimental results illustrate that the proposed method can estimate the prediction uncertainty and avoid overconfidence in fault diagnosis with OOD data. Also, the developed approach is shown to perform robustly under various levels of noise, which indicates a high potential for use in practical applications.

Item Type:
Journal Article
Journal or Publication Title:
IEEE Transactions on Instrumentation and Measurement
Uncontrolled Keywords:
/dk/atira/pure/subjectarea/asjc/3100/3105
Subjects:
?? data modelsestimationevidence theoryevidence theoryfault diagnosisfault diagnosisneural networksreliabilitytrustworthy aiuncertaintyuncertainty estimationvgg neural networksinstrumentationelectrical and electronic engineering ??
ID Code:
190348
Deposited By:
Deposited On:
12 Apr 2023 15:40
Refereed?:
Yes
Published?:
Published
Last Modified:
15 Jul 2024 23:39