Temporal Dynamics and Predictive Modelling of Streamflow and Water Quality Using Advanced Statistical and Ensemble Machine Learning Techniques

Abstract

Changes in water quality are closely linked to seasonal fluctuations in streamflow, and a thorough understanding of how these variations interact across different time scales is important for the efficient management of surface water bodies such as rivers, lakes, and reservoirs. The aim of this study is to explore the potential connection between streamflow, rainfall, and water quality and propose an optimised ensemble model for the prediction of a water quality index (WQI). This study modelled the changes in five water quality parameters such as ammonia nitrogen (NH3-N), phosphate (PO43−), pH, turbidity, total dissolved solids (TDS), and their associated WQI caused by rainfall and streamflow. The analysis was conducted across three temporal scales, weekly, monthly, and seasonal, using a generalised additive model (GAM) in Toowoomba, Australia. TDS, turbidity, and WQI exhibited a significant nonlinear variation with the changes in streamflow in the weekly and monthly scales. Additionally, pH demonstrated a significant linear to weakly linear correlation with discharge across the three temporal scales. For the accurate prediction of WQI, this study proposed an ensemble model integrating an extreme gradient boosting (XGBoost) and Bayesian optimisation (BO) algorithm, using streamflow as an input across the same temporal scales. The results for the three temporal scales provided the best accuracy of monthly data, based on the accuracy metrics R2 (0.91), MAE (0.20), and RMSE (0.42). The comparison between the test and predicted data indicated that the prediction model overestimated the WQI at some points. This study highlights the efficiency of integrating rainfall, streamflow, and water quality correlations for WQI prediction, which can provide valuable insights for guiding future water management strategies in similar catchment areas, especially amidst changing climatic conditions.

Keywords

• streamflow
• water quality
• generalised additive model
• XGBoost regressor
• Bayesian optimisation