Machine Learning for Water Quality Assessment Based on Macrophyte Presence

Abstract

The ecological state of the Danube River, as the world’s most international river basin, will always be the focus of scientists in the field of ecology and environmental engineering. The concentration of orthophosphate anions in the river is one of the main indicators of the ecological state, i.e., water quality and level of eutrophication. The sedentary nature and ability to survive in river sections, combined with the presence of high levels of orthophosphate anions, make macrophytes an appropriate biological parameter for in situ prediction of in-river monitoring processes. However, a preliminary literature review identified a lack of comprehensive analysis that can enable the prediction of the ecological state of rivers using biological parameters as the input to machine learning (ML) techniques. This work focuses on comparing eight state-of-the-art ML classification models developed for this task. The data were collected at 68 sampling sites on both river sides. The predictive models use macrophyte presence scores as input variables, and classes of the ecological state of the Danube River based on orthophosphate anions, converted into a binary scale, as outputs. The results of the predictive model comparisons show that support vector machines and tree-based models provided the best prediction capabilities. They are also a low-cost and sustainable solution to assess the ecological state of the rivers.

WaQuPs: A ROS-Integrated Ensemble Learning Model for Precise Water Quality Prediction

“…In 2022, Krtolica et al [15] conducted a study concerning a crucial aspect of water quality assessment through the presence of macrophytes. Their research was grounded in existing literature pertaining to water quality assessment, highlighting the significance of considering macrophytes as valuable indicators of water quality.…”

Prediction of hydrological and water quality data based on granular-ball rough set and k-nearest neighbor analysis

“…Among them, k-nearest neighbor (kNN) is one of the most basic and simple machine learning algorithms [53,54]. It is also used for interannual remote sensing monitoring of eelgrass beds [55], algae toxicity identification [56] and water quality assessment [57,58].…”