ABSTRACT Traditional deep learning-based fault diagnosis methods for diesel engine valve leakage, often rely on large datasets. This study proposes the GAP-VGG19-BN method based on VGG19 and transfer learning. Global average pooling (GAP) is adopted to replace the full connection layer and improve parameter reduction. Batch Normalization (BN) layers are set after each convolution layer to enhance the computational speed. We build a small-scale dataset using the time–frequency and the wavelet methods constructed from the diesel engine cylinder head vibration signal. The GAP-VGG19-BN is pre-trained through ImageNet; then, the well-trained parameters are transferred onto the constructed dataset. The BN method is adopted to make the learning rate approach obtain an optimal global solution of learning rate. Experimental results show that the proposed method has higher accuracy, stronger robustness, less computational cost, and better generalisation for small-scale datasets. The proposed method enriches the fault diagnosis method of diesel engine valve leakage.
Read full abstract