返回
Ensemble learning models with a Bayesian optimization algorithm for mineral prospectivity mapping
DOI:10.1016/j.oregeorev.2022.104916.png)
摘要
En 中文
Machine learning algorithms have been widely applied in mineral prospectivity mapping (MPM). In this study, we implemented ensemble learning of extreme gradient boosting (XGBoost) and random forest (RF) models to create MPM for magmatic hydrothermal tin polymetallic deposits in Xianghualing District, southern Hunan Province, China. Machine-learning models often require careful adjustment of the learning parameters and model hyperparameters for optimal global performance. However, parameter tuning often entails tedious calculations and sufficient expert experience, which is a time-consuming and labor-intensive process. To obtain the global optimal performance of the XGBoost and RF models, a Bayesian optimization algorithm (BOA) was employed with the aid of 5-fold cross validation to search for the most appropriate hyperparameters of the XGBoost and RF models. After the Bayesian optimization, the AUC values of both models were significantly improved, indicating that the BOA is a powerful optimization tool. The optimization results provide a reference for the empirical hyperparameter setting of ensemble learning models. Through a comparative study, the XGBoost model was shown to be superior to the RF model in terms of accuracy, precision, recall, F1 score, and kappa coefficient. In addition, the receiver operating characteristic curves and prediction-area curves showed that the XGBoost model outperformed the RF model, indicating that the XGBoost model had better prediction ability and stability in the case area. In this study, the XGBoost model shows great potential for MPM, offering a significant improvement over the BOA method.
Keyword:
Mineral prospectivity mapping
Ensemble learning
XGBoost
Random forest
Bayesian optimization
K-fold cross validation
AI总结
对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。
期刊
IF:
3.6
论文数:
6.6K
被引数:
2.2W
机构
暂无机构信息
引用论文
Jurassic magmatism related Pb-Zn-W-Mo polymetallic mineralization in the central Nanling Range, South China: Geochronologic, geochemical, and isotopic evidence from the Huangshaping deposit
Ore Geology Reviews
IF3.6
Genesis of the Huangshaping W-Mo-Cu-Pb-Zn polymetallic deposit in Southeastern Hunan Province, China: Constraints from fluid inclusions, trace elements, and isotopes
Ore Geology Reviews
IF3.6
An Ensemble Deep Learning Method for Vehicle Type Classification on Visual Traffic Surveillance Sensors
IEEE ACCESS
IF3.6

