Transfer Learning for Bearing Fault Diagnosis based on Graph Neural Network with Dilated KNN and Adversarial Discriminative Domain Adaptation

Abstract

Graph neural networks (GNNs) have emerged as a forefront in deep learning, notably influencing research in mechanical fault diagnosis. Transfer learning, 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 extracts node embeddings from the continuous wavelet transform graph of raw vibration signals. Subsequently, a graph generation layer 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 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.