Automated machine learning achieves accurate water quality prediction with reduced parameter requirements
Universidade Federal de Juiz de Fora · University of Milan · +2 more institutions
Abstract
Accurate water quality assessment is critical for environmental monitoring and public health. Conventional Water Quality Index (WQI) computation methods, however, often rely on numerous parameters and labor-intensive processes, thus limiting their practicality for rapid assessments. While Machine Learning (ML) offers promising alternatives, the development of high-performing models typically demands extensive expertise and computational resources. This study addresses the latter gap by leveraging Automated Machine Learning (AutoML), specifically the AutoGluon platform, to predict WQI from a reduced set of readily available water quality parameters. Our objectives were to (i) evaluate the predictive performance…
Citation impact
- FWCI
- 46.91
- Percentile
- 100%
- References
- 81
Authors
8Topics & keywords
- Interpretability
- Random forest
- Hyperparameter
- Pipeline (software)
- Predictive modelling
- Water quality
- Ensemble learning
- Decision tree
- Clean water and sanitation
Funding
- CDCoordenação de Aperfeiçoamento de Pessoal de Nível Superior
- CNConselho Nacional de Desenvolvimento Científico e TecnológicoAwards: 2021 AV02 0062/22, APQ-04458-23, 307688/2022-4, BPD-00083-22, 409433/2022-5, 304646/2025-3, 304646/2025- 3, APQ-02513-22
- FDFinanciadora de Estudos e ProjetosAward: 2021 AV02 0062/22
- FDFundação de Amparo à Pesquisa do Estado de Minas GeraisAward: APQ-04458-23
- UFUniversidade Federal de Juiz de Fora