Long-Term Water Quality Prediction with Patch Savitsky-Golay Filtering and Transformer
Document Type
Conference Proceeding
Publication Date
1-1-2024
Abstract
In many fields, time series prediction is gaining more and more attention, e.g., air pollution, geological hazards, and network traffic prediction. Water quality prediction is based on historical data to predict future water quality. However, it is difficult to learn a representation map from a time series that captures the trends and fluctuations to effectively remove noise from time series data and capture complex nonlinear relationships. To solve these problems, this work proposes a time series prediction model, called PSGT for short, which integrates Patch Savitsky-Golay filtering and Transformer. First, this work adopts a Patching method to embed sub-time series data and obtains the trends and semantic information of the time series. Second, it uses the Savitsky-Golay filtering to effectively remove the noise data in the patch and improve the prediction accuracy. Third, it uses a Transformer mechanism to address the nonlinear problem of water quality time series and improve long-term prediction capability. Two real-world datasets are utilized to evaluate the proposed PSGT, and experiments prove that PSGT performs better than other benchmark models by at least 6%.
Identifier
85217851472 (Scopus)
ISBN
[9781665410205]
Publication Title
Conference Proceedings - IEEE International Conference on Systems, Man and Cybernetics
External Full Text Location
https://doi.org/10.1109/SMC54092.2024.10831237
ISSN
1062922X
First Page
4827
Last Page
4832
Grant
4232049
Fund Ref
Natural Science Foundation of Beijing Municipality
Recommended Citation
Lin, Yongze; Qiao, Junfei; Bi, Jing; Yuan, Haitao; Zhai, Jiahui; and Zhou, Meng Chu, "Long-Term Water Quality Prediction with Patch Savitsky-Golay Filtering and Transformer" (2024). Faculty Publications. 713.
https://digitalcommons.njit.edu/fac_pubs/713