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

• Published in: IEEE transactions on reliability Vol. 72; no. 3; pp. 975 - 988
• Main Authors: Wang, Huan, Li, Yan-Fu
• Format: Journal Article
• Language: English
• Published: New York IEEE 01.09.2023; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Algorithms; Datasets; Deep learning; Fault diagnosis; Feature extraction; high-speed aeronautical (HSA) bearing; Industrial applications; Labels; Machine learning; Mechanical systems; Noise measurement; Noise robustness; noisy label; Support vector machines; Training
• DOI: 10.1109/TR.2022.3190942
• ISSN: 0018-9529

Fault diagnosis is an essential means to ensure the regular operation of mechanical systems. The existing data-driven algorithms are developed based on the assumption that the given label is entirely correct. However, mislabeling is common, which often occurs in industrial applications. These methods will overfit these mislabeled samples, resulting in inferior generalization. To this end, this article proposes a novel multistage true label distribution learning algorithm. Specifically, based on the training characteristics of data-driven algorithms on noisy datasets, a novel multistage adversarial loss function (MSA-Loss) is proposed. MSA-Loss can make the model construct the true label distribution from noisy datasets, prevent the model from overfitting the noisy samples, and finally keep the model with good generalization. The proposed method can be easily applied to any existing data-driven algorithm to improve its performance on noisy datasets. Our method is verified on high-speed aeronautical bearing and motor datasets, which prove that MSA-Loss has an excellent performance in noisy label scenarios. It can significantly improve the potential of existing diagnostic models in practical industrial applications.