國家衛生研究院 NHRI:Item 3990099045/16730

NHRI > National Institute of Environmental Health Sciences > Pau-Chung Chen > Periodical Articles > Item 3990099045/16730

Please use this identifier to cite or link to this item: http://ir.nhri.org.tw/handle/3990099045/16730

Title:	Effects of feature selection methods in estimating SO2 concentration variations using machine learning and stacking ensemble approach
Authors:	Wong, PY;Zeng, YT;Su, HJ;Lung, SCC;Chen, YC;Chen, PC;Hsiao, TC;Adamkiewicz, G;Wu, CD
Contributors:	National Institute of Environmental Health Sciences
Abstract:	Statistical-based feature selection methods have been used for dimension reduction, but only a few studies have explored the impact of selected features on machine learning models. This study aims to investigate the effects of statistical and machine learning-based feature selection methods on spatial prediction models for estimating variations in SO2 concentrations. We collected daily SO2 observations from 1994 to 2018 along with predictor variables such as land-use/land cover allocations, roads, landmarks, meteorological factors, and satellite images, resulting in a total of 428 geographic predictors. Important features were identified using statistical-based feature selection methods including SelectKBest, stepwise feature selection, elastic net, and machine learning-based methods such as random forest. The selected features from the four feature selection methods were fitted to machine learning algorithms including gradient boosting, Cat- Boost, XGBoost, and stacking ensemble to establish prediction models for estimating SO2 concentrations. SHapley Additive exPlanations (SHAP) was applied to explain the contribution of each selected feature to the model's prediction capability. The results showed that stacking ensemble model outperformed the three single machine learning algorithms. Among the four feature selection methods, the random forest method yielded the highest prediction accuracy (R2=0.80) in the training model, followed by stepwise selection (R2=0.75), SelectKBest (R2=0.75), and elastic net (R2=0.72) in the stacking ensemble model. These results were robust after several validation tests. Our findings suggested that the random forest feature selection method was more suitable for developing machine learning models for air pollution estimation. The identified features also provide important information for urban air pollution management.
Date:	2025-02
Relation:	Environmental Technology and Innovation. 2025 Feb;37:Article number 103996.
Link to:	http://dx.doi.org/10.1016/j.eti.2024.103996
JIF/Ranking 2023:	http://gateway.webofknowledge.com/gateway/Gateway.cgi?GWVersion=2&SrcAuth=NHRI&SrcApp=NHRI_IR&KeyISSN=2352-1864&DestApp=IC2JCR
Cited Times(WOS):	https://www.webofscience.com/wos/woscc/full-record/WOS:001399960600001
Cited Times(Scopus):	https://www.scopus.com/inward/record.url?partnerID=HzOxMe3b&scp=85213288196
Appears in Collections:	[Pau-Chung Chen] Periodical Articles [Yu-Cheng Chen] Periodical Articles

Files in This Item:

File	Description	Size	Format
ISI001399960600001.pdf		7290Kb	Adobe PDF	35	View/Open

Loading...