Data prediction of soil heavy metal content by deep composite model

Purpose The content of heavy metals in the soil is directly related to the control of soil pollution, but due to the limitations of manpower and material resources, it is difficult to detect them in detail; researchers usually need to predict the content of soil heavy metals in unknown areas based on existing data. Therefore, how to choose an effective method to complete this process has become a challenging problem. Materials and methods In this paper, a deep composite model (DCM) is proposed. The model is based on radial basis function neural network (RBFNN), then, uses self-adaptive learning based particle swarm optimization algorithm (SLPSO) to generate the weight and bias of the output layer of RBFNN and employs adaptive adjustment based root mean square back-propagation (ARMSProp) to optimize all variables of RBFNN, so as to improve the prediction accuracy of the model for soil heavy metal content. When using this model to predict soil heavy metal content, the Pearson coefficient is used as a comparison index to compare the correlation between different heavy metals and heavy metals to be predicted, and finally the content of heavy metals with a Pearson coefficient greater than 0.5 is selected as the input of the model variable. Results and discussion First in the validation of the proposed SLPSO algorithm, the effectiveness of SLPSO and the feasibility of being applied to the DCM model have been proved. Then, the DCM was applied to the prediction of soil heavy metal content in six new urban areas of Wuhan in China, the experimental results show that the predicted value of soil heavy metal content of DCM is closer to the actual value than other comparison models, and the four error indicator values of DCM are also significantly lower than other comparison models, especially when compared with RBFNN, the MAPE and SMAPE of DCM have dropped by 8.6% and 3.9%, respectively. Conclusions We can conclude that the deep composite model proposed in this paper obtains a good prediction accuracy when predicting soil heavy metal content; it has certain feasibility and can be used as an effective method for soil heavy metal content prediction.