Imbalanced Fault Identification via Embedding-augmented Gaussian Prototype Network with Meta-Learning Perspective
Abstract In practical engineering, the number of acquired fault samples from different categories might be in great difference due to the little probability of key equipment happening to malfunctioned. When training the imbalanced data, more methods focus on balancing the number of samples between different categories which may be time-consuming and easy to over-fit. To address this problem, we proposed Embedding-augmented Gaussian Prototype Network (EGPN) which applied a new training mechanism from the perspective of meta-learning. We only train the categories with large samples and the remaining categories only appeared in the testing process to calculate untrained prototypes. EGPN includes a feature embedding augmented module, weighted prototype module and metric module. Firstly, ordinary convolution and dilated convolution are mixed to capture different frequency bands simultaneously, and residual-attention module is added to highlight key features and suppress unimportant features. Different prototypes are calculated by weighting to the embedding vectors through Gaussian covariance matrix. Finally, the classification is taken according to the modified distance. The experiments in two datasets indicating that the proposed method can effectively recognize the untrained categories with only a few samples using as the prototypes, which can tackle the problem of identifying imbalanced fault data efficiently.