A new machine learning technique for predicting river water quality using AVOA-RNN
Abstract Water quality monitoring plays a critical role in safeguarding human health and environmental sustainability. However, existing machine learning models such as KNN, SVM, and CNN struggle with imbalanced and small-sample datasets, reducing their effectiveness for real-time water quality assessment. To overcome these limitations, this study introduces an innovative hybrid African Vulture Optimization Algorithm–Recurrent Neural Network (AVOA-RNN) framework. The novelty of the approach lies in three aspects: (i) the integration of AVOA with RNN to automatically tune hyper-parameters and select discriminative features, (ii) the incorporation of Synthetic Minority Oversampling Technique (SMOTE) to mitigate class imbalance and enhance minority-class recognition, and (iii) the evaluation on a newly collected Cauvery River water-quality dataset. Experimental results demonstrate that AVOA-RNN achieves 97% classification accuracy, outperforming CNN, LSTM, GA-RNN, and PSO-RNN baselines by 6–15%. These findings highlight the robustness, adaptability, and superior predictive power of the proposed framework for imbalanced water quality datasets.
Authors
- K Kalaivanan.
- Devi Mani
- Karpagalakshmi RC
- Vellingiri J
- Rajkumar Y.
Institutions
- Alliance University (IN)
- SRM University (IN)
- KPR Institute of Engineering and Technology (IN)
- King Khalid University (SA)
- Vellore Institute of Technology University (IN)
Publication Details
- Journal
- Water Science
- Published
- 2026-04-12
- DOI
- https://doi.org/10.1007/s44533-025-00005-5
- Primary Topic
- Hydrological Forecasting Using AI
- Type
- article
- Field-Weighted Citation Impact
- 0.00