Abstract
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.







Similar content being viewed by others
References
Alexandridis A, Chondrodima E, Sarimveis H (2013) Radial basis function network training using a nonsymmetric partition of the input space and particle swarm optimization. IEEE Trans Neural Networks Learn Syst 24(2):219–230
Babu GS, Suresh S (2013) Sequential projection-based metacognitive learning in a radial basis function network for classification problems. IEEE Trans Neural Networks Learn Syst 24(2):194–206
Chander K, Dyckmans J, Joergensen R, Meyer B (2001) Different sources of heavy metals and their long-term effects on soil microbial properties. Biol Fertil Soils 34(4):241–247
Chatterjee A, Siarry P (2011) Nonlinear inertia weight variation for dynamic adaptation in particle swarm optimization. Comput Oper Res 33(3):859–871
Cheng R, Jin Y (2014) A competitive swarm optimizer for large scale optimization. IEEE Trans Cybern 45(2):191–204
Cheng SJ, Miao JM, Wu SJ (2013) Use of metamodeling optimal approach promotes the performance of proton exchange membrane fuel cell (PEMFC). Appl Energy 105:161–169
Chodak M, Niklińska M, Beese F (2007) Near-infrared spectroscopy for analysis of chemical and microbiological properties of forest soil organic horizons in a heavy-metal-polluted area. Biol Fertil Soils 44(1):171–180
Duchi J, Hazan E, Singer Y (2011) Adaptive subgradient methods for online learning and stochastic optimization. J Mach Learn Res 12(7):257–269
Fernández-Navarro F, Hervás-Martínez C, Gutierrez PA (2013) Generalised gaussian radial basis function neural networks. Soft Comput 17(3):519–533
Gálvez A, Iglesias A (2013) A new iterative mutually coupled hybrid GA–PSO approach for curve fitting in manufacturing. Appl Soft Comput 13(3):1491–1504
Hameed AA, Karlik B, Salman MS (2016) Back-propagation algorithm with variable adaptive momentum. Knowl-Based Syst 114:79–87
Han M, Xi J (2004) Efficient clustering of radial basis perceptron neural network for pattern recognition. Pattern Recogn 37(10):2059–2067
Jia X, Hu B, Marchant BP, Zhou L, Shi Z, Zhu Y (2019) A methodological framework for identifying potential sources of soil heavy metal pollution based on machine learning: a case study in the Yangtze Delta, China. Environ Pollut 250:601–609
Kandeler E, Tscherko D, Bruce KD, Stemmer M, Hobbs PJ, Bardgett RD, Amelung W (2000) Structure and function of the soil microbial community in microhabitats of a heavy metal polluted soil. Biol Fertil Soils 32(5):390–400
Kennedy J, Eberhart R (1995) Particle swarm optimization. Proc IEEE Int Conf Neural Networks 4:1942–1948
Kennedy J, Mendes R (2002) Population structure and particle swarm performance. Proc IEEE Congr Evol Comput 2:1671–1676
Kolassa J, Reichle RH, Liu Q, Alemohammad H (2017) Estimating surface soil moisture from SMAP observations using a neural network technique. Remote Sens Environ 204:43–59
Li C (2012) A self-learning particle swarm optimizer for global optimization problems. IEEE Trans on Cybern 42(3):627–646
Li Q, Wang C, Yue T, Li B (2010) Method for spatial variety of soil organic matter based on radial basis function neural network. Trans Chin Soc Agric Eng 26(1):87–93
Liang J, Suganthan P (2006) Dynamic multi-swarm particle swarm optimizer with a novel constraint-handling mechanism. IEEE Trans Evol Comput 10:9–16
Liang JJ, Qin AK, Suganthan PN, Subramanian B (2006) Comprehensive learning particle swarm optimizer for global optimization of multimodal functions. IEEE Trans Evol Comput 10(3):281–295
Oh SK, Kim E, Pedrycz W, Seo K (2014) Fuzzy radial basis function neural networks with information granulation and its parallel genetic optimization. Fuzzy Sets Syst 237:96–117
Ratnaweera A, Halgamuge SK, Watson HC (2004) Self-organizing hierarchical particle swarm optimizer with time-varying acceleration coefficients. IEEE Trans Evol Comput 8(3):240–255
Sedki A, Ouazar D (2012) Hybrid particle swarm optimization and differential evolution for optimal design of water distribution systems. Adv Eng Inform 26(3):582–591
Shen W, Guo X, Wu C, Wu DD (2011) Forecasting stock indices using radial basis function neural networks optimized by artificial fish swarm algorithm. Knowl-Based Syst 24(3):378–385
Shi Y, Obalahnahatti B (1998) A modified particle swarm optimizer. Proc IEEE Congr Evol Comput 6:69–73
Tabari H, Hosseinzadehtalaei P, Willems P (2015) Short-term forecasting of soil temperature using artificial neural network. Meteorol Appl 22(3):576–585
Villa S, Salzo S, Baldassarre L, Verri A (2013) Accelerated and inexact forward-backward algorithms. SIAM J Optim 23(3):1607–1633
Wang H, Sun H, Li C, Rahnamayan S, Pan J (2013) Diversity enhanced particle swarm optimization with neighborhood search. Inf Sci 223:119–135
Wang F, Zhang H, Li K, Lin Z, Yang J, Shen X (2018) A hybrid particle swarm optimization algorithm using adaptive learning strategy. Inf Sci 436:162–177
Wu H, Yang F, Li H, Li Q, Zhang F, Ba Y, Cui L, Sun L, Lv T, Wang N, Zhu J (2019) Heavy metal pollution and health risk assessment of agricultural soil near a smelter in an industrial city in China. Int J Environ Health Res 30(2):1–13
Xie Y, Chen T, Lei M, Yang J, Guo Q, Song B, Zhou X (2011) Spatial distribution of soil heavy metal pollution estimated by different interpolation methods: accuracy and uncertainty analysis. Chemosphere 82(3):468–476
Xiong T, Bao Y, Hu Z, Chiong R (2015) Forecasting interval time series using a fully complex-valued RBF neural network with DPSO and PSO algorithms. Inf Sci 305:77–92
Yu H, Reiner PD, Xie T, Bartczak T, Wilamowski BM (2014) An incremental design of radial basis function networks. IEEE Trans Neural Networks Learn Syst 25(10):1793–1803
Zhan ZH, Zhang J, Li Y, Chung SH (2010) Adaptive particle swarm optimization. IEEE Trans Cybern 39(6):1362–1381
Funding
This work is supported by National Nature Science Foundation of China (Grant No.61272278); Major Technical Innovation Projects of Hubei Province (ID:2018ABA099); Natural Science Foundation of Hubei Province of China (Grant No.2015CFA061); National Science Fund for Youth of Hubei Province of China (Grant No.2018CFB408).
Author information
Authors and Affiliations
Corresponding author
Additional information
Responsible editor: Jun Zhou
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Cao, W., Zhang, C. Data prediction of soil heavy metal content by deep composite model. J Soils Sediments 21, 487–498 (2021). https://doi.org/10.1007/s11368-020-02793-y
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11368-020-02793-y


