Wavelet-Prototypical Network Based on Fusion of Time and Frequency Domain for Fault Diagnosis.

Yu Wang,Lei Chen,Yang Liu,Lipeng Gao
DOI: https://doi.org/10.3390/s21041483
IF: 3.9
2021-01-01
Sensors
Abstract:Neural networks for fault diagnosis need enough samples for training, but in practical applications, there are often insufficient samples. In order to solve this problem, we propose a wavelet-prototypical network based on fusion of time and frequency domain (WPNF). The time domain and frequency domain information of the vibration signal can be sent to the model simultaneously to expand the characteristics of the data, a parallel two-channel convolutional structure is proposed to process the information of the signal. After that, a wavelet layer is designed to further extract features. Finally, a prototypical layer is applied to train this network. Experimental results show that the proposed method can accurately identify new classes that have never been used during the training phase when the number of samples in each class is very small, and it is far better than other traditional machine learning models in few-shot scenarios.
What problem does this paper attempt to address?