A holistic semi-supervised method for imbalanced fault diagnosis of rotational machinery with out-of-distribution samples

Zhangjun Wu, Renli Xu, Yuansheng Luo and Haidong Shao

Reliability Engineering and System Safety, 2024, vol. 250, issue C

Abstract: Fault diagnosis plays a critical role in ensuring the reliability and safety of industrial systems. Despite the success of semi-supervised learning in fault diagnosis, challenges persist in dealing with imbalanced datasets and out-of-distribution (OOD) samples, particularly in industrial settings where fault incidents are rare. The rarity of fault incidents in industrial contexts leads to dataset imbalances, while changes in conditions may introduce OOD samples. To address these challenges, this study proposes a novel semi-supervised fault diagnosis method that simultaneously considers the imbalanced distribution of fault categories and OOD samples. Firstly, the semi-supervised framework has been presented, which combines pseudo-labeling and consistency regularization techniques. Then, the self-adaptive threshold has been devised to balance confidence levels among fault classes using dataset-specific and class-specific thresholds, thereby enhancing the model's robustness in real-world scenarios. Additionally, the OOD detection strategy based on contrastive learning has been proposed, wherein predictions from two different views of each sample estimate its “likelihood†of belonging to in-distribution (ID) or OOD categories. This strategy improves system safety by alerting operators to potential anomalies. Comprehensive experimental validations and comparisons were conducted on two rotational machinery fault datasets, demonstrating that the proposed method achieves superior performance compared to the alternative methods and is applicable for practical application.

Keywords: Fault diagnosis; Semi-supervised learning; Imbalanced dataset; Out-of-distribution samples (search for similar items in EconPapers)
Date: 2024
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (2)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0951832024003697
Full text for ScienceDirect subscribers only

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:reensy:v:250:y:2024:i:c:s0951832024003697

DOI: 10.1016/j.ress.2024.110297

Access Statistics for this article

Reliability Engineering and System Safety is currently edited by Carlos Guedes Soares

More articles in Reliability Engineering and System Safety from Elsevier
Bibliographic data for series maintained by Catherine Liu ().