Transfer Learning for Bearing Fault Diagnosis basedon Graph Neural Network with Dilated KNN andAdversarial Discriminative Domain Adaptation

Tang Tang,Zeyuan Liu,Chuanhang Qiu,Ming Chen,Ying Yu
DOI: https://doi.org/10.1088/1361-6501/ad3016
IF: 2.398
2024-03-06
Measurement Science and Technology
Abstract:Graph neural networks (GNNs) have emerged as a forefront in deep learning, notably influencing research in mechanical fault diagnosis. Transfer learning (TL), particularly through domain adaptation (DA) techniques, has found application in machinery fault diagnosis by training models under one working condition and deploying them under another. While efforts have been made to integrate GNNs with DA techniques to alleviate data distribution discrepancies by investigating the inter-sample relationships, challenges persist: reliance on K-Nearest Neighbor (KNN) for graph generation emphasizes close relationships, neglecting distant ones; batch processing limits real-time fault diagnosis; and transfer between different-sized bearings is nearly unexplored. To address these limitations, a novel framework for GNN- based domain adaptation in machinery fault diagnosis is proposed. Initially, a convolutional neural network (CNN) extracts node embeddings from the Continuous Wavelet Transform (CWT) graph of raw vibration signals. Subsequently, a graph generation layer (GGL) based on dilated KNN captures both close and distant sample relationships, addressing the long-range dependency issue. Two GNN blocks are then applied for inter-sample relationships investigation and further feature extraction with the outputs directed to a linear classifier during source domain pretraining. Following pretraining, adversarial discriminative domain adaptation (ADDA) is leveraged to mitigate domain distribution discrepancies. Additionally, a novel graph construction method that combines existing training samples with a new single sample is proposed, enabling fault prediction with single instances for real-time online fault diagnosis. Evaluation on datasets with varying working conditions and bearings of different sizes demonstrates the superior performance of our method to other comparison methods.
engineering, multidisciplinary,instruments & instrumentation
What problem does this paper attempt to address?