Simultaneously mapping the 3D distributions of multiple heavy metals in an industrial site using deep learning and multisource auxiliary data

Abstract

Three-dimensional (3D) distributions of multiple soil pollutants in industrial site are crucial for risk assessment and remediation. Yet, their 3D prediction accuracies are often low because of the strong variability of pollutants and availability of 3D covariate data. This study proposed a patch-based multi-task convolution neural network (MT-CNN) model for simultaneously predicting the 3D distributions of Zn, Pb, Ni, and Cu at an industrial site. By integrating neighborhood patches from multisource covariates, the MT-CNN model captured both horizontal and vertical pollution information, and outperformed the widely-used methods such as random forest (RF), ordinary Kriging (OK), and inverse distance weighting (IDW) for all the 4 heavy metals, with R² values of 0.58, 0.56, 0.29 and 0.23 for Zn, Pb, Ni and Cu, respectively. Besides, the MT-CNN model achieved more stable predictions with reasonable accuracy, in comparison with the single-task CNN model. These results highlighted the potential of the proposed MT-CNN in simultaneously mapping the 3D distributions of multiple pollutants, while balancing the model training, maintaining and accuracy for low-cost rapid assessment of soil pollution at industrial sites.