☆ 4.7 Article

Metric-based meta-learning model for few-shot fault diagnosis under multiple limited data conditions

MECHANICAL SYSTEMS AND SIGNAL PROCESSING (2021)

Journal

MECHANICAL SYSTEMS AND SIGNAL PROCESSING

Volume 155, Issue -, Pages -

Publisher

ACADEMIC PRESS LTD- ELSEVIER SCIENCE LTD

DOI: 10.1016/j.ymssp.2020.107510

Keywords

Metric-based meta-learning; Few-shot learning; Feature space; Fault diagnosis; Limited data conditions

Funding

National Key Research and Development Program of China [2017YFB0602700]

Ask authors/readers for more resources

Protocol

Community support

Reagent

Community support

Automated Summary New
Abstract

The article proposes a Feature Space Metric-based Meta-learning Model (FSM3) to address the challenge of few-shot fault diagnosis. Experimental results demonstrate that the method outperforms baseline methods in fault diagnosis tasks under various limited data conditions. Additionally, the time complexity and implementation difficulty have been analyzed to show the relatively high feasibility of the method.

The real-world large industry has gradually become a data-rich environment with the development of information and sensor technology, making the technology of data-driven fault diagnosis acquire a thriving development and application. The success of these advanced methods depends on the assumption that enough labeled samples for each fault type are available. However, in some practical situations, it is extremely difficult to collect enough data, e.g., when the sudden catastrophic failure happens, only a few samples can be acquired before the system shuts down. This phenomenon leads to the few-shot fault diag-nosis aiming at distinguishing the failure attribution accurately under very limited data conditions. In this paper, we propose a new approach, called Feature Space Metric-based Meta-learning Model (FSM3), to overcome the challenge of the few-shot fault diagnosis under multiple limited data conditions. Our method is a mixture of general supervised learning and episodic metric meta-learning, which will exploit both the attribute informa-tion from individual samples and the similarity information from sample groups. The experiment results demonstrate that our method outperforms a series of baseline methods on the 1-shot and 5-shot learning tasks of bearing and gearbox fault diagnosis across var-ious limited data conditions. The time complexity and implementation difficulty have been analyzed to show that our method has relatively high feasibility. The feature embedding is visualized by t-SNE to investigate the effectiveness of our proposed model. (c) 2021 Elsevier Ltd. All rights reserved.

Authors

I am an author on this paper

Click your name to claim this paper and add it to your profile.

Reviews

Primary Rating

4.7

Not enough ratings

Secondary Ratings

Novelty

Significance

Scientific rigor

Rate this paper

Meta-Generating Deep Attentive Metric for Few-Shot Classification

Fei Zhou, Lei Zhang, Wei Wei

Summary: Learning to generate a task-aware base learner is a promising direction for tackling the few-shot learning problem. Existing methods that use a fixed metric or classifier often fail to generalize to challenging cases due to their limited discriminative capacity. To address this, we propose a novel deep metric meta-generation method that learns to adaptively generate a specific metric for a new few-shot learning task. By utilizing a three-layer deep attentive network, our method is able to generate a discriminative metric for each task. Additionally, we introduce a tailored variational autoencoder that establishes a multi-modal weight distribution conditioned on cross-class sample pairs, allowing us to better adapt to new few-shot learning tasks with satisfactory generalization performance.

IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY (2022)