Abstract:The defect detection of industrial products is an essential task in industrial production. In the field of tire defect detection, X-ray imaging sensors are employed by tire factories to investigate the inner structure of the tires and identify defects. In recent years, deep learning has become a good alternative to manual inspection, but one major challenge of utilizing deep learning models is the requirement of large datasets with precisely annotated labels. In industrial applications, the process of annotating labels is often dependent on domain experts, which introduces inter- and intra-observer variability. Consequently, industrial datasets often contain noisy labels, which will deteriorate the performance of convolutional neural networks (CNNs) during the training process. In this study, we propose an approach called prototype-based supervised contrastive learning (PSCL) to address label noise problems. First, we utilize a supervised contrastive learning (SCL) framework to pull together features of samples belonging to the same class and produce class prototypes based on these features. Subsequently, label correction is performed using the updated class prototypes. In return, these corrected labels will contribute to the training of a reliable network, which ensures the provision of better prototypes. To assess the effectiveness of our method, we conduct extensive experiments on both synthetic noisy datasets and industrial tire datasets. The results demonstrate the robustness of our approach even in datasets with a large proportion of noisy labels. Therefore, our approach provides a promising solution for mitigating label noise problems in industrial datasets, offering great potential for practical industrial applications.

Contrastive Regularization Guided Label Refurbishment for Fault Diagnosis under Label Noise

Prototype-Based Supervised Contrastive Learning Method for Noisy Label Correction in Tire Defect Detection

LAFD-Net: Learning with Noisy Pseudo-Labels for Semisupervised Bearing Fault Diagnosis

Meta-Self-Training Based on Teacher–Student Network for Industrial Label-Noise Fault Diagnosis

Multi-Scale Adaptive-Routing Capsule Contrastive Network-Based Intelligent Fault Diagnosis Method for Rotating Machinery under Noisy Environment and Labels

Robust Mechanical Fault Diagnosis with Noisy Label Based on Multistage True Label Distribution Learning.

Curriculum-Based Federated Learning for Machine Fault Diagnosis with Noisy Labels

An Enhanced Intelligent Fault Diagnosis Method to Combat Label Noise

A Robust Intelligent Fault Diagnosis Method for Rotating Machinery under Noisy Labels

Contrastive Learning Joint Regularization for Pathological Image Classification with Noisy Labels

Iterative Error Self-Correction for Robust Fault Diagnosis of Mechanical Equipment with Noisy Label

Intelligent Fault Diagnosis With Noisy Labels via Semi-Supervised Learning on Industrial Time Series

Dual Contrastive Learning for Semi-Supervised Fault Diagnosis Under Extremely Low Label Rate

An Adaptive Symmetric Loss in Dynamic Wide-Kernel ResNet for Rotating Machinery Fault Diagnosis Under Noisy Labels

A Fault Diagnosis Framework Insensitive to Noisy Labels Based on Recurrent Neural Network

Intelligent Fault Diagnosis with Noisy Labels Via Semisupervised Learning on Industrial Time Series

A Two-Stage Semi-Supervised Learning Framework for Fault Diagnosis of Rotating Machinery

A Novel Normalized Recurrent Neural Network for Fault Diagnosis with Noisy Labels

Label Correction Using Contrastive Prototypical Classifier for Noisy Label Learning.

Reflective Learning with Label Noise

Two Wrongs Don't Make a Right: Combating Confirmation Bias in Learning with Label Noise.