基于Optuna优化GBDT模型的静力触探土层分类研究

    Optuna-Optimized GBDT Approach for Soil Stratification Analysis in Cone Penetration Testing

    • 摘要: 传统的土层分类方法主要依赖于现场试验与专家评估的结合,然而这种方法往往伴随着较高的成本,并且在土层的划分过程中可能引入一定的主观性。近些年,随着大数据与人工智能的快速发展,机器学习在岩土领域展现出其独特的优势。为了克服传统土层分类方法的局限性,提出了一种基于Optuna超参数优化框架的梯度提升决策树(GBDT)算法模型,用于土层分类的预测。利用静力触探试验(CPT)所测得的数据,将其划分为训练集与测试集,并采用四个评估指标对预测结果进行全面评估。此外,将模型与KNN、RF、MLP、XGBoost、LightGBM与CatBoost几种不同算法的预测结果进行了比较。研究结果表明,经过Optuna调参框架优化后的GBDT算法模型,在预测效果上相较于其他几种模型展现出更优的性能,具有更高的预测精度和稳定性。在实际工程应用中,该算法模型的预测精度达到0.8以上,少部分达到0.9以上,具有较好的预测效果。因此,提出的Optuna优化GBDT算法模型为土层分类问题提供了一种有效的解决方案,并为该领域的进一步研究提供了有益的参考。

       

      Abstract: Traditional soil layer classification methods primarily rely on the combination of in-situ testing and expert evaluation. However, this approach is often associated with high costs and may introduce subjectivity during soil stratification. In recent years, with the rapid development of big data and artificial intelligence, machine learning has demonstrated unique advantages in geotechnical engineering. To overcome the limitations of conventional soil classification methods, this study proposes an Optuna hyperparameter-optimized Gradient Boosting Decision Tree (GBDT) algorithm model for soil layer classification prediction. Utilizing data obtained from Cone Penetration Testing (CPT), the dataset was divided into training and test sets, with four evaluation metrics employed to comprehensively assess prediction performance. Furthermore, the model was compared with several alternative algorithms including KNN, RF, MLP, XGBoost, LightGBM, and CatBoost. Results indicate that the Optuna-optimized GBDT model exhibits superior performance compared to other models, demonstrating higher prediction accuracy and stability. In practical engineering applications, the model achieved prediction accuracies exceeding 0.8, with some cases reaching over 0.9, indicating satisfactory predictive effectiveness. Therefore, the proposed Optuna-optimized GBDT algorithm provides an effective solution for soil layer classification and offers valuable references for further research in this field.

       

    /

    返回文章
    返回