Water quality prediction: A data-driven approach exploiting advanced machine learning algorithms with data augmentation

IF 2.7 4区 环境科学与生态学 Q2 WATER RESOURCES Journal of Water and Climate Change Pub Date : 2023-12-20 DOI:10.2166/wcc.2023.403
Karthick K, S. Krishnan, R. Manikandan
{"title":"Water quality prediction: A data-driven approach exploiting advanced machine learning algorithms with data augmentation","authors":"Karthick K, S. Krishnan, R. Manikandan","doi":"10.2166/wcc.2023.403","DOIUrl":null,"url":null,"abstract":"\n \n Water quality assessment plays a crucial role in various aspects, including human health, environmental impact, agricultural productivity, and industrial processes. Machine learning (ML) algorithms offer the ability to automate water quality evaluation and allow for effective and rapid assessment of parameters associated with water quality. This article proposes an ML-based classification model for water quality prediction. The model was tested with 14 ML algorithms and considers 20 features that represent various substances present in water samples and their concentrations. The dataset used in the study comprises 7,996 samples, and the model development involves several stages, including data preprocessing, Yeo–Johnson transformation for data normalization, principal component analysis (PCA) for feature selection, and the application of the synthetic minority over-sampling technique (SMOTE) to address class imbalance. Performance metrics, such as accuracy, precision, recall, and F1 score, are provided for each algorithm with and without SMOTE. LightGBM, XGBoost, CatBoost, and Random Forest were identified as the best-performing algorithms. LightGBM achieved the highest accuracy of 96.25% without SMOTE, while XGBoost attained the highest precision of 0.933. The application of SMOTE enhanced the performance of CatBoost. These findings provide valuable insights for ML-based water quality assessment, aiding researchers and professionals in decision-making and management.","PeriodicalId":49150,"journal":{"name":"Journal of Water and Climate Change","volume":"104 10","pages":""},"PeriodicalIF":2.7000,"publicationDate":"2023-12-20","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":null,"platform":"Semanticscholar","paperid":null,"PeriodicalName":"Journal of Water and Climate Change","FirstCategoryId":"93","ListUrlMain":"https://doi.org/10.2166/wcc.2023.403","RegionNum":4,"RegionCategory":"环境科学与生态学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q2","JCRName":"WATER RESOURCES","Score":null,"Total":0}
引用次数: 0

Abstract

Water quality assessment plays a crucial role in various aspects, including human health, environmental impact, agricultural productivity, and industrial processes. Machine learning (ML) algorithms offer the ability to automate water quality evaluation and allow for effective and rapid assessment of parameters associated with water quality. This article proposes an ML-based classification model for water quality prediction. The model was tested with 14 ML algorithms and considers 20 features that represent various substances present in water samples and their concentrations. The dataset used in the study comprises 7,996 samples, and the model development involves several stages, including data preprocessing, Yeo–Johnson transformation for data normalization, principal component analysis (PCA) for feature selection, and the application of the synthetic minority over-sampling technique (SMOTE) to address class imbalance. Performance metrics, such as accuracy, precision, recall, and F1 score, are provided for each algorithm with and without SMOTE. LightGBM, XGBoost, CatBoost, and Random Forest were identified as the best-performing algorithms. LightGBM achieved the highest accuracy of 96.25% without SMOTE, while XGBoost attained the highest precision of 0.933. The application of SMOTE enhanced the performance of CatBoost. These findings provide valuable insights for ML-based water quality assessment, aiding researchers and professionals in decision-making and management.
查看原文
分享 分享
微信好友 朋友圈 QQ好友 复制链接
本刊更多论文
水质预测:利用先进的机器学习算法和数据增强技术的数据驱动方法
水质评估在人类健康、环境影响、农业生产力和工业流程等各个方面都发挥着至关重要的作用。机器学习(ML)算法能够自动进行水质评价,并能有效、快速地评估与水质相关的参数。本文提出了一种基于 ML 的水质预测分类模型。该模型使用 14 种 ML 算法进行了测试,并考虑了代表水样中各种物质及其浓度的 20 个特征。研究中使用的数据集包括 7,996 个样本,模型开发涉及多个阶段,包括数据预处理、用于数据归一化的 Yeo-Johnson 转换、用于特征选择的主成分分析 (PCA),以及用于解决类不平衡问题的合成少数过度采样技术 (SMOTE)。每种算法在有 SMOTE 和没有 SMOTE 的情况下,都有准确度、精确度、召回率和 F1 分数等性能指标。LightGBM、XGBoost、CatBoost 和随机森林被认为是性能最好的算法。在没有 SMOTE 的情况下,LightGBM 的准确率最高,达到 96.25%,而 XGBoost 的精度最高,达到 0.933。SMOTE 的应用提高了 CatBoost 的性能。这些发现为基于 ML 的水质评估提供了宝贵的见解,有助于研究人员和专业人员进行决策和管理。
本文章由计算机程序翻译,如有差异,请以英文原文为准。
求助全文
约1分钟内获得全文 去求助
来源期刊
CiteScore
4.80
自引率
10.70%
发文量
168
审稿时长
>12 weeks
期刊介绍: Journal of Water and Climate Change publishes refereed research and practitioner papers on all aspects of water science, technology, management and innovation in response to climate change, with emphasis on reduction of energy usage.
期刊最新文献
Analysis of different hypotheses for modeling air–water exchange and temperature evolution in a tropical reservoir Accounting for climate change in the water infrastructure design: evaluating approaches and recommending a hybrid framework Climatic characteristics and main weather patterns of extreme precipitation in the middle Yangtze River valley Water quality prediction: A data-driven approach exploiting advanced machine learning algorithms with data augmentation Consequence assessment of the La Giang dike breach in the Ca river system, Vietnam
×
引用
GB/T 7714-2015
复制
MLA
复制
APA
复制
导出至
BibTeX EndNote RefMan NoteFirst NoteExpress
×
×
提示
您的信息不完整,为了账户安全,请先补充。
现在去补充
×
提示
您因"违规操作"
具体请查看互助需知
我知道了
×
提示
现在去查看 取消
×
提示
确定
0
微信
客服QQ
Book学术公众号 扫码关注我们
反馈
×
意见反馈
请填写您的意见或建议
请填写您的手机或邮箱
已复制链接
已复制链接
快去分享给好友吧!
我知道了
×
扫码分享
扫码分享
Book学术官方微信
Book学术文献互助
Book学术文献互助群
群 号:481959085
Book学术
文献互助 智能选刊 最新文献 互助须知 联系我们:info@booksci.cn
Book学术提供免费学术资源搜索服务,方便国内外学者检索中英文文献。致力于提供最便捷和优质的服务体验。
Copyright © 2023 Book学术 All rights reserved.
ghs 京公网安备 11010802042870号 京ICP备2023020795号-1