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.

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

Semi-supervised meta-learning networks with squeeze-and-excitation attention for few-shot fault diagnosis

Attention-based Deep Meta-Transfer Learning for Few-Shot Fine-Grained Fault Diagnosis

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

Semi-supervised Multi-Scale Attention-Aware Graph Convolution Network for Intelligent Fault Diagnosis of Machine under Extremely-Limited Labeled Samples

Gradient Flow-Based Meta Generative Adversarial Network for Data Augmentation in Fault Diagnosis.

Adaptive model-agnostic meta-learning network for cross-machine fault diagnosis with limited samples

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

Smeta-LU: A Self-Supervised Meta-Learning Fault Diagnosis Method for Rotating Machinery Based on Label Updating

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

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

A novel meta-transfer learning approach via convolutional multi-head self-attention network for few-shot fault diagnosis

Metric-based Meta-Learning Model for Few-Shot Fault Diagnosis under Multiple Limited Data Conditions

A Novel Cross-Domain Fault Diagnosis Method Based on Model Agnostic Meta-Learning

Novel Cross-Domain Fault Diagnosis Method Based on Model-Agnostic Meta-Learning Embedded in Adaptive Threshold Network

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

Global receptive field graph attention network for unsupervised domain adaptation fault diagnosis in variable operating conditions

Meta-Learning with Adaptive Learning Rates for Few-Shot Fault Diagnosis

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

Multi-Scale Recursive Semi-Supervised Deep Learning Fault Diagnosis Method with Attention Gate

Self-attention Metric Learning Based on Multi-Scale Feature Fusion for Few-shot Fault Diagnosis