Evaluation of machine learning models for accurate prediction of heavy metals in coal mining region soils in Bangladesh.

Abstract

Coal mining soils are highly susceptible to heavy metal pollution due to the discharge of mine tailings, overburden dumps, and acid mine drainage. Developing a reliable predictive model for heavy metal concentrations in this region has proven to be a significant challenge. This study employed machine learning (ML) techniques to model heavy metal pollution in soils within this critical ecosystem. A total of 91 standardized soil samples were analyzed to predict the accumulation of eight heavy metals using four distinct ML algorithms. Among them, random forest model outer performed in predicting As (0.79), Cd (0.89), Cr (0.63), Ni (0.56), Cu (0.60), and Zn (0.52), achieving notable R squared values. The feature attribute analysis identified As-K, Pb-K, Cd-S, Zn-Fe₂O₃, Cr- Fe₂O₃, Ni-Al₂O₃, Cu-P, and Mn- Fe₂O₃ relationships resembled with correlation coefficients among them. The developed models revealed that the contamination factor for metals in soils indicated extremely high levels of Pb contamination (CF ≥ 6). In conclusion, this research offers a robust framework for predicting heavy metal pollution in coal mining soils, highlighting critical areas that require immediate conservation efforts. These findings emphasize the necessity for targeted environmental management and mitigation to reduce heavy metal pollution in mining sites.