MTC-GAN Bearing Fault Diagnosis for Small Samples and Variable Operating Conditions

Abstract

In response to the challenges of bearing fault diagnosis under small sample sizes and variable operating conditions, this paper proposes a novel method based on the two-dimensional analysis of vibration acceleration signals and a Multi-Task Conditional Generative Adversarial Network (MTC-GAN). This method first constructs two-dimensional images of vibration signals by leveraging the physical properties of the bearing acceleration signals and employs Local Binary Patterns (LBP) to extract subtle texture features from these images, thereby generating fault feature signatures with high discriminative power across different operating conditions. Subsequently, MTC-GAN is utilized for data augmentation, and the trained discriminator is used to perform fault classification tasks, improving classification accuracy under conditions with small sample sizes. Experimental results demonstrate that the proposed method achieves excellent fault diagnosis accuracy and robustness under both small sample sizes and varying operating conditions. Compared to traditional methods, this approach exhibits higher efficiency and reliability in handling complex operating conditions and data scarcity.