Work

A Hybrid Deep Learning Model for Water Quality Prediction

Year: 2025

Type: article

Abstract: Water quality prediction is crucial for water environment management, playing a key role in preventing pollution and ensuring water resource safety. To improve prediction accuracy, this paper proposes a CEEMDAN-CNN-LSTM model enhanced by a self-attention mechanism. First, the Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN) algorithm decomposes the water quality time series into multiple Intrinsic Mode Functions (IMFs), reducing noise and addressing nonlinearities in the data. Next, Convolutional Neural Networks (CNN) extract spatial features from the decomposed IMFs, improving the model’s spatial learning capability. Then, a Long Short-Term Memory (LSTM) network models the temporal features, capturing long-term dependencies in the water quality data. Finally, a self-attention mechanism assigns varying importance to features at different time steps, improving the model’s focus on key information. Experimental results demonstrate that the proposed model outperforms traditional methods, achieving high accuracy across multiple water quality indicators. This model effectively addresses non-stationarity and noise in the data, demonstrating strong generalization and robustness, providing a novel and efficient approach to water quality prediction.

Source: IEEE Access

DOI: 10.1109/access.2025.3580741

Authors: Qiliang Zhu, Xueting Yu, Liang Zhao, Linjun Xu

Institution: North China University of Water Resources and Electric Power

Language: English

Topic: Hydrological Forecasting Using AI

Subfield: Environmental Engineering

Field: Environmental Science

Domain: Physical Sciences

SDG: Clean water and sanitation

Open Access status: gold

Estimated APC paid: $2,075