Prediction of zinc, cadmium, and arsenic in european soils using multi-end machine learning models

Abstract

Heavy metal contamination in soil is a major environmental and public health concern, especially in regions with substantial industrial and agricultural activities. Conventional predictive models often focus on single contaminants, limiting their utility for comprehensive environmental monitoring. This study addressed these limitations by developing an advanced multi-end ensemble convolutional neural network model capable of simultaneously predicting the concentrations of cadmium, arsenic, and zinc in European soils. A comprehensive dataset with 18 diverse factors was prepared, including soil properties, climatic factors, and anthropogenic activities. Moreover, the model compared four ensemble learning techniques in contamination prediction, including simple averaging, snapshot ensembles, integrated stacking, and separate stacking. Among these, the separate stacking model with random forest regressor meta-model achieved the highest accuracy, with a mean spared error of 0.0378, a mean absolute error of 0.0785, and a coefficient of determination of 0.79 in the testing phases. Sensitivity analysis highlighted farming area, road length, nitrogen content, and mean annual temperature as key factors influencing metal concentrations. To enhance accessibility, a GUI-based web application was developed, allowing users to enter relevant factors and receive real-time predictions of contamination levels. This application empowers stakeholders, such as environmental regulators and policymakers, to make informed, data-driven decisions for targeted remediation. These findings underscore the critical role of integrated machine learning approaches in environmental science, offering a powerful tool for identifying contamination hotspots, supporting soil health management, and promoting sustainable land use.