Wastewater Discharge Quality Prediction Using Stratified Sampling and Wavelet De-Noising ANFIS Model
Document Type
Article
Publication Date
7-1-2020
Publication Title
Computers & Electrical Engineering
Volume
85
First page number:
1
Last page number:
15
Abstract
The monitoring of wastewater quality is vitally important for the stability of an ecosystem. Among many machine learning techniques proposed for predicting the quality of wastewater, the adaptive neuro-fuzzy inference system (ANFIS) can achieve the best accuracy. However, due to data size limitations, the uneven distribution of randomly sampled training data can cause out-of-range prediction errors in the ANFIS model. In this study, a general-purpose input parameter selection method is proposed, combined with an optimized wavelet de-noising ANFIS model, to predict salinity parameters in wastewater discharge samples from the Las Vegas Wash, Nevada, USA. A statistical, stratified sampling strategy is used to preprocess the wastewater quality dataset. Compared with existing artificial intelligence models, the experimental results prove that the proposed model has the best performance, in which the R2 testing value achieves 0.976, 0.975, 0.988, and 0.986 in predicting chloride, sulfate, electrical conductivity, and total dissolved solids, respectively.
Keywords
Wastewater quality prediction; Machine learning; ANFIS; WAVELET-ANFIS; Stratified sampling
Disciplines
Electrical and Computer Engineering | Engineering
Language
English
Repository Citation
Fu, Z.,
Cheng, J.,
Yang, M.,
Batista, J.,
Jiang, Y.
(2020).
Wastewater Discharge Quality Prediction Using Stratified Sampling and Wavelet De-Noising ANFIS Model.
Computers & Electrical Engineering, 85
1-15.
http://dx.doi.org/10.1016/j.compeleceng.2020.106701