IEEE Access (Jan 2020)

A Collaborative Compound Neural Network Model for Soil Heavy Metal Content Prediction

  • Wenqi Cao,
  • Cong Zhang

DOI
https://doi.org/10.1109/ACCESS.2020.3009248
Journal volume & issue
Vol. 8
pp. 129497 – 129509

Abstract

Read online

The prediction of soil heavy metal content is an important part of the management of soil heavy metal pollution, but it is often ignored. At present, there are few studies on the prediction of soil heavy metal content, and it is an urgent problem to choose an efficient method for soil heavy metal content prediction. In this paper, a collaborative compound neural network model (CCNN) was put forward to predict the soil heavy metal content, this model uses wavelet neural network (WNN) as the basic prediction model, and at the same time proposes a parallel bird swarm algorithm (PBSA) to solve the parameter optimization problem of WNN, based on the bird swarm algorithm (BSA), the PBSA not only increases the gathering behavior of individual, but also adopts sine transformation based on fitness difference ratio to carry out the following behavior of beggars to improve the global optimization ability, besides that, the acceptance criterion is used to compare the fitness of individuals after updating to avoid falling into a local optimum. Soil heavy metal content data from Yinchuan city of Ningxia and six new urban areas in Wuhan, China are used to make prediction experiments respectively, through compare with support vector machine (SVM), radial basis function neural network (RBFNN), WNN and bird swarm algorithm optimizes wavelet neural network (BSA-WNN), the experimental results demonstrate that the predicted value of the CCNN is closer to the actual value and has better prediction performance.

Keywords