Adaptive Meta Transfer Learning with Efficient Self-Attention for Few-Shot Bearing Fault Diagnosis

Jun Zhao,Tang Tang,Ying Yu,Jingwei Wang,Tianyuan Yang,Ming Chen,Jie Wu
DOI: https://doi.org/10.1007/s11063-022-10918-2
IF: 2.565
2022-07-05
Neural Processing Letters
Abstract:The success of these meta-learning methods in few-shot bearing fault diagnosis depends strictly on the assumption that the meta-training set and the meta-testing set share the same distribution, which inevitably leads to undesirable performance degradation in varying working conditions. In this paper, we propose a novel Adaptive Siamese-based Meta Transfer Learning Networks (ASTN) for cross-domain fault diagnosis with limited data. Our algorithm learns meta-knowledge from episodic metric meta-learning and further adapts the prior experience to the target domain with a transfer learning strategy. The experiment results demonstrate that our proposed approach outperforms a series of existing meta-learning methods with fewer auxiliary samples. Meanwhile, with the increment of the complexity in the meta-training set, the proposed ASTN achieves good cross-domain generalization.
computer science, artificial intelligence
What problem does this paper attempt to address?