Machine fault diagnosis based on multi-head deep learning network

Abstract

Feature extraction and utilization is of great importance for the problem of machine fault diagnosis. In this paper, multihead deep learning network is proposed to achieve machine health status classification using features of different sizes. Firstly, statistical characteristics which reflect machine signal status of time domain and frequency domain are summarized to compose feature vectors as one-dimensional network input. Secondly, Mel power spectrum and its incremental characteristics are utilized as two-dimensional network input of three channels. Lastly, the multi-head network is introduced to analyze both one-dimensional and two-dimensional features using two different sub neural networks and classify the machine health status according to the joint feature analyzing result. The experiments on bearing working status database of Case Western Reserve University show that the proposed method has good mechanical signal classification ability and better stability. Moreover, our final test accuracy of fault diagnosis on 16 kinds of bearing working signals can reach up to about 99.53%.