An Uncertainty-Aware Deep Learning Model for Reliable Detection of Steel Wire Rope Defects

Wenting Yi, Wai Kit Chan, Hiu Hung Lee, Steven T. Boles, Xiaoge Zhang

Research output: Journal article publicationJournal articleAcademic researchpeer-review

Abstract

As safety is a top priority in mission-critical engineering applications, uncertainty quantification emerges as a linchpin to the successful deployment of AI models in these high-stakes domains. In this article, we seamlessly encode a simple and principled uncertainty quantification module spectral-normalized neural Gaussian process (SNGP) into GoogLeNet to detect various defects in steel wire ropes (SWRs) accurately and reliably. To this end, the developed methodology consists of three coherent steps. In the first step, raw magnetic flux leakage (MFL) signals in waveform associated with normal and defective SWRs that are manifested in the number of broken wires are collected via a dedicated experimental setup. Next, the proposed approach utilizes Gramian angular field to transform the MFL signal in 1-D time series into 2-D images while preserving key spatial and temporal structures in the data. Third, built atop the backbone of GoogLeNet, we systematically integrate SNGP by adding the spectral normalization (SN) layer to normalize the weights and replacing the output layers with a Gaussian process (GP) in the main network and auxiliary classifiers of GoogLeNet accordingly, where SN enables to preserve the distance in data transformation and GP makes the output layer of neural network distance aware when assigning uncertainty. Comprehensive comparisons with the state-of-the-art models highlight the advantages of the developed methodology in classifying SWR defects and identifying out-of-distribution (OOD) SWR instances. In addition, a thorough ablation study is performed to quantitatively illustrate the significant role played by SN and GP in the principledness of the estimated uncertainty toward detecting SWR instances with varying OODness.

Original languageEnglish
Pages (from-to)1187-1201
Number of pages15
JournalIEEE Transactions on Reliability
Volume73
Issue number2
DOIs
Publication statusPublished - 1 Jun 2024

Keywords

  • Deep learning
  • defect detection
  • Feature extraction
  • Gaussian process (GP)
  • Reliability
  • Safety
  • steel wire rope (SWR)
  • Task analysis
  • Uncertainty
  • uncertainty quantification
  • Wires

ASJC Scopus subject areas

  • Safety, Risk, Reliability and Quality
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'An Uncertainty-Aware Deep Learning Model for Reliable Detection of Steel Wire Rope Defects'. Together they form a unique fingerprint.

Cite this