At present, the fault diagnosis methods for rolling bearings are all based on research with fewer fault categories, without considering the problem of multiple faults. In practical applications, the coexistence of multiple operating conditions and faults can lead to an increase in classification difficulty and a decrease in diagnostic accuracy. To solve this problem, a fault diagnosis method based on an improved convolution neural network is proposed. The convolution neural network adopts a simple structure of three-layer convolution. The average pooling layer is used to replace the common maximum pooling layer, and the global average pooling layer is used to replace the full connection layer. The BN layer is used to optimize the model. The collected multi-class signals are used as the input of the model, and the improved convolution neural network is used for fault identification and classification of the input signals. The experimental data of XJTU-SY and Paderborn University show that the method proposed in this paper has a good effect on the multi-classification of bearing faults.
Keywords: convolution neural network; multi-classification problem; rolling bearing.