Data-Driven Prognostics of Industrial Pumps: Condition Assessment and Time to Change Estimation Using Ensemble Methods

Document Type : Original Article

Authors

1 Department of Industrial Engineering, University of Kurdistan

2 Department of Mechanical Engineering, Islamic Azad University, Science and Research Branch, Tehran

10.22116/jiems.2026.556663.1623
Abstract
Abstract
Unplanned pump failures inflict high operational costs, rendering traditional maintenance strategies ‎inefficient. While Machine Learning (ML) has advanced fault diagnosis, a critical gap remains in ‎simultaneously classifying operational states and predicting the "Time Remaining until a State Change" ‎‎(TRSC) using real-world, imbalanced data. This study addresses this necessity by developing an integrated ‎predictive maintenance framework for industrial centrifugal pumps. Leveraging a four-year vibration ‎dataset (2020–2024), we employ Random Forest (RF), XGBoost, and Multi-Layer Perceptron (MLP), ‎utilizing SMOTE and jittering augmentation to mitigate data scarcity and imbalance. The study makes two ‎primary contributions: (1) accurate classification of four operational states (Normal to Failure), and (2) ‎precise regression of TRSC to optimize spare parts logistics. Results indicate that ensemble models ‎‎(RF/XGBoost) achieve classification accuracies exceeding 92% and TRSC prediction with an RMSE ‎below 25 days, significantly outperforming MLP. Furthermore, SHAP analysis reveals that horizontal and ‎axial vibrations are the dominant precursors to failure. These findings offer a robust, interpretable tool for ‎shifting towards condition-based maintenance, ensuring reliability and cost efficiency.‎

Keywords


