Two novelty learning models developed based on deep cascade forest to address the environmental imbalanced issues: A case study of drinking water quality prediction

Abstract

Environmental quality data sets are typically imbalanced, because environmental pollution events are rarely observed in daily life. Prediction of imbalanced data sets is a major challenge in machine learning. Our recent work has shown deep cascade forest (DCF), as a base learning model, is promising to be recommended for environmental quality prediction. Although some traditional models were improved by introducing the cost matrix, little is known about whether cost matrix could enhance the prediction performance of DCF. Additionally, feature extraction is also an important way to potentially improve the model's ability to predict the imbalanced data. Here, we developed two novelty learning models based on DCF: cost-sensitive DCF (CS-DCF) and DCF that combines unsupervised learning models and greedy methods (USM-DCF-G). Subsequently, CS-DCF and USM-DCF-G were successfully verified by an imbalanced drinking water quality data set. Our data presented both CS-DCF and USM-DCF-G show better prediction performance than that of DCF alone did. In particular, USM-DCF-G shows the best performance with the highest F1-score (95.12 ± 2.56%), after feature extraction and selection by using unsupervised learning models and greedy methods. Thus, the two learning models, especially USM-DCF-G, were promising learning models to address environmental imbalanced issues and accurately predict environmental quality.

Keywords: Cost-sensitive; Deep cascade forest; Environmental imbalance issues; Feature extraction; Feature selection.