Gearbox compound fault diagnosis method based on deep adversarial graph convolution transfer learning network under low label ratios

Abstract

Abstract Advancement in measurement techniques has dramatically contributed to the development of the modern manufacturing industry. As the primary fault causing unplanned downtime of mechanical equipment, gearbox compound faults are usually coupled by single faults with unequal severity and are difficult to obtain. In industrial scenarios, monitoring data for extreme operating conditions is not available in advance, and labeling samples is time-consuming and costly. However, current research on unseen compound fault diagnosis relies on numerous labeled samples to train the model, and few studies are conducted on transfer learning and unseen compound fault diagnosis under low label ratios. To address the issue, a gearbox compound fault diagnosis method based on deep adversarial graph convolution transfer learning network (DAGCTLN) under low label ratios is proposed. Specifically, a novel DAGCTLN model, including a feature extractor, two label classifiers, and a discriminator, is constructed to realize the diagnosis of faults in the transfer domain and unseen compound faults in the source and target domains. The feature extractor of a three-layer graph convolutional network is presented to achieve deep extraction of fault features under low label ratios. Then a domain space adversarial mechanism between the feature extractor and discriminator is used to achieve global alignment of the source and target domain features. Furthermore, two label classifiers are constructed, and the adversarial adaptation of the decision boundary is realized by maxi-min the classifier difference to achieve subdomain alignment of the same class features in all domains. Experimental results indicate that DAGCTLN can achieve the highest fault diagnosis accuracy in the transfer domain compared to state-of-the-art algorithms. The average diagnosis accuracy of compound faults in all domains can reach 98.41% even if the label ratio is only 0.1, which provides guiding significance for the safe operation and predictive maintenance of mechanical equipment.