UM  > Faculty of Science and Technology
Residential Collegefalse
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 Name18th International Conference on Wavelet Analysis and Pattern Recognition, ICWAPR 2021
Source PublicationInternational Conference on Wavelet Analysis and Pattern Recognition
Volume2021-December
Conference Date04-05 December 2021
Conference PlaceAdelaide, Australia
CountryAustralia
PublisherIEEE
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.

KeywordWater Quality Prediction Attention Mechanism Principal Component Analysis Long Short-term Memory Networks Spar-rows Search Algorithm
DOI10.1109/ICWAPR54887.2021.9736154
URLView the original
Indexed ByCPCI-S
Language英語English
WOS Research AreaComputer Science ; Engineering ; Mathematics
WOS SubjectComputer Science, Artificial Intelligence ; Computer Science, Interdisciplinary Applications ; Engineering, Electrical & Electronic ; Mathematics, Applied
WOS IDWOS:000806728800009
Scopus ID2-s2.0-85127615396
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionFaculty of Science and Technology
DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Affiliation1.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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Xiao-Feng Wang]'s Articles
[Sheng-Nan Wei]'s Articles
[Li-Xiang Xu]'s Articles
Baidu academic
Similar articles in Baidu academic
[Xiao-Feng Wang]'s Articles
[Sheng-Nan Wei]'s Articles
[Li-Xiang Xu]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Xiao-Feng Wang]'s Articles
[Sheng-Nan Wei]'s Articles
[Li-Xiang Xu]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.