Prediction of Wastewater Effluent Water Quality Based on Variational Mode Decomposition and Deep Learning Algorithm

MEI Dan, ZHANG Heng

Journal of Changjiang River Scientific Research Institute ›› 2025, Vol. 42 ›› Issue (9) : 67-74.

PDF(6187 KB)
PDF(6187 KB)
Journal of Changjiang River Scientific Research Institute ›› 2025, Vol. 42 ›› Issue (9) : 67-74. DOI: 10.11988/ckyyb.20240721
Water Environment And Water Ecology

Prediction of Wastewater Effluent Water Quality Based on Variational Mode Decomposition and Deep Learning Algorithm

Author information +
History +

Abstract

[Objective] This study aims to propose a novel method for predicting effluent water quality in wastewater treatment plants, in order to enhance prediction accuracy and address the inadequate generalizability of existing models, thereby providing robust support for the operational optimization of wastewater treatment plants. [Methods] The proposed prediction framework primarily includes the following steps: First, the water quality sequence was decomposed into multiple subsequences with different characteristics using the variational mode decomposition (VMD) method. Subsequently, a comprehensive evaluation indicator (CEI) was introduced, based on which the deep learning algorithm with optimal prediction performance was selected for each decomposed subsequence. Four deep learning algorithms were involved in this study. Finally, the predicted values from each sub-model were aggregated to obtain the final effluent quality prediction. Taking the effluent chemical oxygen demand (COD) concentration of a wastewater treatment plant in Wuhan, Hubei Province as the research object, the proposed prediction framework was validated through a case study. The performance of the proposed framework was evaluated by comparing the prediction performance with that of single models. [Results] The effluent COD concentration data from a wastewater treatment plant in Wuhan were used for validation. The results showed that by decomposing the COD time series into different intrinsic mode functions (IMFs) using VMD, the complexity of the COD time series was effectively reduced. This provided simplified components for subsequent prediction, enabling the prediction model to better capture underlying patterns in the data and consequently improve prediction performance. Meanwhile, by introducing the CEI, four key evaluation indicators—mean absolute error (MAE), root mean square error (RMSE), standard deviation (STD), and mean absolute percentage error (MAPE)—were successfully integrated. This allowed for a comprehensive consideration of multi-dimensional error conditions when selecting the optimal prediction algorithm for each IMF subsequence, ensuring the comprehensiveness and accuracy of the selected algorithm. Finally, predictions were made for each different IMF based on the selected algorithm with optimal prediction performance. The results showed that this method effectively improved the overall model’s prediction accuracy, with the RMSE reaching 0.485. This confirmed that the proposed prediction framework achieved significant improvement in prediction performance compared to single models, providing strong support for accurate effluent water quality prediction in wastewater treatment plants. [Conclusions] The proposed water quality prediction framework based on VMD and multiple deep learning algorithms achieves high-precision prediction of effluent COD concentration in wastewater treatment plants by reasonably decomposing the water quality sequence and adaptively selecting prediction algorithms. The framework overcomes the limitations of existing single prediction models in handling complex nonlinear relationships, providing more accurate water quality predictions to support energy-saving and consumption-reduction decision-making in wastewater treatment plants. With significant practical value, it can be further extended in the future to predict other water quality indicators and be applied to wastewater treatment plants of different scales and types, thereby promoting intelligent operation and management in the wastewater treatment industry.

Key words

water quality prediction / variational mode decomposition(VMD) / comprehensive evaluation indicator(CEI) / optimal sub-model selection / deep learning algorithms

Cite this article

Download Citations
MEI Dan , ZHANG Heng. Prediction of Wastewater Effluent Water Quality Based on Variational Mode Decomposition and Deep Learning Algorithm[J]. Journal of Changjiang River Scientific Research Institute. 2025, 42(9): 67-74 https://doi.org/10.11988/ckyyb.20240721

References

