TY - JOUR

T1 - Reliable and Intelligent Fault Diagnosis with Evidential VGG Neural Networks

AU - Zhou, H.

AU - Chen, W.

AU - Cheng, L.

AU - Williams, D.

AU - De Silva, C.W.

AU - Xia, M.

PY - 2023/3/7

Y1 - 2023/3/7

N2 - 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.

AB - 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.

KW - Data models

KW - Estimation

KW - Evidence theory

KW - evidence theory

KW - fault diagnosis

KW - Fault diagnosis

KW - Neural networks

KW - Reliability

KW - Trustworthy AI

KW - Uncertainty

KW - uncertainty estimation

KW - VGG neural networks

U2 - 10.1109/TIM.2023.3250308

DO - 10.1109/TIM.2023.3250308

M3 - Journal article

VL - 72

SP - 1

EP - 12

JO - IEEE Transactions on Instrumentation and Measurement

JF - IEEE Transactions on Instrumentation and Measurement

SN - 0018-9456

M1 - 3508612

ER -