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

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

Research and application of recognition method for collapse column based on LightGBM and DNNFL

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

     

    Abstract: In order to solve the problem of low recognition accuracy of collapse column in unbalanced seismic attribute datasets, a novel collapse column recognition method named "LightGBM-DNNF" was proposed, which based on the light gradient boosting machine (LightGBM) and the use of focal loss to improve deep neural networks (DNN). 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 classification training and prediction of geological structure types. Precision (P), Recall (R), F1-score (F1-score), and area under the curve (AUC) were introduced as evaluation metrics, and comparative experiments and ablation experiments were conducted based on datasets from three mining areas. The experimental results show that compared with the traditional machine learning and single ensemble learning algorithm, the F1-score and AUC values of the LightBM-DNNFL model are both above 93%, indicating its effectiveness in recognizing collapse column and exhibiting stronger generalization ability.

     

/

返回文章
返回