Prediction of soil heavy metal contents in urban residential areas and the strength of deep learning: A case study of Beijing

Abstract

Predicting soil heavy metal (SHM) content is crucial for understanding SHM pollution levels in urban residential areas and guide efforts to reduce pollution. However, current research indicates low SHM prediction accuracy in urban areas. Therefore, we employed a deep learning method (fully connected deep neural network) alongside four other methods (muti-layer perceptron, radial basis function neural network, multiple stepwise linear regression, and Kriging interpolation) to predict SHM content in the urban residential areas of Beijing and demonstrated the strength of deep learning in improving prediction accuracy. We found the contents of the evaluated heavy metals (Cd, Cu, Pb, and Zn) exhibited significant correlations with numerous other soil physicochemical properties and environmental factors. The prediction accuracy for Cu, Pb, and Zn contents was relatively high across different methods. Notably, deep learning showed considerable strength in predicting the contents of the four heavy metals, with the R2 for the test set of the model ranging from 0.75 to 0.91. Compared to other methods, deep learning achieved markedly higher prediction accuracy according to different accuracy evaluation indicators (e.g., deep learning showed increases in the cumulative R2 of the four heavy metals ranging from 53.16 % to 187.36 % compared to other methods). Our study indicates that deep learning can significantly improve SHM content prediction accuracy in urban areas and is highly applicable in urban residential areas with complex environmental influences.