Machine learning prediction of STEMI incidence with SHAP interpretation of environmental determinants

利用SHAP解释环境决定因素的机器学习预测STEMI发生率

阅读:1

Abstract

ST-segment elevation myocardial infarction (STEMI) is a life-threatening cardiovascular event influenced by meteorological conditions and air pollution. Traditional statistical methods often fail to capture the complex, nonlinear relationships between environmental factors and STEMI risk. This study analyzed hospitalization data from Harbin, China (2014-2023), alongside meteorological and air pollution data. Seven machine learning models, including LightGBM and XGBoost, were used to predict STEMI risk. Key predictors were identified via recursive feature elimination, and model performance was assessed with nested cross-validation. Shapley Additive Explanations (SHAP) were applied to interpret the impact of key predictors. Model performance was evaluated using metrics such as AUC, accuracy, and recall. LightGBM achieved the best performance, with an AUC of 0.84 on the validation set, demonstrating high predictive accuracy and generalizability. Moreover, the LightGBM model improved its interpretability and predictive power by incorporating lagged effects along with recursive feature elimination (RFE). SHAP analysis identified PM[Formula: see text], SO[Formula: see text], NO[Formula: see text], and meteorological factors (e.g., air pressure, humidity, wind speed) as critical contributors, with significant lag effects observed. These findings underscore the cumulative impact of prolonged environmental exposure on cardiovascular health. This study developed a robust, interpretable predictive model that elucidates the complex relationship between environmental factors and STEMI incidence. The results provide valuable insights for early prevention, public health policymaking, and resource allocation.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。