Prediction of heavy metal removal performance of sulfate-reducing bacteria using machine learning

Abstract

A robust modeling approach for predicting heavy metal removal by sulfate-reducing bacteria (SRB) is currently missing. In this study, four machine learning models were constructed and compared to predict the removal of Cd, Cu, Pb, and Zn as individual ions by SRB. The CatBoost model exhibited the best predictive performance across the four subsets, achieving R2 values of 0.83, 0.91, 0.92, and 0.83 for the Cd, Cu, Pb, and Zn models, respectively. Feature analysis revealed that temperature, pH, sulfate concentration, and C/S (the mass ratio of chemical oxygen demand to sulfate) had significant impacts on the outcomes. These features exhibited the most effective metal removal at 35 °C and sulfate concentrations of 1000–1200 mg/L, with variations observed in pH and C/S ratios. This study introduced a new modeling approach for predicting the treatment of metal-containing wastewater by SRB, offering guidance for optimizing operational parameters in the biological sulfidogenic process.