Albuflaseh, Abdulrahman. 2025. “Machinery Imbalance Failure Prediction Based on Pump – Motor Vibration Readings.” Rochester Institute of Technology. Accessed From. https://repository.rit.edu/theses/12056/.
Bergstra, James, and Yoshua Bengio. 2012. “Random Search for Hyper-Parameter Optimization.” The Journal of Machine Learning Research 13 (1): 281–305.
Bhattarai, Aasma, Amit Gupta, Abishek Kafle, Prajwal Sapkota, Sailesh Chitrakar, Ole Gunnar Dahlhaug, and Suman Pradhan. 2023. “Application of Machine Learning Algorithm for Fault Detection in Pump.” In International Conference on the Efficiency and Performance Engineering Network, 235–47. Springer.
Breiman, Leo. 2001. “Random Forests.” Machine Learning 45: 5–32.
Cabrera, D., Villacís, M., Cerrada, M., Sánchez, R., Li, C., Sancho, F., Long, J., & Estupiñan, E. (2023). Adversarial Fault Detector Guided by One-Class Learning for a Multistage Centrifugal Pump. IEEE/ASME Transactions on Mechatronics, 28, 1395-1403. https://doi.org/10.1109/TMECH.2022.3226347.
Chao, Q., Tao, J., Wei, X., & Liu, C. (2020). Identification of cavitation intensity for high-speed aviation hydraulic pumps using 2D convolutional neural networks with an input of RGB-based vibration data. Measurement Science and Technology, 31. https://doi.org/10.1088/1361-6501/ab8d5a.
Chawla, Nitesh V, Kevin W Bowyer, Lawrence O Hall, and W Philip Kegelmeyer. 2002. “SMOTE: Synthetic Minority over-Sampling Technique.” Journal of Artificial Intelligence Research 16: 321–57.
Chen, Tianqi, and Carlos Guestrin. 2016. “Xgboost: A Scalable Tree Boosting System.” In Proceedings of the 22nd Acm Sigkdd International Conference on Knowledge Discovery and Data Mining, 785–94.
Diaby, Hadi, Derek D. Snaidauf, Don Williams, Edward Bowen, and Ephraim Schoenbrun. 2023. “Using AI-Enabled Predictive Maintenance to Help Maximize Asset Value.” www.deloitte.com/us/AIInstitute.
Dutta, N., Kaliannan, P., & Paramasivam, S. (2022). A comprehensive review on fault detection and analysis in the pumping system. International Journal of Ambient Energy, 43, 6878 - 6898. https://doi.org/10.1080/01430750.2022.2056917.
Ehsan Heydari. 2025. “Pump Vibration Data at 4 Points and Three Different Directions.” figshare. https://doi.org/https://doi.org/10.6084/m9.figshare.29553872.v1.
EN 13306:2017. 2017. “BSI Standards Publication Maintenance Maintenance. Maintenance Terminology. BS EN 13306:2017.” British Standard Institution, London, UK.
Fabić, Marko. 2025. “Application of AI-Based Predictive Maintenance for Industrial Processes.” Tehnicki Glasnik 19: 154–58. https://doi.org/10.31803/tg-20250404124706.
Gao, Yangde, Zahoor Ahmad, and Jong Myon Kim. 2024. “Fault Diagnosis of Rotating Machinery Using an Optimal Blind Deconvolution Method and Hybrid Invertible Neural Network.” Sensors 24 (1). https://doi.org/10.3390/s24010256.
Garousi, Masoud Hatami, Mahdi Karimi, Paolo Casoli, Massimo Rundo, and Rasoul Fallahzadeh. 2024. “Vibration Analysis of a Centrifugal Pump with Healthy and Defective Impellers and Fault Detection Using Multi-Layer Perceptron.” Eng 5 (4): 2511–30. https://doi.org/10.3390/eng5040131.
Gawde, Shreyas, Shruti Patil, Satish Kumar, Pooja Kamat, Ketan Kotecha, and Sultan Alfarhood. 2024. “Explainable Predictive Maintenance of Rotating Machines Using LIME, SHAP, PDP, ICE.” IEEE Access 12 (March): 29345–61. https://doi.org/10.1109/ACCESS.2024.3367110.
Goodfellow, Ian, Yoshua Bengio, Aaron Courville, and Yoshua Bengio. 2016. Deep Learning. Vol. 1. MIT press Cambridge.
Henderson, James, and Mark Sanders. 2025. “AI Driven Predictive Maintenance: Reducing Downtime and Enhancing Productivity in Manufacturing Environments,” 0–15. https://doi.org/10.20944/preprints202504.0602.v1.
Hu, Q., Ohata, E., Silva, F., Ramalho, G., Han, T., & Filho, P. (2020). A new online approach for classification of pumps vibration patterns based on intelligent IoT system. Measurement, 151, 107138. https://doi.org/10.1016/j.measurement.2019.107138.
Iglewicz, Boris, and David Hoaglin. 1993. “The ASQC Basic References in Quality Control: Statistical Techniques.” How to Detect and Handle Outliers 16: 1–87.
Johnson, Justin M, and Taghi M Khoshgoftaar. 2019. “Survey on Deep Learning with Class Imbalance.” Journal of Big Data 6 (1): 1–54.
Karakulov, I. (2022). CLASSIFICATION OF THE TECHNICAL CONDITION OF WATER PUMPING EQUIPMENT USING CONVOLUTIONAL NEURAL NETWORKS. Applied Mathematics and Control Sciences. https://doi.org/10.15593/2499-9873/2022.2.02.
Kazemi, Z., Andersen, P., Lemming, J., & Larsen, P. (2023). Artificial Intelligence-Based Diagnosis of Hydraulic Centrifugal Pumps Using Vibration Sensor: Feature Extraction and Failure Classification. 2023 IEEE 32nd International Symposium on Industrial Electronics (ISIE), 1-6. https://doi.org/10.1109/ISIE51358.2023.10227930.
Khan, Khalid, Muhammad Sohaib, Azaz Rashid, Saddam Ali, Hammad Akbar, Abdul Basit, and Tanvir Ahmad. 2021. “Recent Trends and Challenges in Predictive Maintenance of Aircraft’s Engine and Hydraulic System.” Journal of the Brazilian Society of Mechanical Sciences and Engineering 43 (8): 1–17. https://doi.org/10.1007/s40430-021-03121-2.
Kim, Heonho, Taewoong Ryu, Chanhee Lee, Hyeonmo Kim, Eunchul Yoon, Bay Vo, Jerry Chun-Wei Lin, and Unil Yun. 2022. “EHMIN: Efficient Approach of List Based High-Utility Pattern Mining with Negative Unit Profits.” Expert Systems with Applications 209: 118214.
Konieczny, J., & Stojek, J. (2021). Use of the K-Nearest Neighbour Classifier in Wear Condition Classification of a Positive Displacement Pump. Sensors (Basel, Switzerland), 21. https://doi.org/10.3390/s21186247.
Kumar, Shahil, Krish Kumar Raj, Maurizio Cirrincione, Giansalvo Cirrincione, Vincenzo Franzitta, and Rahul Ranjeev Kumar. 2024. “A Comprehensive Review of Remaining Useful Life Estimation Approaches for Rotating Machinery.” Energies 17 (22): 5538.
Lundberg, S. M., & Lee, S. I. (2017). A unified approach to interpreting model predictions. Advances in neural information processing systems, 30.
https://doi.org/10.48550/arXiv.1705.07874
Maguluri, L., Suganthi, D., Dhote, G., Kapila, D., Jadhav, M., & Neelima, S. (2024). AI-enhanced predictive maintenance in hybrid roll-to-roll manufacturing integrating multi-sensor data and self-supervised learning. The International Journal of Advanced Manufacturing Technology. https://doi.org/10.1007/s00170-024-14263-7.
McKinney, Wes. 2010. “Data Structures for Statistical Computing in Python.” Scipy 445 (1): 51–56.
Menanno, Marialuisa, and Anita Salsano. 2023. “Centrifugal Pump Fault Diagnosis Using a Predictive Maintenance Model,” no. July 2023. https://doi.org/10.46254/eu6.20230422.
Mohammed, Awsan. 2023. “Data Driven-Based Model for Predicting Pump Failures in the Oil and Gas Industry.” Engineering Failure Analysis 145: 107019.
Ney, J., & Wehn, N. (2025). ECNN: A Low-complex, Adjustable CNN for Industrial Pump Monitoring Using Vibration Data. 2025 IEEE Symposium on Computational Intelligence on Engineering/Cyber Physical Systems Companion (CIES Companion), 1-5. https://doi.org/10.1109/CIESCompanion65073.2025.11010817.
Nugroho, Waego Hadi, Samingun Handoyo, Hsing-Chuan Hsieh, Yusnita Julyarni Akri, and Donna DwinitaAdelia. 2022. “Modeling Multioutput Response Uses Ridge Regression and MLP Neural Network with Tuning Hyperparameter through Cross Validation.” International Journal of Advanced Computer Science and Applications 13 (9).
Oguzhan das, et al. 2023. “Machine Learning for Fault Analysis in Rotating Machinery: A Comprehensive Review.” Heliyon 9 (6): e17584. https://doi.org/10.1016/j.heliyon.2023.e17584.
Patil, S., & Jadhav, M. (2025). An Overview on Industrial Waste Water Submersible Pump. International Journal on Science and Technology. https://doi.org/10.71097/ijsat.v16.i1.1793.
Pedregosa, Fabian, Gaël Varoquaux, Alexandre Gramfort, Vincent Michel, Bertrand Thirion, Olivier Grisel, Mathieu Blondel, Peter Prettenhofer, Ron Weiss, and Vincent Dubourg. 2011. “Scikit-Learn: Machine Learning in Python.” The Journal of Machine Learning Research 12: 2825–30.
Peng, G., Huang, X., Zhou, L., Zhou, G., & Zhou, H. (2020). Solid-liquid two-phase flow and wear analysis in a large-scale centrifugal slurry pump. Engineering Failure Analysis, 114, 104602. https://doi.org/10.1016/j.engfailanal.2020.104602.
Powers, David M W. 2020. “Evaluation: From Precision, Recall and F-Measure to ROC, Informedness, Markedness and Correlation.” ArXiv Preprint ArXiv:2010.16061.
Prabu, S., Senthilraja, R., Ali, A., Jayapoorani, S., & Arun, M. (2025). AI-Driven Predictive Maintenance for Smart Manufacturing Systems Using Digital Twin Technology. International Journal of Computational and Experimental Science and Engineering. https://doi.org/10.22399/ijcesen.1099.
Rana, S. (2025). AI-DRIVEN FAULT DETECTION AND PREDICTIVE MAINTENANCE IN ELECTRICAL POWER SYSTEMS: A SYSTEMATIC REVIEW OF DATA-DRIVEN APPROACHES, DIGITAL TWINS, AND SELF-HEALING GRIDS. American Journal of Advanced Technology and Engineering Solutions. https://doi.org/10.63125/4p25x993.
Riedemann, J., Reyes, E., Sarasiri, N., Pena, R., & Andrade, I. (2024). Renewable Source-Based Water Pumping Electrification in Mines: Current technologies and future trends. IEEE Electrification Magazine, 12, 38-44. https://doi.org/10.1109/MELE.2023.3348349.
Rimal, Yagyanath, Navneet Sharma, and Abeer Alsadoon. 2024. “The Accuracy of Machine Learning Models Relies on Hyperparameter Tuning: Student Result Classification Using Random Forest, Randomized Search, Grid Search, Bayesian, Genetic, and Optuna Algorithms.” Multimedia Tools and Applications 83 (30): 74349–64.
Rojas, L., Peña, Á., & García, J. (2025). AI-Driven Predictive Maintenance in Mining: A Systematic Literature Review on Fault Detection, Digital Twins, and Intelligent Asset Management. Applied Sciences. https://doi.org/10.3390/app15063337.
Rojek, Izabela, Małgorzata Jasiulewicz-Kaczmarek, Mariusz Piechowski, and Dariusz Mikołajewski. 2023. “An Artificial Intelligence Approach for Improving Maintenance to Supervise Machine Failures and Support Their Repair.” Applied Sciences (Switzerland) 13 (8). https://doi.org/10.3390/app13084971.
Rossum, Van. 2009. “Python 3 Reference Manual.” (No Title).
Saeed, Ali, Muazzam A. Khan, Usman Akram, Waeal J. Obidallah, Soyiba Jawed, and Awais Ahmad. 2025. “Deep Learning Based Approaches for Intelligent Industrial Machinery Health Management and Fault Diagnosis in Resource-Constrained Environments.” Scientific Reports 15 (1): 1114.
Sarwar, Ayesha, Muhammad Faheem Mushtaq, Urooj Akram, Furqan Rustam, Ameer Hamza, Vaibhav Rupapara, and Saleem Ullah. 2023. “IoT Networks Attacks Detection Using Multi-Novel Features and Extra Tree Random-Voting Ensemble Classifier (ER-VEC).” Journal of Ambient Intelligence and Humanized Computing 14 (12): 16637–51.
Sathiya, V, M Chinnadurai, and S Ramabalan. 2022. “Mobile Robot Path Planning Using Fuzzy Enhanced Improved Multi-Objective Particle Swarm Optimization (FIMOPSO).” Expert Systems with Applications 198: 116875.
Shorten, Connor, and Taghi M Khoshgoftaar. 2019. “A Survey on Image Data Augmentation for Deep Learning.” Journal of Big Data. https://doi.org/10.1186/s40537-019-0197-0.
Saxena, A. (2025). Application of Data Science in Pump Maintenance. International Journal For Multidisciplinary Research. https://doi.org/10.36948/ijfmr.2025.v07i01.37055.
Simion, D., Postolache, F., Fleacă, B., & Fleacă, E. (2024). AI-Driven Predictive Maintenance in Modern Maritime Transport—Enhancing Operational Efficiency and Reliability. Applied Sciences. https://doi.org/10.3390/app14209439.
Singh, Dalwinder, and Birmohan Singh. 2020. “Investigating the Impact of Data Normalization on Classification Performance.” Applied Soft Computing 97: 105524.
Thakkar, D., & Kumar, R. (2024). AI-Driven Predictive Maintenance for Industrial Assets using Edge Computing and Machine Learning. Journal for Research in Applied Sciences and Biotechnology. https://doi.org/10.55544/jrasb.3.1.55.
Ucar, Aysegul, Mehmet Karakose, and Necim Kırımça. 2024. “Applied Sciences Artificial Intelligence for Predictive Maintenance Applications:”
Um, Terry T, Franz M J Pfister, Daniel Pichler, Satoshi Endo, Muriel Lang, Sandra Hirche, Urban Fietzek, and Dana Kulić. 2017. “Data Augmentation of Wearable Sensor Data for Parkinson’s Disease Monitoring Using Convolutional Neural Networks.” In Proceedings of the 19th ACM International Conference on Multimodal Interaction, 216–20.
Vijayan, Midhula, Preeth Raguraman, and R Mohan. 2021. “A Fully Residual Convolutional Neural Network for Background Subtraction.” Pattern Recognition Letters 146: 63–69.
Vithi, Nontuthuzelo Lindokuhle, and Colin Chibaya. 2024. “Advancements in Predictive Maintenance: A Bibliometric Review of Diagnostic Models Using Machine Learning Techniques.” Analytics 3 (4): 493–507. https://doi.org/10.3390/analytics3040028.
Wada, I., Bello, S., Ige, O., Chianumba, E., & Adebayo, S. (2024). AI-driven predictive maintenance and optimization of renewable energy systems for enhanced operational efficiency and longevity. International Journal of Science and Research Archive. https://doi.org/10.30574/ijsra.2024.13.1.1992.
Xia, Yuxuan, Shanshan Jiang, Lingyi Meng, and Xin Ju. 2024. “XGBoost-B-GHM: An Ensemble Model with Feature Selection and GHM Loss Function Optimization for Credit Scoring.” Systems 12 (7).
Yu, B., Lee, J., Kim, Y., Lee, T., Kim, Y., Jang, H., Park, J., & Park, J. (2023). A Study on Fault Classification for Condition-Based Maintenance of Auxiliary Pump in Ship’s Engine Systems Using HHT and CNN. Journal of Applied Reliability. https://doi.org/10.33162/jar.2023.12.23.4.366.
Zhang, J., Zhang, H., He, Y., & Tao, W. (2016). A comprehensive review on advances and applications of industrial heat pumps based on the practices in China. Applied Energy, 178, 800-825. https://doi.org/10.1016/J.APENERGY.2016.06.049.
Zhi, Yifan, Qian Huang, Fudong Tao, Huairui Li, Da Hao, Haoyang Qin, and Qiang Fu. 2025. “Research on Fault Diagnosis of Vertical Centrifugal Pump Based on Multi-Channel Information Fusion.” Processes 13 (4). https://doi.org/10.3390/pr13041152.
Zhu, R., Li, Y., Huang, Q., Li, S., Zhang, X., Li, H., & Fu, Q. (2023). Research on Vertical SEC Centrifugal Pump Multi-Fault Diagnosis Based on WPT–SVM. E

Articles in Press, Accepted Manuscript
Available Online from 12 July 2026