Wastewater Discharge Quality Prediction Using Stratified Sampling and Wavelet De-Noising ANFIS Model
Computers & Electrical Engineering
First page number:
Last page number:
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.
Wastewater quality prediction; Machine learning; ANFIS; WAVELET-ANFIS; Stratified sampling
Electrical and Computer Engineering | Engineering
Wastewater Discharge Quality Prediction Using Stratified Sampling and Wavelet De-Noising ANFIS Model.
Computers & Electrical Engineering, 85