Toggle Main Menu Toggle Search

Open Access padlockePrints

River Water Quality Prediction in Malaysia Based on Extra Tree Regression Model Coupled with Linear Discriminant Analysis (LDA)

Lookup NU author(s): Dr Zainal Ahmad, Dr Jie ZhangORCiD


Full text for this publication is not currently held within this repository. Alternative links are provided below where available.


© 2021 Elsevier B.V.Poor river water quality as a result of rapid urbanization has led to various disease outbreak and destruction of wildlife ecosystem. Although continuous water quality monitoring has been proposed as the immediate solution, the complexity and expensive nature of such system often offset its purpose. Therefore, numerous studies have been made to forecast water quality utilizing different machine learning algorithms. In this paper, a comparative study was carried out to develop an efficient water quality index (WQI) prediction model based on more attainable monitoring parameters. Three different variation of ensemble decision tree models were analysed and compared, namely: Random forest regression (RF), extra tree regression (ETR), and decision tree + AdaBoost (BTR). These models were coupled with principle component analysis (PCA) and linear discriminant analysis (LDA) to reduce the dimensions of the input data. The results show that (ETR-LDA) model outperform the other ensemble tree models with an R2 value up to 0.88. The ETR-LDA combination consistently score a higher R2 values even when trained with a much smaller input dimension (i.e. 3) resulting in a faster training time. This model could positively contribute towards the long-term water quality management effort in a cost-effective manner.

Publication metadata

Author(s): Djarum DH, Ahmad Z, Zhang J

Publication type: Book Chapter

Publication status: Published

Book Title: 31st European Symposium on Computer Aided Process Engineering

Year: 2021

Volume: 50

Pages: 1491-1496

Print publication date: 25/06/2021

Online publication date: 18/07/2021

Acceptance date: 02/04/2020

Series Title: Computer Aided Chemical Engineering

Publisher: Elsevier B.V.

Place Published: Amsterdam


DOI: 10.1016/B978-0-323-88506-5.50230-8

Library holdings: Search Newcastle University Library for this item

ISBN: 9780323885065