A novel hybrid model for long-term water quality prediction with the ‘decomposition–inputs–prediction’ hierarchical optimization framework

Abstract

Accurate, stable, and long-term water quality predictions are essential for water pollution warning and efficient water environment management. In this study, a hierarchical water quality prediction (HWQP) model was developed based on ‘data decomposition–predictor screening–efficient prediction’ via wavelet decomposition, Spearman correlation analysis, and long short-term memory network, respectively. The observed data from 14 stations in the Huaihe River–Hongze Lake system, including ammonia nitrogen (AN) and chemical oxygen demand (COD), were used to make long-term water quality predictions. The results suggested that, compared to existing water quality prediction models, the HWQP model has higher accuracy, with the root mean square errors of 6 and 17% for simulating AN and COD, respectively. The AN and COD concentrations will range from 0 to 1 mg/l and from 3 to 5 mg/l at 12 stations, respectively, and the COD concentrations will exceed the water quality target at Stations 4 and 5. The established model has great potential to address the challenges associated with the water environment.