我们的服务符合学术规范和道德
专业 高端让您使用时没有后顾之忧

矿业安全与环保2024年第05期:基于LightGBM和DNNFL的陷落柱识别方法研究与应用

来源:期刊VIP网 时间:


作者:王怀秀;王慧;

单位:北京建筑大学电气与信息工程学院;

摘要:为了解决不平衡地震属性数据集中陷落柱识别准确率较低的问题,提出了一种基于轻梯度提升机(Light Gradient Boosting Machine, LightGBM)和利用焦点损失(Focal Loss)改进深度神经网络(Deep Neural Networks, DNN)相结合的陷落柱识别方法LightGBM-DNNF。首先通过相关性分析和重要性分析进行属性优选;其次提取LightGBM叶子节点的路径作为新的特征,并与原始数据集组合成新的数据集;最后输入到DNNFL模型中进行分类训练,预测地质构造类型。引入精确率(P)、召回率(R)、F1分数(F_(1-score))、曲线下面积(A_(UC))作为评价指标,基于3个矿区的数据集开展对比实验和消融实验。实验结果表明,与传统的机器学习和单一的集成学习算法相比,LightGBM-DNNFL模型的F_(1-score)和A_(UC)值均在93%以上,能有效识别陷落柱,且模型泛化能力更强。

关键词:陷落柱识别;;深度神经网络;;Focal Loss;;LightGBM;;不平衡分类

基金资助:国家重点研发计划项目(2018YFC0807806)