JOURNAL OF YANGTZE RIVER SCIENTIFIC RESEARCH INSTI ›› 2020, Vol. 37 ›› Issue (10): 21-27.DOI: 10.11988/ckyyb.201908087

• WATER RESOURCES AND ENVIRONMENT • Previous Articles     Next Articles

Water Quality Prediction Using an ARIMA-SVR Hybrid Model

LUO Xue-ke1, HE Yun-xiao2, LIU Peng2, LI Wen2   

  1. 1. Beijing Institute of Graphic Communication, Beijing 102600, China;
    2. Research Institute of Electromechanical Engineering, North China University of Technology, Beijing 100144, China
  • Received:2019-07-11 Revised:2019-08-23 Online:2020-10-01 Published:2020-10-29

Abstract: The research aims to tackle the difficulty in obtaining the mechanism of water quality change in complex waters and modeling water quality prediction as well as the low accuracy of prediction. A water quality prediction method integrating Autoregressive Integrated Moving Average (ARIMA) model and Support Vector Machine Regression (SVR) model is established. After pretreatment, the data are linearly fitted by ARIMA model, and then the residual is predicted by SVR model to compensate for the non-linear change. The monitoring data of pH value and dissolved oxygen in Chaohu Lake basin from 2004 to 2015 are selected as experimental samples. Analysis by HP filtering method shows that the two sets of data have different trends and fluctuation characteristics. The prediction effect of the hybrid model is analyzed through comparative study on evaluation indicators. The results conclude that the present model has high accuracy with the correlation coefficients between predicted values and observed values of pH value and dissolved oxygen both reaching 0.99, the root mean square error 0.20 and 0.61, and the mean absolute percentage error 2.2% and 6.6%, respectively. The ARIMA-SVR hybrid model is of high prediction accuracy and strong generalization ability, suitable for water quality prediction in complex waters.

Key words: water quality prediction, ARIMA, SVR, hybrid model, dissolved oxygen

CLC Number: