Open Access iconOpen Access



A New Random Forest Applied to Heavy Metal Risk Assessment

Ziyan Yu1, Cong Zhang1,*, Naixue Xiong2, Fang Chen1

1 Wuhan Polytechnic University, Department of Mathematics and Computer Science, Wuhan, 430023, China
2 Northeastern State University, Department of Mathematics and Computer Science, Tahlequah, OK, 74464, USA

* Corresponding Author: Cong Zhang. Email: email

Computer Systems Science and Engineering 2022, 40(1), 207-221.


As soil heavy metal pollution is increasing year by year, the risk assessment of soil heavy metal pollution is gradually gaining attention. Soil heavy metal datasets are usually imbalanced datasets in which most of the samples are safe samples that are not contaminated with heavy metals. Random Forest (RF) has strong generalization ability and is not easy to overfit. In this paper, we improve the Bagging algorithm and simple voting method of RF. A W-RF algorithm based on adaptive Bagging and weighted voting is proposed to improve the classification performance of RF on imbalanced datasets. Adaptive Bagging enables trees in RF to learn information from the positive samples, and weighted voting method enables trees with superior performance to have higher voting weights. Experiments were conducted using G-mean, recall and F1-score to set weights, and the results obtained were better than RF. Risk assessment experiments were conducted using W-RF on the heavy metal dataset from agricultural fields around Wuhan. The experimental results show that the RW-RF algorithm, which use recall to calculate the classifier weights, has the best classification performance. At the end of this paper, we optimized the hyperparameters of the RW-RF algorithm by a Bayesian optimization algorithm. We use G-mean as the objective function to obtain the optimal hyperparameter combination within the number of iterations.


Cite This Article

APA Style
Yu, Z., Zhang, C., Xiong, N., Chen, F. (2022). A new random forest applied to heavy metal risk assessment. Computer Systems Science and Engineering, 40(1), 207-221.
Vancouver Style
Yu Z, Zhang C, Xiong N, Chen F. A new random forest applied to heavy metal risk assessment. Comput Syst Sci Eng. 2022;40(1):207-221
IEEE Style
Z. Yu, C. Zhang, N. Xiong, and F. Chen "A New Random Forest Applied to Heavy Metal Risk Assessment," Comput. Syst. Sci. Eng., vol. 40, no. 1, pp. 207-221. 2022.


cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 2052


  • 3440


  • 1


Share Link