A Gearbox Fault Diagnosis Method Based on Graph Neural Networks and Markov Transform Fields

Many current fault diagnosis methods tend to ignore the temporal correlation in signals, leading to a loss of critical fault information. Additionally, traditional diagnostic models often face challenges in terms of noise immunity, generalization, and handling non-Euclidean structured data. To address these issues, we propose a novel fault diagnosis approach that combines Graph Neural Networks (GNN) with the Markov Transform Field (MTF). We first use the Markov Transform Field to convert vibration signals into two-dimensional images, preserving temporal correlation and preventing the loss of crucial fault information. Next, we use a Graph Convolutional Neural Network (GCN) to process graph-structured data, capturing global structural information. Finally, we introduce the Graph Attention Network (GAT) to dynamically adjust node weights based on their relative importance, enhancing the overall model performance. In this paper, we introduce a new fault diagnosis model, GCN-GAT, and evaluate it using the CWRU bearing dataset and a custom-built planetary gearbox dataset. The results show that our model maintains high fault detection accuracy even in the presence of significant noise and variable load conditions. This indicates that our approach demonstrates strong robustness and generalization, providing an effective solution for complex fault diagnosis tasks.