Amir Naghibi
Researcher
Improved TDS forecasting in data-scarce regions using CEEMDAN and AI-driven hydro-climatic analysis
Author
Summary, in English
Total dissolved solids (TDS) are a key water quality parameter, reflecting the concentration of dissolved salts in aquatic systems. Accurate TDS forecasting is essential for sustainable water resource management, particularly in data-scarce regions. This study proposes a novel and generalized AI-based framework to forecast TDS up to six months ahead using a limited set of hydro-climatic input variables. The methodology combines Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN) for signal denoising and pattern extraction with advanced machine learning models, including Random Forest (RF) and a hybrid Grey Wolf Optimization–Support Vector Machine (GWO-SVM). To enhance model transferability, only four widely available input variables—precipitation, evaporation, discharge, and chloride concentration—were used. Historical data from 1975 to 2016 were collected from three hydrometric stations representing distinct climatic conditions. Forecasting was conducted both with and without the inclusion of lagged TDS values. The CEEMDAN-GWO-Linear SVM model achieved high accuracy (R2 = 0.70–0.96) across different forecast horizons. Additionally, CEEMDAN significantly improved the predictive performance of both SVM and RF models. Feature importance analysis using RF ranked chloride concentration, discharge, precipitation, and evaporation as the most influential variables in TDS prediction. The proposed framework offers a robust, data-efficient solution for mid-term water quality forecasting.
Department/s
- Division of Water Resources Engineering
- Centre for Advanced Middle Eastern Studies (CMES)
- MECW: The Middle East in the Contemporary World
- LTH Profile Area: Water
Publishing year
2025-08
Language
English
Publication/Series
Environmental Modelling and Software
Volume
192
Document type
Journal article
Publisher
Elsevier
Topic
- Earth Observation
Keywords
- CEEMDAN
- Random forests
- Support vector machine
- Time series modeling
- Water quality
Status
Published
ISBN/ISSN/Other
- ISSN: 1364-8152