Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Journal articles
  4. Spatially adaptive machine learning models for predicting water quality in Hong Kong
 
research article

Spatially adaptive machine learning models for predicting water quality in Hong Kong

Wang, Qiaoli
•
Li, Zijun
•
Cai, Jiannan
Show more
May 21, 2023
Journal of Hydrology

Water quality prediction in the spatially heterogeneous environment is challenging as the importance of water quality parameters (WQPs) and the performance of prediction models may vary across space. Thus, this study proposed spatially adaptive machine learning models to predict water quality status in Hong Kong. First, spatial clusters with relatively homogeneous water quality were adaptively detected using dynamically constrained agglomerative clustering and partitioning. Then, the optimal prediction models were constructed for each cluster by locally evaluating the prediction performance of six standalone machine learning models, including multi-layer perceptron neural network (MLPNN), support vector machine (SVM), random forest (RF), extremely ran-domized tree (ET), eXtreme gradient boosting (XGBoost) and categorical gradient boosting (CatBoost), as well as four novel hybrid models (MLPNN-SVM, ET-CatBoost, MLPNN-CatBoost and XGBoost-CatBoost). Finally, a sensitivity analysis was conducted to explore the minimum sets of indicative WQPs to achieve more cost-efficient water quality prediction based on locally optimal prediction models. The results revealed that the water quality in the study area was spatially heterogeneous and four spatially contiguous clusters were identified. MLPNN-SVM, ET-CatBoost, MLPNN-CatBoost and CaBboost performed best in Cluster 1 to Cluster 4, with R2 values of 0.917, 0.906, 0.901 and 0.937 and RMSE values of 1.978, 0.843, 2.020 and 1.572, respectively. The results of the sensitivity analysis indicated that acceptable local prediction results can be obtained using fewer WQPs. It is conducive to issuing timely water quality warnings and striving for more time for water pollution remediation.

  • Details
  • Metrics
Type
research article
DOI
10.1016/j.jhydrol.2023.129649
Web of Science ID

WOS:001002623500001

Author(s)
Wang, Qiaoli
Li, Zijun
Cai, Jiannan
Zhang, Mengsheng
Liu, Zida
Xu, Yu
Li, Rongrong  
Date Issued

2023-05-21

Publisher

ELSEVIER

Published in
Journal of Hydrology
Volume

622

Article Number

129649

Subjects

Engineering, Civil

•

Geosciences, Multidisciplinary

•

Water Resources

•

Engineering

•

Geology

•

water quality prediction

•

spatially adaptive strategy

•

locally optimal models

•

machine learning

•

performance evaluation

•

artificial-intelligence

•

regionalization

•

variability

•

parameters

•

index

•

river

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
LBI  
Available on Infoscience
June 19, 2023
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/198407
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés