Multi-task learning framework for predicting water quality using non-linear machine learning technique

Abstract

Predicting the quality of water is a very important issue in an ecosystem and it can be used to control the increase of water contamination. Also, water quality prediction is a prominent complex non-linear multi-target learning problem and extracting a relevant subset of features from a large number of features with multiple targets is a challenging task. Existing water quality prediction model not focused on multi-target learning process simultaneously and not identifying the non-linear relationship between the features and target variables. Therefore, this study proposes a multi-task learning method dealing with multi-target regression using non-linear machine learning technique. Finally, experiments are conducted to build a prediction model based on the proposed methods to evaluate accuracy on water quality dataset. The experimental results indicate that our method increases the overall accuracy of the experimental dataset compared with the existing methods with the reduced number of significant features.