Polarized Direct Cross-Attention Message Passing in GNNs for Machinery Fault Diagnosis

TL;DR

PolaDCA is introduced, a novel relational learning framework that enables adaptive message passing through data-driven graph construction that dynamically infers attention weights from three semantically distinct node features without requiring fixed adjacency matrices.

Abstract

The reliability of safety-critical industrial systems hinges on accurate and robust fault diagnosis in rotating machinery. Conventional graph neural networks (GNNs) for machinery fault diagnosis face limitations in modeling complex dynamic interactions due to their reliance on predefined static graph structures and homogeneous aggregation schemes. To overcome these challenges, this paper introduces polarized direct cross-attention (PolaDCA), a novel relational learning framework that enables adaptive message passing through data-driven graph construction. Our approach builds upon a direct cross-attention (DCA) mechanism that dynamically infers attention weights from three semantically distinct node features (such as individual characteristics, neighborhood consensus, and neighborhood diversity) without requiring fixed adjacency matrices. Theoretical analysis establishes PolaDCA's superior noise robustness over conventional GNNs. Extensive experiments on industrial datasets (i.e., XJTUSuprgear, CWRUBearing and Three-Phase Flow Facility datasets) demonstrate state-of-the-art diagnostic accuracy and enhanced generalization under varying noise conditions, outperforming seven competitive baseline methods. The proposed framework provides an effective solution for safety-critical industrial applications.

Figures (12)

• Figure 1: Framework of DCA-GNN
• Figure 2: Framework of DCA-based message passing
• Figure 3: Framework of PolaDCA-based message passing
• Figure 4: Accuracy comparison on CWRUBering data with gaussian noise
• Figure 5: Accuracy comparison on TFF data with gaussian noise