Kalman filtering assimilated machine learning methods significantly improve the prediction performance of water quality parameters

作者：Zhenyu Gao, Guoqiang Wang, Jinyue Chen, Lei Fang, Shilong Ren, A Yinglan, Shuping Ji, Ruobing Liu, Qiao Wang · 发表于：Ecological Informatics · 年份：2025 · DOI：10.1016/j.ecoinf.2025.103337 · 被引用次数：4 · 研究领域：Hydrological Forecasting Using AI、Water Quality Monitoring Technologies、Water Quality Monitoring and Analysis

Accurate water quality prediction is essential for effective water pollution prevention and emergency responses. However, existing research on machine learning (ML)-based data assimilation methods remains limited, particularly in terms of addressing the combined impacts of climate change and anthropogenic activities. To address this gap, we proposed a novel ‘ML–Kalman filter (KF)’ data assimilation framework and evaluated its performance in the Dahei River Basin, a representative semi-arid watershed. Our results demonstrated significant improvements in predicting key water quality parameters, including total nitrogen (TN), total phosphorus (TP), and the permanganate index (COD Mn ), through the integration of KF with four ML models (LSTM, RF, XGBoost, and SVR). The accuracy enhancement ranged from 4.3 % to 17.6 %, with TP showing the most substantial improvement (9.2 %–17.6 %), followed by TN (6.4 %–11.1 %) and COD Mn (4.3 %–12.1 %). After assimilation, the models exhibited the following performance ranking for TN based on the coefficient of determination (R 2 ): LSTM–KF (R 2 = 0.909) > RF–KF (R 2 = 0.886) > SVR–KF (R 2 = 0.840) > XGBoost–KF (R 2 = 0.797), with similar trends observed for TP and COD Mn . The proposed framework demonstrates strong portability and applicability across different monitoring sections and temporal resolutions, offering a robust solution for regions with limited monitoring capabilities and challenging climatic conditions. These findings provide valu...