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