Rolling bearings are crucial mechanical components in the mechanical industry. Timely intervention and diagnosis of system faults are essential for reducing economic losses and ensuring product productivity. To further enhance the exploration of unlabeled time-series data and conduct a more comprehensive analysis of rolling bearing fault information, this paper proposes a fault diagnosis technique for rolling bearings based on graph node-level fault information extracted from 1D vibration signals. In this technique, 10 categories of 1D vibration signals from rolling bearings are sampled using a sliding window approach. The sampled data is then subjected to wavelet packet decomposition (WPD), and the wavelet energy from the final layer of the four-level WPD decomposition in each frequency band is used as the node feature. The weights of edges between nodes are calculated using the Pearson correlation coefficient (PCC) to construct a node graph that describes the feature information of rolling bearings under different health conditions. Data augmentation of the node graph in the dataset is performed by randomly adding nodes and edges. The graph convolutional neural network (GCN) is employed to encode the augmented node graph representation, and deep graph contrastive learning (DGCL) is utilized for the pre-training and classification of the node graph. Experimental results demonstrate that this method outperforms contrastive learning-based fault diagnosis methods for rolling bearings and enables rapid fault diagnosis, thus ensuring the normal operation of mechanical systems. The proposed WPDPCC-DGCL method offers two advantages: (1) the flexibility of wavelet packet decomposition in handling non-smooth vibration signals and combining it with the powerful multi-scale feature encoding capability of GCN for richer characterization of fault information, and (2) the construction of graph node-level fault samples to effectively capture underlying fault information. The experimental results demonstrate the superiority of this method in rolling bearing fault diagnosis over contrastive learning-based approaches, enabling fast and accurate fault diagnoses for rolling bearings and ensuring the normal operation of mechanical systems.

Traditional fault diagnosis models often overlook the interconnections between segments of vibration data, resulting in the loss of critical feature information. Therefore, an efficient fault diagnosis model tailored for rolling bearings is proposed in this paper. The 1D vibration signals are first preprocessed using ensemble empirical mode decomposition (EEMD) to generate multiple intrinsic mode functions (IMF) as individual nodes. The percentage distance between each node is calculated using the Wasserstein distance (WD) to capture the relationships between nodes and use it as the edge weights to construct a node graph. An improved multi-head graph attention network (MGAT) model is established to extract features and perform classification on the node graph. This MGAT model effectively utilizes the relationships between nodes and enhances the accuracy of fault diagnosis. The experimental results demonstrate that the proposed method achieves higher accuracy compared to similar models while requiring less processing time.