Fault diagnosis of air compressors using transfer learning: A comparative study of pre-trained networks and hyperparameter optimization

Abstract

Air compressors are critical components in many industries whose catastrophic failure results in huge financial losses and downtime leading to accidents. Hence, real time fault diagnosis of air compressor is essential to predict the health condition of air compressor and plan scheduled maintenance thereby reducing financial losses and accidents. Fault diagnosis using transfer learning aids in real time fault detection. In the present study, five air compressor conditions were considered namely, check valve fault, inlet and outlet reed valve fluttering fault, inlet reed valve fluttering fault, outlet reed valve fluttering fault, and good condition. The raw vibration data was converted to radar plot images that were pre-processed and classified using four pre-trained networks (ResNet-50, GoogLeNet, AlexNet, and VGG-16). The hyperparameters like epochs, batch size, optimizer, train-test split ratio, and learning rate were varied to find out the best network for air compressor fault diagnosis. ResNet-50 among all other pre-trained networks produced the maximum classification accuracy (average of five trials) of 98.72%.