Intelligent Fault Diagnosis Method Based on Cross-Device Secondary Transfer Learning of Efficient Gated Recurrent Unit Network

Abstract

In response to the issues of low model recognition accuracy and weak generalization in mechanical equipment fault diagnosis due to scarce data, this paper proposes an innovative solution, a cross-device secondary transfer-learning method based on EGRUN (efficient gated recurrent unit network). This method utilizes continuous wavelet transform (CWT) to transform source domain data into images. The EGRUN model is initially trained, and shallow layer weights are frozen. Subsequently, random overlapping sampling is applied to the target domain data to enhance data and perform secondary transfer learning. The experimental results demonstrate that this method not only significantly improves the model’s ability to learn fault features but also enhances its classification accuracy and generalization performance. Compared to current state-of-the-art algorithms, the model proposed in this study shows faster convergence speed, higher diagnostic accuracy, and superior robustness and generalization, providing an effective approach to address the challenges arising from scarce data and varying operating conditions in practical engineering scenarios.