A weak prior embedding-based method for transfer fault diagnosis of rolling bearing

摘要

Transfer learning extends the application scope of deep intelligent fault diagnosis models. However, when operation conditions are significantly varied, existing transfer learning methods may encounter a bottleneck. The unsupervised methods will mismatch features, while supervised methods are constrained by insufficient target labeled instances. A weak prior embedding-based domain adaption network (WPEDAN) is proposed to resolve this problem. Specifically, the cluster ability of source data pre-trained network to target data is emphasized and enhanced directly and indirectly so as to obtain discriminative feature clusters. The target prior template samples are embedded and automatically delivered to the most relevant feature clusters, so the class information of target instances can be estimated based on similarity and supervised training on target is following. Moreover, an improved conditional maximum mean discrepancy (ICMMD) is developed to further align conditional distri-bution. Experiments on two challenging bearing datasets show that proposed method achieves more than 99% diagnosis accuracy in all tasks, and is significantly ahead of the comparison methods in complex transfer tasks.

关键词