Wastewater discharge quality prediction using stratified sampling and wavelet de-noising ANFIS model. (July 2020)

• Record Type: Journal Article
• Authors: Fu, Z.; Cheng, J.; Yang, M.; Batista, J.; Jiang, Y.
• Is Part Of: Computers & electrical engineering. Volume 85(2020)
• Publication Date: 2020-07
• Subjects: Wastewater quality prediction -- Machine learning -- ANFIS -- WAVELET-ANFIS -- Stratified sampling; Computer engineering -- Periodicals; Electrical engineering -- Periodicals; Electrical engineering -- Data processing -- Periodicals; Ordinateurs -- Conception et construction -- Périodiques; Électrotechnique -- Périodiques; Électrotechnique -- Informatique -- Périodiques; Computer engineering; Electrical engineering; Electrical engineering -- Data processing; Periodicals; Electronic journals; 621.302854
• DOI: 10.1016/j.compeleceng.2020.106701
• Languages: English
• ISSNs: 0045-7906

Abstract

Highlights: A general input parameter selection method is proposed, combined with an optimized wavelet de-noising ANFIS model to predict wastewater quality. A network structure selection method is given to enhance the stability and scalability of ANFIS. Statistical stratified sampling is employed to proportionally partition the dataset. The experimental results confirm that the proposed model outperforms other existing AI models. 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 R 2 testing value achieves 0.976, 0.975, 0.988, and 0.986 in predicting chloride, sulfate, electrical conductivity, and total dissolved solids, respectively. Graphical abstract: Image, graphical abstract