Fault diagnosis model of convolutional neural network based on maximum mean difference
In view of the difficulty of bearing fault data collection in engineering scenarios and the low accuracy and stability of bearing fault diagnosis under small samples,a fault diagnosis method of rolling bearing under small samples,namely convolutional neural network(CNN)diagnosis model(method)based on maximum mean discrepancy(MMD)was proposed.Firstly,the fault simulation signal was obtained according to the bearing fault mechanism,and the confrontation training model between the simulation signal and a few real samples was constructed based on the generative adversarial network,the pseudo-domain samples were obtained,and the training data set was expanded.Secondly,the cross-entropy loss and MMD were used as the optimization criteria of the CNN,the scaling factor was introduced,the network was dynamically optimized,and the scaling factor of 0.05 was selected as the optimal network structure parameter according to the test results,and the training model of fault diagnosis was constructed.Finally,the training set of the model was composed of the pseudo-domain samples with 1 024 data points and the real samples,which was normalized and input into the constructed network model.With MMD as constraint,convolution and pooling operations were carried out to achieve feature extraction,and the model was optimized by back propagation to achieve iterative updating of the diagnostic model parameters.The experimental results show that the proposed method can significantly improve the accuracy of bearing fault diagnosis and recognition under small samples.When the number of samples is only 16,the recognition rate can reach more than 95%,which proves that the method can still obtain a high fault recognition rate in bearing fault diagnosis under small samples.
rolling bearingfault diagnosissmall sample sizegenerative adversarial networkconvolutional neural network(CNN)maximum mean discrepancy(MMD)cross-entropy loss
万方数据
维普
