水力发电学报
            首 页   |   期刊介绍   |   编委会   |   投稿须知   |   下载中心   |   联系我们   |   学术规范   |   编辑部公告   |   English

水力发电学报

• •    

水库出库流量预测的机器学习模型性能与可解释性研究

  

  • 出版日期:2026-09-03 发布日期:2026-09-03

Exploring the Performance and Interpretability of Machine‑Learning Models for Reservoir Outflow Forecasting

  • Online:2026-09-03 Published:2026-09-03

摘要: 为对比不同机器学习算法在水库出库流量预测精度与可解释性差异,基于三峡水库2008至2025年日尺度入库、出库流量及水位实测数据,选取极端梯度提升树(Extreme Gradient Boosting, XGBoost)、反向传播神经网络(Back Propagation, BP)、长短期记忆网络(Long Short-Term Memory, LSTM)、广义可加模型(Generalized Additive Model, GAM)四种机器学习模型,在全系列、汛期、枯水期三类水文情景下对比出库流量预测精度,并引入SHAP(SHapley Additive exPlanations, SHAP)加性解释模型解析各模型特征贡献机制。结果表明:XGBoost 在三类情景预测精度均最优(R2=0.803~0.884,RMSE=1420~3569 m3/s),RMSE均控制在实测流量均值20%以内。SHAP分析显示,四类模型特征响应趋势整体相近,但高水位条件下XGBoost、GAM与BP、LSTM的驱动方向相反。精度与可解释性在多情景下不完全一致,模型选型须将决策逻辑的物理合理性纳入评价体系。

Abstract: To compare the prediction accuracy and interpretability of different machine?learning algorithms for reservoir outflow forecasting, this study employs observed daily inflow, outflow, and current water level data of the Three Gorges Reservoir from 2008 to 2025. Four models—Extreme Gradient Boosting (XGBoost), Back Propagation (BP) neural network, Long Short-Term Memory (LSTM) network, and Generalized Additive Model (GAM)—are selected. Their outflow prediction accuracy is compared under three hydrological scenarios: the full period, the flood season (June–October), and the dry season (November–May of the following year). The SHapley Additive exPlanations (SHAP) framework is introduced to interpret the feature contribution mechanism of each model. The results show that XGBoost achieved the highest prediction accuracy in all three scenarios(R2=0.803~0.884,RMSE=1420~3569 m3/s),The RMSE values are all within 20% of the mean observed outflow. SHAP analysis reveals that the driving directions of XGBoost and GAM are opposite to those of BP and LSTM under high water levels, despite generally similar feature response trends across the four models. Given the misalignment between prediction performance and interpretability across scenarios, model selection must prioritize the physical plausibility of decision logic alongside accuracy metrics.

京ICP备13015787号-3
版权所有 © 2013《水力发电学报》编辑部
编辑部地址:中国北京清华大学水电工程系 邮政编码:100084 电话:010-62783813
本系统由北京玛格泰克科技发展有限公司设计开发  技术支持:support@magtech.com.cn