Improving Bearing Fault Identification by Using Novel Hybrid Involution-Convolution Feature Extraction With Adversarial Noise Injection in Conditional GANs

Abstract

Bearing faults are critical in machinery; their early detection is vital to prevent costly breakdowns and ensure operational safety. This study presents a pioneering take on addressing the challenges of imbalanced datasets in bearing fault diagnosis. By mitigating issues such as mode collapse and vanishing gradients, our novel method employs Conditional Generative Adversarial Networks (CGANs) with spectral normalization and adaptive adversarial noise injection to generate high-quality bearing fault samples. This enhances generalization and robustness against noisy data, significantly improving the stability of CGAN training. To extract meaningful features from grayscale bearing fault images, we introduce a novel combination of involution and convolution feature extraction method named I-C FFN. This innovative feature extraction method captures both local and global information, making it capable of handling various types of features, including channel-agnostic, spatial-specific, spatial-agnostic, and channel-specific characteristics. Our proposed oversampling methodology helped enhance the performance of proposed classification scheme as well as of benchmark transfer learning models. Having accuracy values between 99.40% to 99.61% for 0 and 1 HP imbalanced dataset respectively, our models outperformed standard transfer learning methodologies. Furthermore, by the inclusion of our proposed Adaptive Adversarial Class- Conditional GAN (AAC-cGAN), the samples quality and the robustness to noise was significantly increased as demonstrated by the quantitative assessment through various Evaluation Metrics employed in this paper. Lastly, the performance of each combination of both the up-sampled and under-sampled methodologies were assessed through multiple metrics to determine the effectiveness of our proposed approach in addressing class imbalance in bearing fault diagnosis.