Deep learning-based water quality index classification using stacked ensemble variational mode decomposition

Abstract

Water is crucial to human survival in general, and determining the WQI (water quality index) is one of the primary aspects. The existing water quality classification models are facing various challenges and gaps that are impeding their effectiveness. These challenges include limited data availability, the intricate nature of water systems, spatial and temporal variability, non-linear relationships, sensor noise, and error, interpretability, and explainability. It is imperative to address these challenges to improve the accuracy and efficacy of the models and to ensure that they continue to serve as reliable tools for monitoring and safeguarding water quality. To solve the issues, this paper proposes a Stacked Ensemble efficient long short-term memory (StackEL) model for an efficient water quality index classification. At first, the raw input data is pre-processed to rescale the input data using data normalization and one-hot encoding. After that, the process known as variational mode decomposition (VMD) is applied to get at the intrinsic mode functions (IMFs). Consequently, feature selection is performed using an extended coati optimization (EX-CoA) algorithm to select the most significant attributes from the feature selection. Here, publicly available datasets, namely the water quality dataset from Kaggle, are used for classification and performed using are used to perform the Stacked Ensemble efficient long short-term memory (StackEL) classification process effectively. To further perfect the proposed prediction model, the Dwarf Mongoose optimization (DMO) method is implemented. Several measures of effectiveness are examined. When compared to other existing models, the suggested model can achieve a high accuracy of 98.85% of the water quality dataset.