Accurate prediction of spatial distribution of soil potentially toxic elements using machine learning and associated key influencing factors identification: A case study in mining and smelting area in southwestern China

Accurate prediction of spatial distribution of potentially toxic elements (PTEs) is crucial for soil pollution prevention and risk control. Achieving accurate prediction of spatial distribution of soil PTEs at a large scale using conventional methods presents significant challenges. In this study, machine learning (ML) models, specially artificial neural network (ANN), random forest (RF), and extreme gradient boosting (XGB), were used to predict spatial distribution of soil PTEs and identify associated key factors in mining and smelting area located in Yunnan Province, China, under the three scenarios: (1) natural + socioeconomic + spatial datasets (NS), (2) NS + irrigation pollution index (IPI) datasets, (3) NS + IPI + deposition (DEPO) datasets. The results highlighted the combination of NS+IPI+DEPO yielded the highest predictive accuracy across ML models. Particularly, XGB exhibited the highest performance for As (R