Optimized AI-MPM: Application of PSO for tuning the hyperparameters of SVM and RF algorithms

• Published in: Computers & geosciences 2025-01, Vol.195, p.105785, Article 105785
• Main Authors: Daviran, Mehrdad, Maghsoudi, Abbas, Ghezelbash, Reza
• Format: Article
• Language: English
• Subjects: AI-MPM; Machine learning algorithms; Particle swarm optimization; Random forest; Support vector machines
• ISSN: 0098-3004
• DOI: 10.1016/j.cageo.2024.105785

Modern computational techniques, particularly Support Vector Machines (SVM) and Random Forest (RF) models, are revolutionizing predictive mineral prospectivity mapping. These advanced systems excel at identifying prime resource locations but require meticulous fine-tuning of their internal settings to achieve peak performance. Careful calibration of these configurations during the learning phase significantly enhances their ability to detect promising deposits. The main goal of this study is to introduce a hybrid model called PSO -SVM and PSO-RF, which aim to combine particle swarm optimization (PSO) with SVM (with RBF kernel) and RF models. This hybrid model automatically adjusts the optimized hyperparameters of SVM and RF, resulting in highly accurate predictions and a wide range of applicability. The PSO algorithm has been applied to fine-tune two main parameters (C and λ) for SVM-RBF and three main parameters (NT, NS, and d) for RF, creating efficient models for both. The proposed hybrid model as well as the traditional versions of SVM and RF models, were tested using a geo-spatial dataset related to Cu mineralization in Kerman belt, SE Iran. Forecasting algorithms were developed by integrating diverse datasets: multi-element concentrations from stream samples, bedrock and fault line maps, indicators of hot fluid interaction, aeromagnetic survey results, coordinates of previously identified copper-rich igneous intrusions, and verified ore body positions as reference points. The models' performance was evaluated using four validation methods: Multi-round data partitioning (K-fold), error classification tables (confusion matrix), true-positive vs. false-positive graphical analysis ((ROC) curve), and P-A plot were used to assess algorithms and models performance. Tests revealed that the PSO-SVM surpassed all competitors. Impressively, this fine-tuned classifier identified prime target zones in merely one-seventh of the region (14%), yet these areas encompassed nearly all verified resource sites (97%). •Advanced ML Models: SVM and RF are changing mineral prospectivity mapping by effectively identifying prime resource locations.•Hybrid Optimization: The study presents PSO-SVM and PSO-RF models, using particle swarm to fine-tune hyperparameters for accuracy.•Diverse Data Integration: Models were tested with geo-spatial datasets from Iran's Kerman belt, including samples, maps, and surveys.•Rigorous Validation: Evaluating performance through K-fold cross-vali