Fault Diagnosis Method of Rolling Bearings Under Different Working Conditions Based on Federated Feature Transfer Learning

Abstract

A rolling bearing fault diagnosis method based on the federated feature transfer learning is proposed for the low accuracy of the diagnosis model in the presence of large differences in data distribution under different working conditions, difficulty in obtaining labeled data and non-sharing of data among different users. This method performs wavelet transformation on the time domain vibration data of rolling bearings to obtain a time-frequency diagram. The priori labeled public data and the multi-user island private data are regarded as the source domain and the target domain. The multi-representation feature extraction structure is introduced to improve the original residual network. Based on an improved residual network and multi-representation features in the source domain and the target domain, every local model and a federated global model are constructed. Through verification of bearing data, the proposed method can establish an effective fault diagnosis model with high fault diagnosis accuracy. It can integrate the knowledge of isolated island data without sharing data among multiple users.