Application of Fault Diagnosis Method Combining Finite Element Method and Transfer Learning for Insufficient Turbine Rotor Fault Samples

Abstract

Deep learning has led to significant progress in the fault diagnosis of mechanical systems. These intelligent models often require large amounts of training data to ensure their generalization capabilities. However, the difficulty of obtaining turbine rotor fault data poses a new challenge for intelligent fault diagnosis. In this study, a turbine rotor fault diagnosis method based on the finite element method and transfer learning (FEMATL) is proposed, ensuring that the intelligent model can maintain high diagnostic accuracy in the case of insufficient samples. This method fully exploits the finite element method (FEM) and transfer learning (TL) for small-sample problems. First, FEM is used to generate data samples with fault information, and then the one-dimensional vibration displacement signal is transformed into a two-dimensional time-frequency diagram (TFD) by taking advantage of the deep learning model to recognize the image. Finally, a pre-trained ResNet18 network was used as the input to carry out transfer learning. The feature extraction layer of the network was trained on the ImageNet dataset and a fully connected layer was used to match the specific classification problems. The experimental results show that the method requires only a small amount of training data to achieve high diagnostic accuracy and significantly reduces the training time.