Assessing the feasibility of using Machine learning algorithms to determine reservoir water quality based on a reduced set of predictors

cris.virtual.author-orcid0000-0003-0344-585X
cris.virtual.author-orcid0000-0003-3999-7250
cris.virtualsource.author-orcidff7a36ab-d209-401b-822d-12191685f04a
cris.virtualsource.author-orcide32458cb-9a33-41e8-9c1a-a373b123c233
dc.abstract.enThe present study analyses the possibility of assessing water quality using the water quality index (WQI) through the application of four different machine learning algorithms (ML): neural network models (NNM), random forest (RF), k-nearest neighbor (KNN), and linear regression (LR). Water quality was determined based on 5 indicators: P, COD, BOD5, N total, and total suspended solids TS. The possibility of predicting water quality (WQI index) based on the reduced number of predictors was then analyzed. It was estimated which indicators have the most significant impact on WQI values. The performance of models using different algorithms, as well as those trained on full and reduced data sets, was compared. The models demonstrate high performance in WQI prediction, achieving an R2 of 0.999 (for NNM and LR) for the entire dataset, 0.988 (KNN) for the dataset using only three types of predictors, and 0.941 for the dataset using only two predictors (RF). The construction and training of ML models for reduced sets and types of predictors will enable early water quality estimation based on only a few selected parameters. The implementation of ML algorithms will enable more effective water quality management and significantly improve the precision of predictions for critical water parameters.
dc.affiliationWydział Inżynierii Środowiska i Inżynierii Mechanicznej
dc.affiliation.instituteKatedra Inżynierii Wodnej i Sanitarnej
dc.affiliation.instituteKatedra Budownictwa i Geoinżynierii
dc.contributor.authorWalczak, Natalia
dc.contributor.authorWalczak, Zbigniew
dc.date.access2025-06-16
dc.date.accessioned2025-06-16T08:08:01Z
dc.date.available2025-06-16T08:08:01Z
dc.date.copyright2025-01-07
dc.date.issued2025
dc.description.accesstimeat_publication
dc.description.bibliographyil., bibliogr.
dc.description.financepublication_act
dc.description.financecost9981,44
dc.description.if7,0
dc.description.numberJune 2025
dc.description.points200
dc.description.versionfinal_published
dc.description.volume175
dc.identifier.doi10.1016/j.ecolind.2025.113556
dc.identifier.eissn1872-7034
dc.identifier.issn1470-160X
dc.identifier.urihttps://sciencerep.up.poznan.pl/handle/item/2849
dc.identifier.weblinkhttps://www.sciencedirect.com/science/article/pii/S1470160X25004868?via%3Dihub
dc.languageen
dc.pbn.affiliationenvironmental engineering, mining and energy
dc.relation.ispartofEcological Indicators
dc.relation.pagesart. 113556
dc.rightsCC-BY
dc.sciencecloudsend
dc.share.typeOPEN_JOURNAL
dc.subject.enreservoir water quality
dc.subject.enWQI
dc.subject.enmachine learning ML
dc.subject.enPCA
dc.titleAssessing the feasibility of using Machine learning algorithms to determine reservoir water quality based on a reduced set of predictors
dc.typeJournalArticle
dspace.entity.typePublication
oaire.citation.volume175