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.

A Generalized Graph Contrastive Learning Framework for Few-Shot Machine Fault Diagnosis

A Graph Representation Learning-Based Method for Fault Diagnosis of Rotating Machinery under Time-Varying Speed Conditions

Self-supervised Knowledge Mining from Unlabeled Data for Bearing Fault Diagnosis under Limited Annotations

Semi-supervised machine fault diagnosis fusing unsupervised graph contrastive learning

Few-shot fault diagnosis based on multi-scale graph convolution filtering for industry

Exploiting Graph Neural Network with One-Shot Learning for Fault Diagnosis of Rotating Machinery

Few-Shot Learning for Fault Diagnosis With a Dual Graph Neural Network

A Fault Diagnosis Method for Rotating Machinery with Semi-supervised Graph Convolutional Network and Images Converted from Vibration Signals

A Self-Attention Legendre Graph Convolution Network for Rotating Machinery Fault Diagnosis

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

A multi-scale graph convolutional network with contrastive-learning enhanced self-attention pooling for intelligent fault diagnosis of gearbox

Domain Discrepancy-guided Contrastive Feature Learning for Few-shot Industrial Fault Diagnosis under Variable Working Conditions

Rotating machinery fault diagnosis based on feature extraction via an unsupervised graph neural network

Adaptive Dynamic Threshold Graph Neural Network: A Novel Deep Learning Framework for Cross-Condition Bearing Fault Diagnosis

A contrastive learning-based fault diagnosis method for rotating machinery with limited and imbalanced labels

Dynamic graph-based feature learning with few edges considering noisy samples for rotating machinery fault diagnosis

CGASNet: A Generalized Zero-Shot Learning Compound Fault Diagnosis Approach for Bearings

LSTM-Based Node-Gated Graph Neural Network for Cross-Condition Few-Shot Bearing Fault Diagnosis

A graph-guided collaborative convolutional neural network for fault diagnosis of electromechanical systems

Dual Contrastive Learning for Semi-Supervised Fault Diagnosis Under Extremely Low Label Rate

Domain Knowledge-Guided Contrastive Learning Framework Based on Complementary Views for Fault Diagnosis With Limited Labeled Data