A novel model with the ability of few-shot learning and quick updating for intelligent fault diagnosis

Abstract

Both of traditional intelligent fault diagnosis (TIFD) based on artificial features and modern intelligent fault diagnosis (MIFD) based on deep learning have made healthy progress in recent times. But, the bulk of methods neglects the actual application environments of intelligent diagnosis: (1) There are only few samples of faults, which greatly limit the popularity of TIFD and MIFD. Therefore, it is urgent to develop intelligent models with the ability of few-shot learning. (2) The performance degradation will occur when the equipment runs for a long time, which requires intelligent diagnosis models to possess the ability of quick updating. In order to remedy these shortcomings, a capsule auto-encoder model based on auto-encoder and capsule network, namely CaAE, is proposed for intelligent fault diagnosis. By constructing a capsule auto-encoder, various meaningful feature capsules are extracted from the input data, and then these capsules are fused adaptively into status capsules by the dynamic routing algorithm to represent health statuses. After that, status capsules are fed into the classifier to distinguish health statuses. The extraction of feature capsules enhances the model’s ability of mining information, which reduces the dependence of CaAE on the number of samples and compresses training time by reducing layers of the network. The loss function in the model introduces the penalty for the samples classified correctly and the constraints on the capsule auto-encoder, which make the model fast in training and excellent in feature extracting. A bearing dataset is utilized to validate the performance of the proposed CaAE. The results indicate that CaAE is suitable for few-shot learning and quick updating of intelligent models. In addition, the model can also achieve satisfactory results under noisy environment.