融合深度学习和大语言模型的可解释河流水质预测模型

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

  • 摘要: 基于物理机制的河流水质预测模型计算复杂,机器学习模型可解释性不足,导致预测与管理决策“两张皮”,难以落地应用。为实现高精度预测与决策支持的深度结合,融合时间卷积网络和注意力机制构建TCN-Attention深度学习模型,预测6项核心水质指标,通过贝叶斯优化自动调优参数;引入SHAP可解释性技术量化多源输入特征贡献,揭示关键驱动因子;基于Qwen3-Next大语言模型自动生成水质等级评价与改善建议。新吴区河流应用结果表明:TCN-Attention在小样本条件下预测性能良好,生活用水量、浊度和水域被识别为最关键影响因子;Qwen3-Next水质等级评价准确率达89.8%,所提改善建议具有针对性与可操作性。可解释水质智能预测方法有效提升城市河流水质预测精度与可解释性,为智能化水环境管理提供可靠技术路径。

     

    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 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.

     

/

返回文章
返回