Adversarial Training-Based Convolutional Neural Network Fault Diagnosis for Bearings with Random Perturbation
摘要
In response to the challenges of feature extraction and model security faced by convolutional neural networks in bearing fault diagnosis, this study proposes a sample segmentation method based on the characteristics of the fault signal, combined with an analysis of the envelope spectrum to generate training samples. Adversarial training is employed to enhance both diagnostic accuracy and model robustness. The proposed method adaptively determines the sample length based on fault feature frequencies, replacing traditional experience-based segmentation strategies, and utilizes color-mapping techniques to visually present the feature information from the envelope spectrum. To further improve the model’s stability under small perturbations, an adversarial layer based on the FGSM-RP algorithm is introduced by applying random perturbations during training. This significantly enhances the model’s ability to defend against such attacks. The experimental results show that the proposed method not only achieves high-precision fault classification but also effectively defends against adversarial attacks from various algorithms. These findings validate the effectiveness of the proposed method and provide new insights for improving roll bearing fault diagnosis.