Rolling bearing fault diagnosis method based on generative adversarial enhanced multi-scale convolutional neural network model

Maoxuan Zhou,Wei Kang,Kun He
2024-03-21
Abstract:In order to solve the problem that current convolutional neural networks can not capture the correlation features between the time domain signals of rolling bearings effectively, and the model accuracy is limited by the number and quality of samples, a rolling bearing fault diagnosis method based on generative adversarial enhanced multi-scale convolutional neural network model is proposed. Firstly, Gram angular field coding technique is used to encode the time domain signal of the rolling bearing and generate the feature map to retain the complete information of the vibration signal. Then, the re-sulting data is divided into a training set, a validation set, and a test set. Among them, the training set is input into the gradient penalty Wasserstein distance generation adversarial network to complete the training, and a new sample with similar features to the training sample is obtained, and then the original training set is expanded. Next, multi-scale convolution is used to extract the fault features of the extended training set, and the feature graph is normalized by example to overcome the influence of the difference in feature distribution. Finally, the attention mechanism is applied to the adaptive weighting of normalized features and the extraction of deep features, and the fault diagnosis is completed by the softmax classifier. Compared with ResNet method, the experimental results show that the proposed method has better generalization performance and anti-noise performance.
Signal Processing,Machine Learning
What problem does this paper attempt to address?
This paper proposes a rolling bearing fault diagnosis method to solve the problems of existing convolutional neural networks failing to effectively capture the correlation features between time-domain signals and limited model accuracy due to the quantity and quality of samples. By using a multi-scale convolutional neural network model enhanced by generative adversarial networks and incorporating Gram angular difference field encoding technology, the time series signals are encoded and the training samples are expanded. The method utilizes multi-scale convolution and attention mechanisms for feature extraction and classification recognition, improving the model's generalization performance and noise resistance. Experimental results show that this method outperforms the ResNet method.