Hybrid Deep Learning Model for Fault Diagnosis in Centrifugal Pumps: A Comparative Study of VGG16, ResNet50, and Wavelet Coherence Analysis
Wasim Zaman,Muhammad Farooq Siddique,Saif Ullah,Faisal Saleem,Jong-Myon Kim
DOI: https://doi.org/10.3390/machines12120905
IF: 2.899
2024-12-11
Machines
Abstract:Significant in various industrial applications, centrifugal pumps (CPs) play an important role in ensuring operational efficiency, yet they are susceptible to faults that can disrupt production and increase maintenance costs. This study proposes a robust hybrid model for accurate fault detection and classification in CPs, integrating Wavelet Coherence Analysis (WCA) with deep learning architectures VGG16 and ResNet50. WCA is initially applied to vibration signals, creating time–frequency representations that capture both temporal and frequency information, essential for identifying subtle fault characteristics. These enhanced signals are processed by VGG16 and ResNet50, each contributing unique and complementary features that enhance feature representation. The hybrid approach fuses the extracted features, resulting in a more discriminative feature set that optimizes class separation. The proposed model achieved a test accuracy of 96.39%, demonstrating minimal class overlap in t-SNE plots and a precise confusion matrix. When compared to the ResNet50-based and VGG16-based models from previous studies, which reached 91.57% and 92.77% accuracy, respectively, the hybrid model displayed better classification performance, particularly in distinguishing closely related fault classes. High F1-scores across all fault categories further validate its effectiveness. This work underscores the value of combining multiple CNN architectures with advanced signal processing for reliable fault diagnosis, improving accuracy in real-world CP applications.
engineering, mechanical, electrical & electronic