Coupling Fault Diagnosis of Bearings Based on Hypergraph Neural Network

Abstract

Coupling faults that simultaneously occur during the operation of mechanical equipment are widespread. These faults encompass a diverse range of high-order coupling relationships, involving multiple base fault types. Based on the advantages of hypergraphs for higher-order relationship descriptions, two coupling fault diagnosis architectures based on the hypergraph neural network are proposed in this paper: 1. In the coupling fault diagnosis framework based on feature generation, the base faults serve as the hypergraph nodes, and each hyperedge connects the base faults. The generator, which consists of the hypergraph neural network, generates coupling faults as negative samples to enforce regularization constraints for the discriminator training. 2. In the coupling fault diagnosis framework based on feature extraction, each node represents a fault mode, and each hyperedge connects nodes with common failure modes. The multi-head attention mechanism extracts the features of base faults, and the common fault features in a hyperedge are aggregated via the hypergraph neural network. The inner product correlation is used to diagnose the fault modes. The results show that the diagnostic accuracy for coupling faults with the two frameworks reaches 88.6% and 86.76%, respectively. Both frameworks can be used for the diagnosis and analysis of high-order coupling faults.