Research on Fault Diagnosis Algorithm Based on Structure Optimization for Convolutional Neural Network

Most of the traditional fault diagnosis methods rely on artificial extraction features and the expert knowledge of related fields, and these algorithms are not accurate, and the ability of robustness and generalization are poor. Convolutional neural network is one of the most widely used deep learning models. Based on its unique convolution-pooling network structure, convolutional neural network has powerful feature extraction and expression capabilities. In this paper, the structure of classical convolutional neural network is optimized, and a fault diagnosis algorithm based on structure optimization for convolutional neural network is proposed. For the characteristics of one-dimensional vibration signals, the convolution kernel of the first convolutional layer is set to a wide volume, and we use the batch normalization algorithm to further improve the convergence speed of the model. Through the experiment of the CWRU bearing fault public data set, the proposed algorithm has more than 99% fault recognition rate.