Abstract:Considering the changing working conditions of rotating machinery in operation, it is often difficult to collect data accurately in some severe fault states, and the lack of data can lead to poor performance of deep learning-based fault diagnosis models. In the few-shot scenario, traditional meta-learning methods yield superior predictions by identifying ideal initialization parameters. When the update gradients of tasks are in different directions, meta-learning modifies the weights of samples in the current tasks to average them out, which may lead to negative transfer between tasks and poor generalization of biases. Based on the above problems, we propose a meta-learning network with anti-interference (AIML) for few-shot fault diagnosis, which is obtained by combining a dynamic fine-tuning technique to increase the gradient agreement of the tasks. AIML consists of two functions: the feature encoding network (FEN) and the base network. AIML uses the property of shared parameters about transfer learning to learn shared feature representations of different tasks, while FEN dynamically adjusts the loss weights of conflicting tasks due to meta-learning with two-level optimization techniques. First, in the internal loop, AIML updates the gradient of the base network while the network parameters of FEN are fixed. Instead of immediately computing the gradient of the optimal parameters for the new task, the FEN is then integrated in the outer loop to learn the meta-representation, and the weights obtained for the various tasks relate to the gradient for meta-optimization. Only during the outer loop stage of the meta-training is the FEN updated. Three publicly available datasets are used to assess the performance of AIML, and the results show that it is more effective at resolving problems involving few-shot fault diagnosis.

Few-shot Transfer Learning Method Based on Meta-Learning and Graph Convolution Network for Machinery Fault Diagnosis

Few-shot transfer learning for intelligent fault diagnosis of machine

Augmentation-based discriminative meta-learning for cross-machine few-shot fault diagnosis

Cross-domain Few-Shot Fault Diagnosis Based on Meta-Learning and Domain Adversarial Graph Convolutional Network

Generic Meta-Transfer Learning Model with Special Neuronal Processing Parameters for Few-Shot Fault Bearing Diagnosis

Transfer Learning with Convolutional Neural Networks for Small Sample Size Problem in Machinery Fault Diagnosis

Industrial fault diagnosis based on few shot learning

Few-Shot Learning for Fault Diagnosis With a Dual Graph Neural Network

Few-shot intelligent fault diagnosis based on an improved meta-relation network

A Meta-learning Method for Few-shot Bearing Fault Diagnosis under Variable Working Conditions

Meta-learning as a promising approach for few-shot cross-domain fault diagnosis: Algorithms, applications, and prospects

An Information Fusion-Based Meta Transfer Learning Method for Few-Shot Fault Diagnosis under Varying Operating Conditions

A Meta-Learning Network with Anti-Interference for Few-Shot Fault Diagnosis

A New Meta-transfer Learning Method with Freezing Operation for Few-shot Bearing Fault Diagnosis

Deep Convolutional Transfer Learning Network: A New Method for Intelligent Fault Diagnosis of Machines with Unlabeled Data

A Generalized Graph Contrastive Learning Framework for Few-Shot Machine Fault Diagnosis

An intelligent fault diagnosis model based on deep neural network for few-shot fault diagnosis

Few shot cross equipment fault diagnosis method based on parameter optimization and feature mertic

Self-Adaptation Graph Attention Network via Meta-Learning for Machinery Fault Diagnosis With Few Labeled Data

An Intelligent Machinery Fault Diagnosis Method Based on GAN and Transfer Learning under Variable Working Conditions

A Few-Shot Machinery Fault Diagnosis Framework Based on Self-Supervised Signal Representation Learning