Few-Shot Cross-Domain Fault Diagnosis of Bearing Driven By Task-Supervised ANIL

Haidong Shao, Xiangdong Zhou, Jian Lin, Bin Liu

Research output: Contribution to journalArticlepeer-review

127 Citations (Scopus)

Abstract

Meta-learning has effectively addressed the limit of deep learning fault diagnosis models that demands a large number of samples. However, existing meta-learning models lack the capacity of feature reuse and task adaptability. To address the cross-domain fault diagnosis tasks with small samples, the feature reuse capability and task adaptability of existing meta-learning models need further improvements. To achieve this goal, this article introduces a new approach built upon the task-supervised almost no inner loop (ANIL). The proposed approach adopts a residual network to optimize the backbone structure of the inner loop, enhancing the feature reuse capability of the meta-learning in the unknown domain. An auxiliary term is introduced to define a supervised task-adaptive loss function, further updating the weight parameters of the inner loop meta-learner by monitoring the states of all meta-diagnostic tasks. The proposed method is used to analyze vibration signals from various bearings. The results demonstrate its superiority over traditional meta-learning methods in multiple sets of cross-domain fault diagnosis tasks with small samples.

Original languageEnglish
Pages (from-to)22892-22902
Number of pages11
JournalIEEE Internet of Things Journal
Volume11
Issue number13
Early online date31 Jan 2024
DOIs
Publication statusPublished - 1 Jul 2024

Funding

This work was supported in part by the National Natural Science Foundation of China under Grant 52275104, and in part by the Science and Technology Innovation Program of Hunan Province under Grant 2023RC3097.

Keywords

  • Task analysis
  • Adaptation models
  • Fault diagnosis
  • Feature extraction
  • Metalearning
  • training
  • optimization

Fingerprint

Dive into the research topics of 'Few-Shot Cross-Domain Fault Diagnosis of Bearing Driven By Task-Supervised ANIL'. Together they form a unique fingerprint.

Cite this