ASG-HOMGAT: A high-order multi-head graph attention network with adaptive small graph structure for rolling bearing fault diagnosis

ding shanshan,Renwen Chen,Hao Liu,Fei Liu,Zhang junyi
DOI: https://doi.org/10.1088/1361-6501/ad3293
IF: 2.398
2024-03-12
Measurement Science and Technology
Abstract:Traditional Euclidean spatial data processing is difficult to capture the inherent relationships of unstructured data such as bearing vibration signals. Representing vibration signals in graphical form helps to preserve their topologi-cal structure and temporal information. Secondly, most existing graph convolutional network methods are based on large graph structured data, which incurs certain memory overhead when aggregating high-order neighborhood node information and ignores important information between samples in the global graph structure. To address these issues, this paper proposes a high-order multi-head graph attention network based on an adaptive small graph struc-ture (ASG-HOMGAT) for fault diagnosis of rolling bearings. Firstly, the adaptive preprocessing layer is used to adaptively denoise and compress the one-dimensional time-domain vibration signal, generating small rule graph data with topological structure. Then, these small graph structured data samples are input into a higher-order graph neural network, which aggregates features from multiple higher-order neighborhoods to achieve richer feature repre-sentations and fully explore the intrinsic correlation between samples. Finally, these features are aggregated into a reinforced representation of graph nodes through a multi head attention mechanism, and a SoftMax classifier is used for fault classification. The experimental results show that the ASG-HOMGAT method has better performance com-pared to mainstream graph neural network diagnostic models. The code and model will be released at: https://github.com/ding-ss/ASG-HOMGAT.
engineering, multidisciplinary,instruments & instrumentation
What problem does this paper attempt to address?