A Curriculum-Based Meta-Learning Approach for Few-Shot Fault Diagnosis

Pei Lai,Fan Zhang,Tianrui Li,Jin Guo,Fei Teng,Ruiqi Liu
DOI: https://doi.org/10.1109/DSInS60115.2023.10455545
2023-01-01
Abstract:Safety-critical systems frequently encounter few-shot fault diagnosis (FSFD) scenarios, where conventional deep learning models may not be as effective. For effective diagnosis of bearings in few-shot scenarios, we present a curriculum-based meta-learning (CBML) approach. In this approach, the task in the meta-training phase is used as a curriculum. The diagnostic knowledge in the task is learned more sufficiently through a curriculum learning strategy. Specifically, a 1DCNN-based backbone network is designed and utilizes an N-way K-shot training mode to extract the diagnostic knowledge from the meta-training tasks. A curriculum metric based on the spectral norm is proposed, which is combined with a curriculum scheduler to implement curriculum learning. The results of the experiments performed on the CWRU and PU datasets demonstrate how effective the curriculum learning strategies are in a variety of FSFD scenarios.
What problem does this paper attempt to address?