arXiv:2511.01837cs.LG2025-11被引 1

用可解释模型揭示美国红河流域水库水温变化规律

Interpretable Machine Learning for Reservoir Water Temperatures in the U.S. Red River Basin of the South

  • 结合机器学习与符号建模,从数据中提取水温驱动因子
  • 最佳模型预测误差仅1.20℃,解释度达R²=0.97
  • 生成简洁可读的数学公式,适合水利与气候研究者

准确预测水库水温(RWT)对可持续水资源管理、生态系统健康和气候韧性至关重要。然而,单纯预测难以揭示其物理机制。本文整合可解释机器学习与符号建模,分析美国红河流域十座水库的超10,000条深度解析温度数据。先采用随机森林(RF)、XGBoost及多层感知机(MLP)等集成与神经网络模型,实现高预测精度(最低RMSE=1.20℃,R²=0.97)。通过SHAP方法量化空气温度、深度、风速、库容等物理因素贡献,发现跨水库模式一致。进一步构建柯尔莫哥洛夫-阿诺德网络(KAN),将数据驱动结果转化为紧凑解析表达式:从仅用7天前气温(R²=0.84)逐步扩展至十变量模型(R²=0.92),但五变量后增益递减。所得方程以线性与有理函数为主,逐步捕捉非线性特征,同时保持可解释性。深度始终为次重要但关键变量,降水影响有限。该框架将黑箱模型转化为透明代理模型,兼顾预测与理解,推动对水库热力动态的认知。

原文摘要 · Abstract (English)

Accurate prediction of Reservoir Water Temperature (RWT) is vital for sustainable water management, ecosystem health, and climate resilience. Yet, prediction alone offers limited insight into the governing physical processes. To bridge this gap, we integrated explainable machine learning (ML) with symbolic modeling to uncover the drivers of RWT dynamics across ten reservoirs in the Red River Basin, USA, using over 10,000 depth-resolved temperature profiles. We first employed ensemble and neural models, including Random Forest (RF), Extreme Gradient Boosting (XGBoost), and Multilayer Perceptron (MLP), achieving high predictive skill (best RMSE = 1.20 degree Celsius, R^2 = 0.97). Using SHAP (SHapley Additive exPlanations), we quantified the contribution of physical drivers such as air temperature, depth, wind, and lake volume, revealing consistent patterns across reservoirs. To translate these data-driven insights into compact analytical expressions, we developed Kolmogorov Arnold Networks (KANs) to symbolically approximate RWT. Ten progressively complex KAN equations were derived, improving from R^2 = 0.84 using a single predictor (7-day antecedent air temperature) to R^2 = 0.92 with ten predictors, though gains diminished beyond five, highlighting a balance between simplicity and accuracy. The resulting equations, dominated by linear and rational forms, incrementally captured nonlinear behavior while preserving interpretability. Depth consistently emerged as a secondary but critical predictor, whereas precipitation had limited effect. By coupling predictive accuracy with explanatory power, this framework demonstrates how KANs and explainable ML can transform black-box models into transparent surrogates that advance both prediction and understanding of reservoir thermal dynamics.

水温预测可解释AI符号建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。