Deep transfer learning rolling bearing fault diagnosis method based on convolutional neural network feature fusion
Di Yu, H.C. Fu, Yanchen Song, Wenjian Xie, Zhijie Xie
Abstract
Abstract Current deep-learning methods are often based on significantly large quantities of labeled fault data for supervised training. In practice, it is difficult to obtain samples of rolling bearing failures. In this paper, a transfer learning-based feature fusion convolutional neural network approach for bearing fault diagnosis is proposed. Specifically, the raw vibration signal features and the corresponding time-frequency image features of the input data are extracted by a one-dimensional convolutional neural network and a pre-trained ConvNeXt, respectively, and connected by a feature fusion strategy. Then, the fine-tuning method based on transfer learning can effectively reduce the reliance on labeled samples in the target domain. A wide convolution kernel is introduced in the time-domain signal feature extraction to increase the receptive field, which is combined with the channel attention mechanism to further optimize the feature quality. Finally, two common bearing datasets are utilized for fault diagnosis experiments. The experimental results show that the proposed model achieves an average accuracy of more than 98.63% in both cross-working conditions and cross-device diagnosis tasks. Meanwhile, anti-noise experiments and ablation experiments further validate the accuracy and robustness of the proposed method.