A multi-scale graph convolutional network with contrastive-learning enhanced self-attention pooling for intelligent fault diagnosis of gearbox.

Recently, the emerging graph convolutional network (GCN) has been applied into fault diagnosis with the aim of providing additional fault features through topological information. However, there are some limitations with these methods. First, the interactions between multi-frequency scales are ignored in existing studies, while they mainly focus on constructing graphs through the relationship between channels/instances. Second, the constructed graph cannot well reflect the topology of noisy samples and lacks robust hierarchical representation learning capability, and the learned graphs have limited interpretability. Hence, a Multi-Scale GCN with Contrastive-learning enhanced Self-attention Pooling (MSGCN-CSP) method is proposed for intelligent fault diagnosis of gearbox. Time–frequency distributions are converted into multi-scale graphs to extract fault features through topological relationships between multi-frequencies. Contrastive-learning is used to implement graph pooling, which enables hierarchical representation learning. Experimental results on two gearbox datasets illustrate that the proposed method offers competitive diagnostic performance and provides good interpretability in establishing GCN. [Display omitted] • A graph modeling strategy is proposed for perceiving multi-scale frequency information. • The contrastive-learning enhanced pooling is proposed for robust hierarchical representation learning. • Physical mechanism of the method is analyzed to demonstrate its interpretability. • Comparative studies show the superiority of the method in terms of accuracy. [ABSTRACT FROM AUTHOR]