Optimization of Feature Selection in Mineral Prospectivity Using Ensemble Learning
Hong Zhang, Miao Xie, Shiyao Dan, Meilin Li, Yunhe Li, Die Yang, Yuanxi Wang
Abstract
In recent years, machine learning (ML) has been extensively used for the quantitative prediction of mineral resources. However, the accuracy of prediction models is often influenced by data quality, feature selection, and algorithm limitations. This research investigates the benefits of data-driven feature optimization techniques in enhancing model accuracy. Using the Lhasa region in Tibet as the study area, this research applies ensemble learning methods, such as random forest and gradient boosting tree techniques, to optimize 43 feature variables encompassing geology, geochemistry, and geophysics. The optimized feature variables are then input into a support vector machine (SVM) model to generate a prospectivity map. The performance characteristics of the SVM, RF_SVM, and GBDT_SVM models are evaluated using ROC curves. The results indicate that the feature-optimized GBDT_SVM model achieves superior classification accuracy and prediction effectiveness, demonstrating that feature optimization is a necessary step for mineral prospectivity mapping, as it can significantly improve the performance of mineral prospectivity prediction.