Residential College | false |
Status | 已發表Published |
LSTM Wastewater Quality Prediction Based on Attention Mechanism | |
Xiao-Feng Wang1; Sheng-Nan Wei2; Li-Xiang Xu1; Jun Pan3; Zhi-Ze Wu1; Timothy C. H. Kwong4; Yuan-Yan Tang5,6 | |
2021-12 | |
Conference Name | 18th International Conference on Wavelet Analysis and Pattern Recognition, ICWAPR 2021 |
Source Publication | International Conference on Wavelet Analysis and Pattern Recognition |
Volume | 2021-December |
Conference Date | 04-05 December 2021 |
Conference Place | Adelaide, Australia |
Country | Australia |
Publisher | IEEE |
Abstract | In order to solve the problem that some key water quality pa-rameters in the wastewater treatment process cannot be obtained in real time, a long and short-term memory network water qual-ity prediction model (SSAA-LSTM) based on SSA and attention mechanism is proposed. The model takes historical data as in-put, constructs models to learn the internal dynamic change law of features, introduces the attention mechanism, assigns different weights to the hidden state of the long and short-term memory network (LSTM) by mapping weighting and learning parame-ter matrix, and uses the sparrow search optimization algorithm (SSA) to achieve the optimal selection of hyperparameters for this model. The high-latitude feature vectors are prone to the dimensional disaster problem, so the SSAA-LSTM model with principal component analysis(PCA-LSTM) is further proposed to reduce the dimensionality of the original data. Finally, the SSAA-LSTM model without principal component analysis (NPCA-LSTM) and the PCA-LSTM model are applied in wastewater quality prediction, and the results show that the PCA-LSTM model has the higher predictive ability. |
Keyword | Water Quality Prediction Attention Mechanism Principal Component Analysis Long Short-term Memory Networks Spar-rows Search Algorithm |
DOI | 10.1109/ICWAPR54887.2021.9736154 |
URL | View the original |
Indexed By | CPCI-S |
Language | 英語English |
WOS Research Area | Computer Science ; Engineering ; Mathematics |
WOS Subject | Computer Science, Artificial Intelligence ; Computer Science, Interdisciplinary Applications ; Engineering, Electrical & Electronic ; Mathematics, Applied |
WOS ID | WOS:000806728800009 |
Scopus ID | 2-s2.0-85127615396 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Affiliation | 1.School of Artificial Intelligence and Big Data, Hefei University, Hefei, China 2.School of Bio-Food and Environment, Hefei University, Hefei, 230601, China 3.Hefei Aichuang Microelectronics Technology Co., Ltd, Hefei, China 4.Tung Wah College, Hong Kong 5.Zhuhai UM Science and Technology Research Institute 6.FST University of Macau, Macau |
Recommended Citation GB/T 7714 | Xiao-Feng Wang,Sheng-Nan Wei,Li-Xiang Xu,et al. LSTM Wastewater Quality Prediction Based on Attention Mechanism[C]:IEEE, 2021. |
APA | Xiao-Feng Wang., Sheng-Nan Wei., Li-Xiang Xu., Jun Pan., Zhi-Ze Wu., Timothy C. H. Kwong., & Yuan-Yan Tang (2021). LSTM Wastewater Quality Prediction Based on Attention Mechanism. International Conference on Wavelet Analysis and Pattern Recognition, 2021-December. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment