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

Abstract

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.