Research on Fault Diagnosis Method Based on Improved CNN

Traditional fault diagnosis methods require complex signal processing and expert experience, and the accuracy of fault identification is low. To solve these problems, a fault diagnosis method based on an improved convolutional neural network (CNN) is proposed. Based on the traditional CNN model, a variety of convolution stride modes were added to extract features of different scales of signals and expand the feature dimension. Firstly, the vibration signals were collected and grouped. Then, the data were divided into a training set and a test set and input into improved CNN for feature extraction and model training to realize fault identification. The proposed model achieved a classification accuracy of 99.3% when testing the vibration data of the armored vehicle. Finally, the proposed model was used to classify different fault types of planetary gearboxes. The gradient-weighted class activation mapping (Grad-CAM) method was used to visualize the classification weight of samples. The results showed that the classification accuracy reaches 98% under various working conditions of the planetary gearbox.