Cross-domain correlation representation for new fault categories discovery in rolling bearings

Chenglong Wang, Jie Nie, Weizhi Nie, Peizhe Yin, Di Niu, Xinyue Liang, Shusong Yu

Abstract: Fault diagnosis technology plays a crucial role in preventing faults and mitigating safety hazards, particularly in domains such as nuclear power, aerospace, and manufacturing. However, obtaining an ample number of fault samples is challenging due to the stringent requirements for safe and reliable equipment operation in real production environments. Currently, various deep learning-based techniques have been studied to mitigate this issue. However, these methods often overlook the identification of novel faults that may emerge in the future. In this paper, we present a novel cross-domain correlation representation approach for discovering new fault categories in rolling bearings. Our approach leverages the knowledge acquired through supervised feature extraction to facilitate the discovery process of new fault categories. In particular, we propose a causal inference module to identify fault-related features by eliminating irrelevant information and retaining the primary influential features that contribute to the occurrence of faults. Additionally, we introduce a pseudo-label generation module for the discovery process of unsupervised samples in new fault categories. This module establishes a learning relationship between labeled and unlabeled samples, enabling the utilization of known category knowledge during the discovery of unknown categories. We verify the effectiveness of our model using the CWRU, PU, and MFPT datasets, showing that it outperforms state-of-the-art techniques in terms of clustering accuracy and normalized mutual information.