Image-based Novel Fault Detection with Deep Learning Classifiers using Hierarchical Labels

Nurettin Sergin,Jiayu Huang,Tzyy-Shuh Chang,Hao Yan
2024-03-27
Abstract:One important characteristic of modern fault classification systems is the ability to flag the system when faced with previously unseen fault types. This work considers the unknown fault detection capabilities of deep neural network-based fault classifiers. Specifically, we propose a methodology on how, when available, labels regarding the fault taxonomy can be used to increase unknown fault detection performance without sacrificing model performance. To achieve this, we propose to utilize soft label techniques to improve the state-of-the-art deep novel fault detection techniques during the training process and novel hierarchically consistent detection statistics for online novel fault detection. Finally, we demonstrated increased detection performance on novel fault detection in inspection images from the hot steel rolling process, with results well replicated across multiple scenarios and baseline detection methods.
Machine Learning,Artificial Intelligence
What problem does this paper attempt to address?
This paper focuses on the problem of using deep learning for new fault detection in industrial manufacturing systems. Existing fault classification models often assume that the types of faults are a known fixed set, but in reality, new types of faults may be encountered. The paper proposes a method to enhance the deep learning model's capability for detecting unknown faults by utilizing hierarchical label information, without sacrificing the model's performance on known faults classification. Specifically, the paper proposes two key points: 1. **Soft labeling technique during the training process**: Improving the novel fault detection technique of deep learning by injecting hierarchical structure information into the loss function, allowing the model to consider both classification accuracy and hierarchical structure consistency during training. 2. **Hierarchical consistency score for online detection**: Proposing a new statistical method for online detection of new faults, by detecting samples that are inconsistent with the known fault hierarchy, in order to improve overall detection performance. The paper demonstrates the superiority of this method on a defect detection image dataset of hot steel rolling process, and proves that the proposed strategies can improve the performance of deep learning-based fault detection methods. In addition, the authors discuss the importance of incorporating hierarchical information into training and detection statistics for anomaly detection, pointing out that although many fault sets have hierarchical structures, these information are often overlooked in practice.