• 中文核心期刊
  • 中国科技核心期刊
  • RCCSE中国核心学术期刊
  • Scopus, DOAJ, CA, AJ, JST收录期刊
高级检索

基于LightGBM和DNNFL的陷落柱识别方法研究与应用

Research and application of recognition method for collapsed columns based on LightGBM and DNNFL

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

     

    Abstract: To address the issue of low recognition accuracy of collapse pillars in imbalanced seismic attribute datasets, a novel collapse pillar recognition method named "LightGBM-DNNF" was proposed, which combined Light Gradient Boosting Machine (LightGBM) and a Deep Neural Network (DNN) improved using Focal Loss. Firstly, attribute selection was performed through correlation and importance analysis. Then, the paths of LightGBM leaf nodes were extracted as new features, which were combined with the original dataset to form a new dataset. Finally, the new dataset was input into the DNNFL model for binary classification and geological structure prediction. Precision, Recall, F1-score, and AUC were introduced as evaluation metrics, and comparative experiments and ablation experiments were conducted on datasets from three mining areas. Experimental results demonstrate that, compared to traditional machine learning and single ensemble learning algorithms, the F1-score and AUC values of the LightGBM-DNNFL model are both above 93%, indicating its effectiveness in recognizing collapse pillars and exhibiting stronger generalization ability.

     

/

返回文章
返回