Water quality prediction using ARIMA-SSA-LSTM combination model

Autor: Tingyu Wang, Wei Chen, Bo Tang
Jazyk: angličtina
Rok vydání: 2024
Předmět:
Zdroj: Water Supply, Vol 24, Iss 4, Pp 1282-1297 (2024)
Druh dokumentu: article
ISSN: 1606-9749
1607-0798
DOI: 10.2166/ws.2024.060
Popis: The water quality index model is a popular tool for evaluating drinking water quality. To overcome low precision and significant errors in the traditional single prediction model, a novel autoregressive integrated moving average (ARIMA)-sparrow search algorithm (SSA)-long short-term memory (LSTM) combination model is proposed to accurately predict residual chlorine, turbidity, and pH in drinking water. First, the ARIMA model is used to extract the linear part of water quality data and output the nonlinear residual. Then, the LSTM model is used to predict the residual, and the SSA is used to find the optimal hyperparameters of the LSTM model, which plays an essential role in reducing the error of the model. To prove the superiority of the model developed, the ARIMA-SSA-LSTM model is compared with SSA-LSTM, whale optimization algorithm-LSTM, PSO-LSTM, ARIMA-LSTM, ARIMA, and LSTM. The results show that the coefficient of determination (R2) of the combination model for residual chlorine, turbidity, and pH are 0.950, 0.990, and 0.998, respectively, which are greater than all comparison models. Therefore, the model is more suitable for the prediction and analysis of water quality data. HIGHLIGHTS This article presents a new combined forecasting method to predict the trend of water quality in water distribution networks.; The autoregressive integrated moving average (ARIMA)-sparrow search algorithm (SSA)-long short-term memory (LSTM) combined model overcomes the limitations of the traditional single model.; SSA can find more suitable hyperparameters for LSTM model, so as to improve the prediction accuracy of LSTM model.;
Databáze: Directory of Open Access Journals
Externí odkaz: