水利水电技术（中英文）2025，Vol.56Issue(8)：19-31,13.DOI:10.13928/j.cnki.wrahe.2025.08.002
基于决策树的水灾预测模型
Flood prediction model based on decision trees
摘要
Abstract
[Objective]Floods are natural disasters triggered by factors such as heavy rainfall,rapid snow and ice melt,and storm surges,often resulting in significant economic losses and severe disruption to daily life.Conventional flood prediction primarily relies on traditional hydrological method and experience-based statistical models.However,in areas lacking long-term and continuous hydrological monitoring data,alternative data-driven method for flood prediction are essential.[Methods]Machine learning algorithms based on decision trees,including Random Forest,XGBoost,and LightGBM,demonstrated excellent performance in classification and regression tasks due to their interpretability and strong functions,making them suitable for flood prediction.A dataset containing 50 000 records and 21 variables was used to evaluate the flood prediction performance of these three algorithms,namely Random Forest,XGBoost,and LightGBM.Their performance was assessed based on prediction accuracy and key variable identification,with the ROC-AUC curve used for comparative analysis.[Results]The result showed that all three models achieved high prediction accuracy.Among them,the XGBoost model exhibited the lowest mean squared error(0.000 186 2)and the highest coefficient of determination(0.925 2).Moreover,the LightGBM model achieved the highest AUC value(0.99)in the ROC-AUC curve.The Random Forest model underperformed the other two across all indicators.[Conclusion]The findings indicate that XGBoost delivers optimal performance for flood probability prediction with lowest prediction errors,while LightGBM is the optimal choice for binary classification tasks,such as predicting flood occurrence.关键词
洪水预测/决策树/随机森林/XGBoost/LightGBMKey words
flood prediction/decision trees/Random Forest/XGBoost/LightGBM分类
建筑与水利引用本文复制引用
左右..基于决策树的水灾预测模型[J].水利水电技术（中英文）,2025,56(8):19-31,13.基金项目
国家社会科学基金项目(21BTJ071) （21BTJ071）
