An interpretable river water quality prediction model by integrating deep learning and large language models

Engineering notes

Engineering notes will be added by the aipentium editorial team.

Chinese explanation / 中文解读

中文解读待补充：本站会优先为大语言模型、生成式AI、ChatGPT相关技术、计算机视觉、深度学习等高价值论文补充中文说明。

Original abstract

Physical mechanism-based models for river water quality prediction involve complicated calculations，whereas machine learning models lack interpretability，resulting in a disconnection between predictions and management decisions that hinders practical application. To deeply integrate high-precision prediction with decision support，a TCN-Attention deep learning model that combines a temporal convolutional network and an attention mechanism is constructed to predict six core water quality indicators，with Bayesian optimization used for automatic parameter tuning. The SHAP interpretability technique is introduced to quantify the contributions of multi-source input features and reveal key driving factors. Based on the Qwen3-Next large language model，water quality grade evaluations and improvement recommendations are automatically generated. Application results for rivers in Xinwu District of Wuxi City demonstrate that the TCN-Attention model achieves good prediction performance under small-sample conditions，with domestic water use，turbidity，and water identified as the most critical influencing factors. Qwen3-Next model achieves an accuracy of 89.8% in water quality grading，and the proposed improvement recommendations are targeted and practicable. The proposed interpretable intelligent water quality prediction method effectively improves the accuracy and interpretability of urban river water quality predictions，providing a reliable technical pathway for smart water environment management.

Links and sources

Comments