WPD-Enhanced Deep Graph Contrastive Learning Data Fusion for Fault Diagnosis of Rolling Bearing

Abstract

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.