Few-shot transfer learning for intelligent fault diagnosis of machine

Abstract

Rotating machinery intelligent diagnosis with large data has been researched comprehensively, while there is still a gap between the existing diagnostic model and the practical application, due to the variability of working conditions and the scarcity of fault samples. To address this problem, few-shot transfer learning method is constructed utilizing meta-learning for few-shot samples diagnosis in variable conditions in this paper. We consider two transfer situations of rotating machinery intelligent diagnosis named conditions transfer and artificial-to-natural transfer, and construct seven few-shot transfer learning methods based on a unified 1D convolution network for few-shot diagnosis of three datasets. Baseline accuracy under different sample capacity and transfer situations are provided for comprehensive comparison and guidelines. What is more, data dependency, transferability, and task plasticity of various methods in the few-shot scenario are discussed in detail, the data analysis result shows meta-learning holds the advantage for machine fault diagnosis with extremely few-shot instances on the relatively simple transfer task. Our code is available at https://github.com/a1018680161/Few-shot-Transfer-Learning.