基于XGBoost−TPE的导水裂隙带发育高度预测模型研究

    Research on prediction model of water-conducting fractured zone development height based on XGBoost−TPE

    • 摘要: 导水裂隙带发育高度的预测是煤矿安全开采与煤层顶板水害防治的重要基础工作。煤层覆岩导水裂隙带的发育机理复杂,且工作面地质工程条件存在显著差异,导致其发育高度的预测方法普遍存在预测精度偏低、泛化能力不足等问题。为了提高导水裂隙带高度预测的准确性和可靠性,收集114座矿井导水裂隙带高度实测数据311组构建样本数据集,涵盖采高、工作面尺寸、开采方式、推进速度、覆岩类型等13个影响因素。针对单一模型预测精度不足、泛化性差等问题,提出了融合极限梯度提升(eXtreme Gradient Boosting,XGBoost)与树结构概率密度估计(Tree−structured Parzen Estimator,TPE)超参数优化的导水裂隙带高度预测方法。首先对原始数据进行数据清洗、数据规范化等数据预处理,然后通过随机森林(Random Forest, RF)、梯度提升决策树(Gradient Boosting Decision Tree, GBDT)、XGBoost 3种算法综合筛选特征变量,利用TPE算法和K折交叉验证寻找最优超参数组合,建立XGBoost−TPE导水裂隙带高度预测模型,并对比了TPE、网格搜索、随机搜索和模拟退火4种优化算法对XGBoost模型预测精度的影响。基于8项评价指标与11种主流模型的对比试验结果显示,XGBoost 模型的平均绝对误差(MAE)、均方根误差(RMSE)、中位数绝对误差(MedAE)及训练时间(TT)较单一模型分别平均降低0.020、0.019、0.018 和 0.183,决定系数(R2)、离群值鲁棒决定系数(D2)则较单一模型分别提升0.118和0.176。研究结果表明,XGBoost集成学习模型具备更优的泛化能力、预测精度与可解释性。

       

      Abstract: Prediction of the development height of the water-conducting fractured zone is an important basic task for the safe mining of coal mines and the prevention and control of coal seam roof water disasters. The development mechanism of the water-conducting fractured zone in coal seam overlying strata is complex, and there are significant differences in the geological and engineering conditions of working faces, resulting in common problems such as low prediction accuracy and insufficient generalization ability in the prediction methods of water-conducting development height. To improve the accuracy and reliability of height prediction of the water-conducting fractured zone, 311 groups of measured data on the height of the water-conducting fractured zone were collected from 114 mines to construct a sample dataset covering 13 influencing factors including mining height, working face size, mining method, advancing speed, and overlying strata type. Aiming at the problems of insufficient prediction accuracy and poor generalization of single models, a prediction method for the height of the water-conducting fractured zone was proposed, which integrates eXtreme Gradient Boosting (XGBoost) and hyperparameter optimization based on Tree-structured Parzen Estimator (TPE). First, data preprocessing such as data cleaning and data standardization was performed on the raw data. Then, feature variables were comprehensively selected by three algorithms: Random Forest (RF), Gradient Boosting Decision Tree (GBDT), and XGBoost. The TPE algorithm and K-fold cross-validation were used to find the optimal hyperparameter combination, and the XGBoost-TPE prediction model for the height of the water-conducting fractured zone was established. The effects of four optimization algorithms (TPE, grid search, random search, and simulated annealing) on the prediction accuracy of the XGBoost model were compared. Comparative test results based on 8 evaluation indicators and 11 mainstream models show that, compared with single models, the XGBoost model reduces the mean absolute error (MAE), root mean square error (RMSE), median absolute error (MedAE), and training time (TT) by an average of 0.020, 0.019, 0.018, and 0.183, respectively, while the coefficient of determination (R2) and outlier-robust coefficient of determination (D2) are increased by 0.118 and 0.176, respectively. The results indicate that the XGBoost ensemble learning model has better generalization ability, prediction accuracy, and interpretability.

       

    /

    返回文章
    返回