[1]
TAYYAB M, JEMAI J, LIM H, et al. A Sustainable Development Framework for a Cleaner Multi-item Multi-stage Textile Production System with a Process Improvement Initiative[J]. Journal of Cleaner Production, 2020, 246: 119055.
[2]
HASSEN E B, ASMARE A M. Predictive Performance Modeling of Habesha Brewery Wastewater Treatment Plant Using Artificial Neural Networks[J]. Chemistry International, 2019, 5(1):87-96.
[3]
LY Q V, TRUONG V H, JI B, et al. Exploring Potential Machine Learning Application Based on Big Data for Prediction of Wastewater Quality from Different Full-scale Wastewater Treatment Plants[J]. Science of The Total Environment, 2022, 832: 154930.
[4]
WANG R, YU Y, CHEN Y, et al. Model Construction and Application for Effluent Prediction in Wastewater Treatment Plant: Data Processing Method Optimization and Process Parameters Integration[J]. Journal of Environmental Management, 2022, 302: 114020.
[5]
NI Q, CAO X, TAN C, et al. An Improved Graph Convolutional Network with Feature and Temporal Attention for Multivariate Water Quality Prediction[J]. Environmental Science and Pollution Research, 2023, 30(5):11516-11529.
[6]
CHING P M L, ZOU X, WU D, et al. Development of a Wide-range Soft Sensor for Predicting Wastewater BOD5 Using an eXtreme Gradient Boosting (XGBoost) Machine[J]. Environmental Research, 2022, 210:112953.
[7]
GENG J, YANG C, LI Y, et al. MPA-RNN: A Novel Attention-based Recurrent Neural Networks for Total Nitrogen Prediction[J]. IEEE Transactions on Industrial Informatics, 2022, 18(10): 6516-6525.
[8]
刘杰, 金勇杰, 田明. 基于VMD和TCN的多尺度短期电力负荷预测[J]. 电子科技大学学报, 2022, 51(4):550-557.
(LIU Jie, JIN Yong-jie, TIAN Ming. Multi-scale Short-term Load Forecasting Based on VMD and TCN[J]. Journal of University of Electronic Science and Technology of China, 2022, 51(4):550-557. (in Chinese))
[9]
许玉格, 曹涛, 罗飞. 基于相关向量机的污水处理出水水质预测模型[J]. 华南理工大学学报(自然科学版), 2014, 42(5):103-108.
Abstract
针对污水生化处理过程复杂、重要出水指标预测困难且误差比较大的情况,提出了一种基于相关向量机的污水处理出水水质预测模型.首先利用模糊单调递增依赖算法对输入数据进行属性约简,并结合经验确定输入属性,然后利用相关向量机建立预测模型,对模型参数进行寻优,以实现最优预测.实验结果表明,文中提出的预测模型预测精度高、泛化能力强,能较好地满足污水处理出水水质的预测要求.
(XU Yu-ge, CAO Tao, LUO Fei. Wastewater Effluent Quality Prediction Model Based on Relevance Vector Machine[J]. Journal of South China University of Technology (Natural Science), 2014, 42(5): 103-108. (in Chinese))
[10]
金岩磊, 何茂慧, 郭涛, 等. 改进VMD融合深度学习在滚动轴承故障诊断中的应用[J]. 热能动力工程, 2023, 38(2):144-152.
(JIN Yan-lei, HE Mao-hui, GUO Tao, et al. Application of Improved VMD Combined with Deep Learning in Rolling Bearing Fault Diagnosis[J]. Journal of Engineering for Thermal Energy and Power, 2023, 38(2):144-152. (in Chinese))
[11]
李涛, 杨腾宇, 刘波, 等. 基于VMD-GRU的地铁隧道台阶法施工地表沉降预测[J]. 华中科技大学学报(自然科学版), 2023, 51(7): 48-54, 62.
(LI Tao, YANG Teng-yu, LIU Bo, et al. Surface Settlement Prediction of Subway Tunnels Constructed by Step Method Based on VMD-GRU[J]. Journal of Huazhong University of Science and Technology (Natural Science), 2023, 51(7): 48-54, 62. (in Chinese))
[12]
ARAGÃO D P, OLIVEIRA E V, BEZERRA A A, et al. Multivariate Data Driven Prediction of COVID-19 Dynamics: Towards New Results with Temperature, Humidity and Air Quality Data[J]. Environmental Research, 2022, 204: 112348.
[13]
LIU W, LIU T, LIU Z, et al. A Novel Deep Learning Ensemble Model Based on Two-stage Feature Selection and Intelligent Optimization for Water Quality Prediction[J]. Environmental Research, 2023, 224:115560.
[14]
MASMOUDI S, ELGHAZEL H, TAIEB D, et al. A Machine-learning Framework for Predicting Multiple Air Pollutants’ Concentrations via Multi-target Regression and Feature Selection[J]. Science of The Total Environment, 2020, 715: 136991.
[15]
刘亚珲, 赵倩. 基于聚类经验模态分解的CNN-LSTM超短期电力负荷预测[J]. 电网技术, 2021, 45(11):4444-4451.
(LIU Ya-hui, ZHAO Qian. Ultra-short-term Power Load Forecasting Based on Cluster Empirical Mode Decomposition of CNN-LSTM[J]. Power System Technology, 2021, 45(11):4444-4451. (in Chinese))
[16]
黄睿, 朱玲俐, 高峰, 等. 基于变分模态分解的卷积长短时记忆网络短期电力负荷预测方法[J]. 现代电力, 2024, 41(1):97-105.
(HUANG Rui, ZHU Ling-li, GAO Feng, et al. Short-term Power Load Forecasting Method Based on Variational Modal Decomposition for Convolutional Long-short-term Memory Network[J]. Modern Electric Power, 2024, 41(1): 97-105. (in Chinese))
[17]
NIU D, SUN L, YU M, et al. Point and Interval Forecasting of Ultra-short-term Wind Power Based on a Data-driven Method and Hybrid Deep Learning Model[J]. Energy, 2022, 254: 124384.
[18]
FAN G, YU P, WANG Q, et al. Short-term Motion Prediction of a Semi-submersible by Combining LSTM Neural Network and Different Signal Decomposition Methods[J]. Ocean Engineering, 2023, 267: 113266.
[19]
WEI M, DENG A, CHEN H, et al. Classification of Wellhead Device Defects Based on CNN-GA-BP Model[J]. Journal of Magnetism and Magnetic Materials, 2023, 572:170606.
PDF(6187 KB)

Accesses

Citation

Detail

Sections
Recommended

/