Heavy metals prediction in coastal marine sediments using hybridized machine learning models with metaheuristic optimization algorithm

Zaher Mundher Yaseen*, Wan Hanna Melini Wan Mohtar, Raad Z. Homod, Omer A. Alawi, Sani I. Abba, Atheer Y. Oudah, Hussein Togun, Leonardo Goliatt, Syed Shabi Ul Hassan Kazmi, Hai Tao

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

1 Scopus citations

Abstract

This study proposes different standalone models viz: Elman neural network (ENN), Boosted Tree algorithm (BTA), and f relevance vector machine (RVM) for modeling arsenic (As (mg/kg)) and zinc (Zn (mg/kg)) in marine sediments owing to anthropogenic activities. A heuristic algorithm based on the potential of RVM and a flower pollination algorithm (RVM-FPA) was developed to improve the prediction performance. Several evaluation indicators and graphical methods coupled with visualized cumulative probability function (CDF) were used to evaluate the accuracy of the models. Akaike (AIC) and Schwarz (SCI) information criteria based on Dickey-Fuller (ADF) and Philip Perron (PP) tests were introduced to check the reliability and stationarity of the data. The prediction performance in the verification phase indicated that RVM-M2 (PBAIS = -o.0465, MAE = 0.0335) and ENN-M2 (PBAIS = 0.0043, MAE = 0.0322) emerged as the best model for As (mg/kg) and Zn (mg/kg), respectively. In contrast with the standalone approaches, the simulated hybrid RVM-FPA proved merit and the most reliable, with a 5 % and 18 % predictive increase for As (mg/kg) and Zn (mg/kg), respectively. The study's findings validated the potential for estimating complex HMs through intelligent data-driven models and heuristic optimization. The study also generated valuable insights that can inform the decision-makers and stockholders for environmental management strategies.

Original languageEnglish
Article number141329
JournalChemosphere
Volume352
DOIs
StatePublished - Mar 2024

Bibliographical note

Publisher Copyright:
© 2024 Elsevier Ltd

Keywords

  • Artificial intelligence
  • Heavy metals
  • Sensitivity analysis
  • Soil contamination

ASJC Scopus subject areas

  • Environmental Engineering
  • Environmental Chemistry
  • General Chemistry
  • Pollution
  • Public Health, Environmental and Occupational Health
  • Health, Toxicology and Mutagenesis

Fingerprint

Dive into the research topics of 'Heavy metals prediction in coastal marine sediments using hybridized machine learning models with metaheuristic optimization algorithm'. Together they form a unique fingerprint.

Cite this