Novel Convolutional Neural Network (NCNN) for the Diagnosis of Bearing Defects in Rotary Machinery

Abstract

This work presents the development of novel convolutional neural network (NCNN) for effective identification of bearing defects from small samples. For effective feature learning from small training data, cost function of convolution neural network (CNN) is modified by adding additional sparsity cost in the existing cost function. A novel trigonometric cross-entropy function is developed to compute the sparsity cost. The proposed cost function introduces sparsity by avoiding unnecessary activation of neurons in the hidden layers of CNN. For identification of bearing defects from small training samples, NCNN-based transfer learning is applied in the following manner. First, the raw vibration signals as well as envelope signals from source domain machine are obtained. Thereafter, these envelope signals are applied to NCNN for the learning of features from the big training data acquitted from the source domain. After feature learning, knowledge gained from NCNN is transferred to do fine-tuning of NCNN from small training samples of target domain. Thereafter, defect identification is carried out by applying the test data of target domain to fine-tuned NCNN. The experimental result validates that the proposed cross-entropy function introduces sparsity in CNN and, hence, creates an effective deep learning which can even work under a situation when training data are not available in abundant.