Improved mapping of heavy metals in agricultural soils using machine learning augmented with spatial regionalization indices

The accurate spatial mapping of heavy metal levels in agricultural soils is crucial for environmental management and food security. However, the inherent limitations of traditional interpolation methods and emerging machine-learning techniques restrict their spatial prediction accuracy. This study aimed to refine the spatial prediction of heavy metal distributions in Guangxi, China, by integrating machine learning models and spatial regionalization indices (SRIs). The results demonstrated that random forest (RF) models incorporating SRIs outperformed artificial neural network and support vector regression models, achieving R2 values exceeding 0.96 for eight heavy metals on the test data. Hierarchical clustering for feature selection further improved the model performance. The optimized RF models accurately predicted the heavy metal distributions in agricultural soils across the province, revealing higher levels in the central-western regions and lower levels in the north and south. Notably, the models identified that 25.78% of agricultural soils constitute hotspots with multiple co-occurring heavy metals, and over 6.41 million people are exposed to excessive soil heavy metal levels. Our findings provide valuable insights for the development of targeted strategies for soil pollution control and agricultural soil management to safeguard food security and public health.