Piston aero-engine fault cross-domain diagnosis based on unpaired generative transfer learning

Abstract

As artificial intelligence (AI) and machine learning continue to advance, they have become key players in signal analysis and pattern recognition. However, challenges such as domain shift and a lack of labeled data hinder the application of machine learning to new, unfamiliar signals, particularly in the development of intelligent diagnostic methods for mechanical faults. In this study, we address these challenges by framing them as the extreme class imbalance problem. We introduce a novel generative transfer learning framework that leverages cycle-consistent adversarial networks with hard constraints (CycleGAN-HCs). This framework is designed to generate unpaired mechanical fault signals, which are then used to train classifiers that can operate across different domains. To enhance feature extraction capabilities for time series data, we developed a new generator model called U-Attention. Additionally, we refined the training loss function to better capture domain-specific and classification-specific features in mechanical vibration signals. The proposed method successfully facilitates feature transfer and data augmentation. Its effectiveness and reliability have been demonstrated through four cross-domain fault diagnosis tasks involving piston aero-engines. Comparative verification shows that the generative transfer learning diagnostic framework outperforms traditional methods, offering superior performance and better generalization in complex mechanical fault diagnosis scenarios.