Machine learning-driven surface water quality prediction: an intuitive GUI solution for forecasting TDS and DO levels
Year: 2025
Type: article
Abstract: ABSTRACT Industrialization and human activities significantly affect water quality, introducing pollutants into aquatic ecosystems. This study employs machine learning models, specifically two hybrid models (the random forest model and the gene expression model) and standalone models, to predict total dissolved solids (TDS) and dissolved oxygen (DO) levels from 11 inputs. Particle swarm optimization (PSO) was used to enhance model performance, with 423 samples analyzed (80% training, 20% testing). k-fold cross-validation assessed model reliability using metrics, such as R2, RMSE, and MAE. The PSO-RF model outperformed others in TDS prediction (R2 = 0.99, RMSE = 0.0001) and showed positive results for DO (R2 = 0.96, RMSE = 0.32). The PSO-GEP model also performed well (R2 = 0.99 for TDS and 0.95 for DO). Shapley analysis indicated high correlations between total solids and turbidity with TDS, and water temperature with DO. New predictive equations and a graphical user interface (GUI) were developed for practical applications.
Source: Water Quality Research Journal
Authors: Bilal Siddiq, Muhammad Faisal Javed, Ali Aldrees
Language: English
Subfield: Environmental Engineering
Field: Environmental Science
Domain: Physical Sciences
Open Access status: hybrid
OpenAlex ID: W7082248963