Partial Transfer Learning Method Based on Inter-Class Feature Transfer for Rolling Bearing Fault Diagnosis

Abstract

Rolling bearing fault diagnosis methods based on transfer learning always assume that the sample classes in the target domain are consistent with those in the source domain during the training phase. However, it is difficult to collect all fault classes in the early stage of mechanical application. The more likely situation is that the training data in the target domain only contain a subset of the entire health state, which will lead to the problem of label imbalance compared with the source domain. The outlier classes in the source domain that do not have corresponding target domain samples for feature alignment will interfere with the feature transfer of other classes. To address this specific challenge, this study introduces an innovative inter-class feature transfer fault diagnosis approach. By leveraging label information, the method distinctively computes the distribution discrepancies among shared classes, thereby circumventing the deleterious influence of outlier classes on the transfer procedure. Empirical evaluations on two rolling bearing datasets, encompassing multiple partial transfer tasks, substantiate that the proposed method surpasses other approaches, offering a novel and efficacious solution for the realm of intelligent bearing fault diagnosis.