Uncertainty quantification of a deep learning model for failure rate prediction of water distribution networks

Predicting the time-dependent pipe failure rate of the water distribution networks (WDNs) is important for planning its renewal budget but also challenging due to the complex factors involved. The recent development of machine learning techniques provides a novel approach for accurate failure rate prediction based on historical data. However, the inherited randomness and uncertainty of water pipe failures and machine learning algorithms are often ignored in the training and prediction process. This article develops an uncertainty quantification framework for deep learning-based WDN system-wide failure rate prediction. The framework integrates a probabilistic long short-term memory (LSTM) model with a Monte Carlo method. The historical climate data and WDN pipe maintenance data over the past 35 years for Cuyahoga County, USA, are used to illustrate this deep-learning model-based uncertainty quantification framework in this study. A statistical time series regression model, ARIMAX, is used as a comparison benchmark. The results show that the LSTM model outperforms the ARIMAX model in prediction accuracy in most years by considering the uncertainties. Besides, the uncertainty range of the LSTM model prediction is 50% of that of the ARIMAX model. The results also identified the major contributing factors to the uncertainties of LSTM machine learning model prediction. The proposed uncertainty framework features excellent extensibility and can be adapted to quantify uncertainties with other types of machine learning models.