Skip to main content

Lund University Publications

LUND UNIVERSITY LIBRARIES

Enhanced TDS Modeling Using an AI Framework Integrating Grey Wolf Optimization with Kernel Extreme Learning Machine

Sayadi, Maryam LU ; Hessari, Behzad ; Montaseri, Majid and Naghibi, Amir LU (2024) In Water (Switzerland) 16(19).
Abstract

Predictions of total dissolved solids (TDS) in water bodies including rivers and lakes are challenging but essential for the effective management of water resources in agricultural and drinking water sectors. This study developed a hybrid model combining Grey Wolf Optimization (GWO) and Kernel Extreme Learning Machine (KELM) called GWO-KELM to model TDS in water bodies. Time series data for TDS and its driving factors, such as chloride, temperature, and total hardness, were collected from 1975 to 2016 to train and test machine learning models. The study aimed to assess the performance of the GWO-KELM model in comparison to other state-of-the-art machine learning algorithms. Results showed that the GWO-KELM model outperformed all other... (More)

Predictions of total dissolved solids (TDS) in water bodies including rivers and lakes are challenging but essential for the effective management of water resources in agricultural and drinking water sectors. This study developed a hybrid model combining Grey Wolf Optimization (GWO) and Kernel Extreme Learning Machine (KELM) called GWO-KELM to model TDS in water bodies. Time series data for TDS and its driving factors, such as chloride, temperature, and total hardness, were collected from 1975 to 2016 to train and test machine learning models. The study aimed to assess the performance of the GWO-KELM model in comparison to other state-of-the-art machine learning algorithms. Results showed that the GWO-KELM model outperformed all other models (such as Artificial Neural Network, Gaussian Process Regression, Support Vector Machine, Linear Regression, Classification and Regression Tree, and Boosted Regression Trees), achieving the highest coefficient of determination (R2) value of 0.974, indicating excellent predictive accuracy. It also recorded the lowest root mean square error (RMSE) of 55.75 and the lowest mean absolute error (MAE) of 34.40, reflecting the smallest differences between predicted and actual values. The values of R2, RMSE, and MAE for other machine learning models were in the ranges of 0.969–0.895, 60.13–108.939, and 38.25–53.828, respectively. Thus, it can be concluded that the modeling approaches in this study were in close competition with each other and, finally, the GWO-KELM model had the best performance.

(Less)
Please use this url to cite or link to this publication:
author
; ; and
organization
publishing date
type
Contribution to journal
publication status
published
subject
keywords
artificial neural networks, classification and regression trees, Gaussian process regression, support vector machine, water quality
in
Water (Switzerland)
volume
16
issue
19
article number
2818
publisher
MDPI AG
external identifiers
  • scopus:85206498483
ISSN
2073-4441
DOI
10.3390/w16192818
language
English
LU publication?
yes
additional info
Publisher Copyright: © 2024 by the authors.
id
c0a9401d-da00-4e10-b103-d2b7c5ad8ef3
date added to LUP
2024-12-10 11:26:14
date last changed
2025-01-02 10:02:01
@article{c0a9401d-da00-4e10-b103-d2b7c5ad8ef3,
  abstract     = {{<p>Predictions of total dissolved solids (TDS) in water bodies including rivers and lakes are challenging but essential for the effective management of water resources in agricultural and drinking water sectors. This study developed a hybrid model combining Grey Wolf Optimization (GWO) and Kernel Extreme Learning Machine (KELM) called GWO-KELM to model TDS in water bodies. Time series data for TDS and its driving factors, such as chloride, temperature, and total hardness, were collected from 1975 to 2016 to train and test machine learning models. The study aimed to assess the performance of the GWO-KELM model in comparison to other state-of-the-art machine learning algorithms. Results showed that the GWO-KELM model outperformed all other models (such as Artificial Neural Network, Gaussian Process Regression, Support Vector Machine, Linear Regression, Classification and Regression Tree, and Boosted Regression Trees), achieving the highest coefficient of determination (R<sup>2</sup>) value of 0.974, indicating excellent predictive accuracy. It also recorded the lowest root mean square error (RMSE) of 55.75 and the lowest mean absolute error (MAE) of 34.40, reflecting the smallest differences between predicted and actual values. The values of R<sup>2</sup>, RMSE, and MAE for other machine learning models were in the ranges of 0.969–0.895, 60.13–108.939, and 38.25–53.828, respectively. Thus, it can be concluded that the modeling approaches in this study were in close competition with each other and, finally, the GWO-KELM model had the best performance.</p>}},
  author       = {{Sayadi, Maryam and Hessari, Behzad and Montaseri, Majid and Naghibi, Amir}},
  issn         = {{2073-4441}},
  keywords     = {{artificial neural networks; classification and regression trees; Gaussian process regression; support vector machine; water quality}},
  language     = {{eng}},
  number       = {{19}},
  publisher    = {{MDPI AG}},
  series       = {{Water (Switzerland)}},
  title        = {{Enhanced TDS Modeling Using an AI Framework Integrating Grey Wolf Optimization with Kernel Extreme Learning Machine}},
  url          = {{http://dx.doi.org/10.3390/w16192818}},
  doi          = {{10.3390/w16192818}},
  volume       = {{16}},
  year         = {{2024}},
}