Abstract:Due to the complexity of electromechanical equipment and the difficulties in obtaining large-scale health monitoring datasets, as well as the long-tailed distribution of data, existing methods ignore certain characteristics of health monitoring data. In order to solve these problems, this paper proposes a method for the fault diagnosis of rolling bearings in electromechanical equipment based on an improved prototypical network-the weight prototypical networks (WPorNet). The main contributions of this paper are as follows: (1) the prototypical networks, which perform well on small-sample classification tasks, were improved by calculating the different levels of influence of support sample distributions in order to achieve the prototypical calculation. The change in sample influence was calculated using the Kullback-Leibler divergence of the sample distribution. The influence change in a specific sample can be measured by assessing how much the distribution changes in the absence of that sample; and (2) The Gramian Angular Field (GAF) algorithm was used to transform one-dimensional time series into two-dimensional vibration images, which greatly improved the application effect of the 2D convolutional neural network (CNN). Through experiments on MAFAULDA and CWRU bearing datasets, it was shown that this network effectively solves the shortcomings of a small number of valid samples and a long-tail distribution in health monitoring data, it enhances the dependency between the samples and the global data, it improves the model's feature extraction ability, and it enhances the accuracy of model classification. Compared with the prototypical network, the improved network model increased the performance of the 2-way 10-shot, 2-way 20-shot, and 2-way 50-shot classification tasks by 5.23%, 5.74%, and 4.37%, respectively, and it increased the performance of the 4-way 10-shot, 4-way 20-shot, and 4-way 50-shot classification tasks by 12.02%, 10.47%, and 4.66%, respectively. Experimental results show that the improved prototypical network model has higher sample classification accuracy and stronger anti-interference ability compared with traditional small-sample classification models.

A Fault Diagnosis Method Using Improved Prototypical Network and Weighting Similarity-Manhattan Distance with Insufficient Noisy Data

An Uncertainty Perception Metric Network for Machinery Fault Diagnosis under Limited Noisy Source Domain and Scarce Noisy Unknown Domain

Fault Diagnosis with Robustness and Lightweight Synergy under Noisy Environment

An improved prototypical network with L2 prototype correction for few-shot cross-domain fault diagnosis

Fault Diagnosis Method and Application Based on Multi-scale Neural Network and Data Enhancement for Strong Noise

Lightweight Network with Variable Asymmetric Rebalancing Strategy for Small and Imbalanced Fault Diagnosis

Method for Diagnosing Bearing Faults in Electromechanical Equipment Based on Improved Prototypical Networks

Multiscale Wavelet Prototypical Network for Cross-Component Few-Shot Intelligent Fault Diagnosis

Uncertainty-based Contrastive Prototype-Matching Network Towards Cross-Domain Fault Diagnosis with Small Data

A Prototype Correction Network with Coordinate Attention for Few-Shot Fault Diagnosis Under Varying Working Conditions

Mechanical equipment fault diagnosis method based on improved deep residual shrinkage network

A Novel Multiscale Lightweight Fault Diagnosis Model Based on the Idea of Adversarial Learning

Imbalanced Fault Diagnosis Based on Supervised Prototypical Contrastive Network

Lightweight Ghost Enhanced Feature Attention Network: An Efficient Intelligent Fault Diagnosis Method under Various Working Conditions

A Novel Brownian Correlation Metric Prototypical Network for Rotating Machinery Fault Diagnosis with Few and Zero Shot Learners

A Deep Convolutional Neural Network with New Training Methods for Bearing Fault Diagnosis under Noisy Environment and Different Working Load

A Multi-Scale Kernel-Based Network with Improved Attention Mechanism for Rotating Machinery Fault Diagnosis under Noisy Environments

A Prototypical Networks-based Multi-task Model for Few-shot Fault Diagnosis

An Anti-Noise One-Dimension Convolutional Neural Network Learning Model Applying on Bearing Fault Diagnosis

A multi-scale convolution neural network for featureless fault diagnosis

Dynamic Temporal Denoise Neural Network with Multi-Head Attention for Fault Diagnosis Under Noise Background