Machine learning based water pipe failure prediction: The effects of engineering, geology, climate and socio-economic factors

Abstract

Underground water pipes deteriorate under the influence of various physical, mechanical, environmental, and social factors. Reliable pipe failure prediction is essential for a proactive management strategy of the water supply network (WSN), which is challenging for the conventional physics-based model. This study applied data-driven machine learning (ML) models to predict water pipe failures by leveraging the historical maintenance data heritage of a large water supply network. A multi-source data-aggregation framework was firstly established to integrate various contributing factors to underground pipe deterioration. The framework defined criteria for the integration of various data sources including the historical pipe break dataset, soil type dataset, topographical dataset, census dataset, and climate dataset. Based on the data, five ML algorithms, including LightGBM, Artificial Neural Network, Logistic Regression, K-Nearest Neighbors, and Support Vector Classification are developed for pipe failure prediction. LightGBM was found to achieve the best performance. The relative importance of major contributing factors on the water pipe failures was analyzed. Interestingly, the socioeconomic factors of a community are found to affect the probability of pipe failures. This study indicates that data-driven analysis that integrates the Machine Learning (ML) techniques and the proposed data integration framework has the potential to support reliable decision-making in WSN management.