A Generative Transfer Learning Method for Extreme Class Imbalance Problem and Applied to Piston Aero-Engine Fault Cross-Domain Diagnosis
Pengfei Shen, Fengrong Bi, Xiaoyang Bi, Xiao Yang, Daijie Tang, Mengshuai Guo
Abstract
Transfer learning (TL) is a powerful approach that enhances the generalizability of cross-domain fault diagnosis. However, the challenge of acquiring high-quality mechanical fault signals limits its application. This article introduces the extreme class imbalance problem in the cross-domain diagnosis, restricting the label space of the target domain while relaxing the restrictions of unsupervised learning. The study proposes a novel generative TL method called fast sparse neural style, which employs sparse representation to capture the domain-invariant fault features as well as the Gram matrix to measure the domain-specific features. Fault features and domain features are proven to be separable in mechanical signals and are fused in the data generation process. Compared to other methods through various cross-domain diagnostic tasks on a piston aero-engine, the proposed method has obvious advantages in tasks with substantial inter-domain differences, demonstrating the potential and research value of generative TL.