Journal of South China University of Technology(Natural Science) >
A Lightweight Multivariate Time Series Prediction Method for Wastewater Treatment
Received date: 2025-04-30
Online published: 2025-06-13
Supported by
the National Natural Science Foundation of China(92467106);the Guangdong Basic and Applied Basic Research Foundation(2021B1515420003);the General College Innovation Team Project of Guangdong Province(2023KCXTDO72)
In wastewater treatment processes, the efficient modeling of key water quality parameters is crucial for achieving process control optimization, anomaly detection, and decision support. However, the process data generally exhibits characteristics such as temporal dependence, multivariable coupling, and non-stationarity under varying operating conditions, posing significant challenges to accurate modeling. To address these issues, this paper proposes a Lightweight Multivariate Time Series Prediction Method for Wastewater Treatment based on the Stationary Wavelet Transform (SWT) and Collaborative Attention (CA) mechanism. This model first performed multi-scale decomposition on wastewater data and used the stationary wavelet transform to extract data features from sequences at different scales. Subsequently, a collaborative attention mechanism based on geometric attention and sparse attention was constructed to effectively capture the complex coupling relationships and temporal features among key water quality parameters. Finally, the features reconstructed via inverse wavelet transform were mapped to the final prediction results through a dual-prediction layer. The model was trained and validated on a measured dataset from a wastewater treatment plant in Dongguan, with multi-step prediction tasks and partial data visualization analyses conducted. Experimental results show that, in the 24-step multi-output prediction tasks, the proposed model achieves a reduction of 9.15% to 37.70% in multi-output root mean square deviation (RMSSD) compared to benchmark models. In other prediction tasks, its accuracy ranks second only to TimesNet, which has a significantly larger parameter scale. These results demonstrate an effective balance between lightweight design and high accuracy, thereby validating the efficacy of the proposed model for time-series prediction in wastewater treatment.
TANG Lili , LIU Yiqi . A Lightweight Multivariate Time Series Prediction Method for Wastewater Treatment[J]. Journal of South China University of Technology(Natural Science), 2026 , 54(1) : 60 -69 . DOI: 10.12141/j.issn.1000-565X.250132
| [1] | 吴菁 .污水处理非稳态特性下核建模方法关键问题的研究[D].广州:华南理工大学,2020. |
| [2] | 方港,袁珑华,王晓明,等 .基于集合卡尔曼-Elman网络的软测量建模方法[J].华南理工大学学报(自然科学版),2023,51(8):126-136. |
| FANG Gang, YUAN Longhua, WANG Xiaoming,et al .Research on ensemble Kalman filter-Elman neural network based soft-sensor model[J].Journal of South China University of Technology(Natural Science Edition),2023,51(8):126-136. | |
| [3] | SHYU H Y, CASTRO C J, BAIR R A,et al .Development of a soft sensor using machine learning algorithms for predicting the water quality of an onsite wastewater treatment system[J].ACS Environmental Au,2023,3(5):308-318. |
| [4] | HERNáNDEZ-DEL-OLMO F, GAUDIOSO E, DURO N,et al .Machine learning weather soft-sensor for advanced control of wastewater treatment plants[J].Sensors,2019,19(14):3139/1-12. |
| [5] | LIU Z, WAN J, MA Y,et al .Online prediction of effluent COD in the anaerobic wastewater treatment system based on PCA-LSSVM algorithm[J].Environmental Science and Pollution Research,2019,26(13):12828-12841. |
| [6] | BAKI O T, ARAS E .Estimation of BOD in wastewater treatment plant by using different ANN algorithms[J].Membrane and Water Treatment,2018,9(6):455-462. |
| [7] | WU J, CHENG H, LIU Y,et al .Modeling of adaptive multi-output soft-sensors with applications in wastewater treatments[J].IEEE Access,2019,7:161887-161898. |
| [8] | 李东,黄道平,许翀,等 .基于协同训练的集成自适应GPR-RVM多输出模型研究[J].华南理工大学学报(自然科学版),2021,49(6):100-108. |
| LI Dong, HUANG Daoping, XU Chong,et al .On integrated adaptive GPR-RVM multi-output model based on co-training algorithm[J].Journal of South China University of Technology(Natural Science Edition),2021,49(6):100-108. | |
| [9] | SHABAN W M, XIE D, ELBAZ K,et al .Real-time water quality prediction of wastewater treatment plants using advanced deep learning networks[J].Journal of Water Process Engineering,2024,65:105775/1-9. |
| [10] | WEN Q, ZHOU T, ZHANG C,et al .Transformers in time series:a survey[EB/OL].(2022-02-15)[2025-04-01].. |
| [11] | ZHOU H, ZHANG S, PENG J,et al .Informer:beyond efficient transformer for long sequence time-series forecasting[C]∥ Proceedings of the Thirty-Fifth AAAI Conference on Artificial Intelligence.Palo Alto:AAAI,2021:11106-11115. |
| [12] | WU H, XU J, WANG J,et al .Autoformer:decomposition transformers with auto-correlation for long-term series forecasting[J].Advances in Neural Information Processing Systems,2021,34:22419-22430. |
| [13] | ZHOU T, MA Z, WEN Q,et al .FEDformer:frequency enhanced decomposed transformer for long-term series forecasting[C]∥ Proceedings of the 39th International Conference on Machine Learning.Baltimore:ML Research Press,2022:27268-27286. |
| [14] | GONG M, ZHAO Y, SUN J,et al .Load forecasting of district heating system based on Informer[J].Energy,2022,253:124179/1-14. |
| [15] | JIANG Y, GAO T, DAI Y,et al .Very short-term residential load forecasting based on deep-autoformer[J].Applied Energy,2022,328:120120/1-13. |
| [16] | JIN Z, FU X, XIANG L,et al .Informer learning framework based on secondary decomposition for multi-step forecast of ultra-short term wind speed[J].Engineering Applications of Artificial Intelligence,2025,139:109702/1-13. |
| [17] | 沈瑜,李江柽,梁栋,等 .特征选择的高效时间序列预测模型[J].哈尔滨工业大学学报,2025-03-21,doi:10.11918/202411025 . |
| SHEN Yu, LI Jiangcheng, LIANG Dong,et al .Efficient time series forecasting model for feature selection[J].Journal of Harbin Institute of Technology,2025-03-21,doi:10.11918/202411025 . | |
| [18] | SIFUZZAMAN M, ISLAM M R, ALI M Z .Application of wavelet transform and its advantages compared to Fourier transform[J].Journal of Physical Sciences,2009,13:121-134. |
| [19] | KUMAR A, TOMAR H, MEHLA V K,et al .Stationary wavelet transform based ECG signal denoising method[J].ISA Transactions,2021,114:251-262. |
| [20] | MICHAU G, FRUSQUE G, FINK O .Fully learnable deep wavelet transform for unsupervised monitoring of high-frequency timeseries[J].Proceedings of the National Academy of Sciences,2022,119(8):e2106598119/1-10. |
| [21] | HAAN P D, COHEN T, BREHMER J .Euclidean,projective,conformal:choosing a geometric algebra for equivariant transformers[C]∥ Proceedings of the 27th International Conference on Artificial Intelligence and Statistics.Valencia:ML Research Press,2024:3088-3096. |
| [22] | CHILD R, GRAY S, RADFORD A,et al .Generating long sequences with sparse transformers[EB/OL].(2019-04-23)[2025-04-01].. |
| [23] | MOJAHED A, BERGMAN L A, VAKAKIS A F .New inverse wavelet transform method with broad application in dynamics[J].Mechanical Systems and Signal Processing,2021,156:107691/1-22. |
| [24] | ZHANG J, LV Y, TAO J,et al .A robust real-time anchor-free traffic sign detector with one-level feature[J].IEEE Transactions on Emerging Topics in Com-putational Intelligence,2024,8(2):1437-1451. |
| [25] | WU H, HU T, LIU Y,et al .TimesNet:temporal 2D-variation modeling for general time series analysis[EB/OL].(2022-10-05)[2025-04-01].. |
| [26] | CHEN H, LUONG V, MUKHERJEE L,et al .SimpleTM:a simple baseline for multivariate time series forecasting[C]∥ Proceedings of the Thirteenth International Conference on Learning Representations.Singapore:OpenReview,2025:3089/1-26. |
/
| 〈 |
|
〉 |