Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learning
BackgroundAcute ST-segment elevation myocardial infarction (STEMI) is a cardiovascular emergency that is associated with a high risk of death. In this study, we developed explainable machine learning models to predict the overall survival (OS) of STEMI patients to help improve prognosis and increase...
Saved in:
| Main Authors: | , , , , , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Frontiers Media S.A.
2025-07-01
|
| Series: | Frontiers in Medicine |
| Subjects: | |
| Online Access: | https://www.frontiersin.org/articles/10.3389/fmed.2025.1594273/full |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1839625136411508736 |
|---|---|
| author | Tao Shi Jianping Yang Jianping Yang Yanji Zhou Sirui Yang Fazhi Yang Xinuo Ma Yujuan Peng Jinfang Pu Hong Wei Lixing Chen |
| author_facet | Tao Shi Jianping Yang Jianping Yang Yanji Zhou Sirui Yang Fazhi Yang Xinuo Ma Yujuan Peng Jinfang Pu Hong Wei Lixing Chen |
| author_sort | Tao Shi |
| collection | DOAJ |
| description | BackgroundAcute ST-segment elevation myocardial infarction (STEMI) is a cardiovascular emergency that is associated with a high risk of death. In this study, we developed explainable machine learning models to predict the overall survival (OS) of STEMI patients to help improve prognosis and increase survival.MethodsAfter applying the inclusion and exclusion criteria, we selected 893 patients who underwent emergency coronary angiography and percutaneous coronary intervention (PCI) for STEMI at the First Affiliated Hospital of Kunming Medical University. The best predictor variables were screened by least absolute shrinkage and selection operator (LASSO) regression. These variables were used to construct Cox proportional hazards regression (coxph) and random survival forest (rfsrc) models. Three criteria (C-index, Brier score, and C/D AUC) were utilised to compare the performance of the two models. Then, by applying the time-dependent variable importance and the partial dependence survival profile, a global explanation of the entire cohort was conducted. Finally, local explanations for individual patients were performed with the SurvSHAP(t) and SurvLIME plots and the ceteris paribus survival profile.ResultsCombining the results of the comparison of the three criteria, the performance of the rfsrc model was shown to be superior to that of the coxph model. LASSO regression was used to screen 11 predictor variables, such as diastolic blood pressure (DBP), Killip class, hyperlipidaemia, global registry of acute coronary events (GRACE) Score, creatine kinase isoenzyme-MB, myoglobin, white blood cells, monocytes, thrombin time, globulin (GLB), and conjugated bilirubin. The global explanation of the whole cohort revealed that DBP, GRACE Score, myoglobin, and monocytes had a significant effect on the OS of STEMI patients in the coxph model and that DBP, GRACE Score, and GLB were the variables that significantly affected the OS of STEMI patients in the rfsrc model. Incorporating a single patient into the model can yield a local explanation of each patient, thus guiding clinicians in developing precision treatments.ConclusionThe rfsrc model outperformed the coxph model in terms of predictive performance. Clinicians can use these predictive models to understand the major risk factors for each STEMI patient and thus develop more individualised and precise treatment strategies. |
| format | Article |
| id | doaj-art-c1cb7029ab8745dc8e38e2d92c575c63 |
| institution | Matheson Library |
| issn | 2296-858X |
| language | English |
| publishDate | 2025-07-01 |
| publisher | Frontiers Media S.A. |
| record_format | Article |
| series | Frontiers in Medicine |
| spelling | doaj-art-c1cb7029ab8745dc8e38e2d92c575c632025-07-18T05:30:26ZengFrontiers Media S.A.Frontiers in Medicine2296-858X2025-07-011210.3389/fmed.2025.15942731594273Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learningTao Shi0Jianping Yang1Jianping Yang2Yanji Zhou3Sirui Yang4Fazhi Yang5Xinuo Ma6Yujuan Peng7Jinfang Pu8Hong Wei9Lixing Chen10Department of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaCollege of Big Data, Yunnan Agricultural University, Kunming, ChinaThe Key Laboratory for Crop Production and Smart Agriculture of Yunnan Province, Kunming, ChinaDepartment of Pediatrics, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaDepartment of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaDepartment of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaDepartment of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaDepartment of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaDepartment of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaDepartment of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaDepartment of Cardiology, The First Affiliated Hospital of Kunming Medical University, Kunming, ChinaBackgroundAcute ST-segment elevation myocardial infarction (STEMI) is a cardiovascular emergency that is associated with a high risk of death. In this study, we developed explainable machine learning models to predict the overall survival (OS) of STEMI patients to help improve prognosis and increase survival.MethodsAfter applying the inclusion and exclusion criteria, we selected 893 patients who underwent emergency coronary angiography and percutaneous coronary intervention (PCI) for STEMI at the First Affiliated Hospital of Kunming Medical University. The best predictor variables were screened by least absolute shrinkage and selection operator (LASSO) regression. These variables were used to construct Cox proportional hazards regression (coxph) and random survival forest (rfsrc) models. Three criteria (C-index, Brier score, and C/D AUC) were utilised to compare the performance of the two models. Then, by applying the time-dependent variable importance and the partial dependence survival profile, a global explanation of the entire cohort was conducted. Finally, local explanations for individual patients were performed with the SurvSHAP(t) and SurvLIME plots and the ceteris paribus survival profile.ResultsCombining the results of the comparison of the three criteria, the performance of the rfsrc model was shown to be superior to that of the coxph model. LASSO regression was used to screen 11 predictor variables, such as diastolic blood pressure (DBP), Killip class, hyperlipidaemia, global registry of acute coronary events (GRACE) Score, creatine kinase isoenzyme-MB, myoglobin, white blood cells, monocytes, thrombin time, globulin (GLB), and conjugated bilirubin. The global explanation of the whole cohort revealed that DBP, GRACE Score, myoglobin, and monocytes had a significant effect on the OS of STEMI patients in the coxph model and that DBP, GRACE Score, and GLB were the variables that significantly affected the OS of STEMI patients in the rfsrc model. Incorporating a single patient into the model can yield a local explanation of each patient, thus guiding clinicians in developing precision treatments.ConclusionThe rfsrc model outperformed the coxph model in terms of predictive performance. Clinicians can use these predictive models to understand the major risk factors for each STEMI patient and thus develop more individualised and precise treatment strategies.https://www.frontiersin.org/articles/10.3389/fmed.2025.1594273/fullacute ST-segment elevation myocardial infarctionLASSO regressionexplainable machine learningCox proportional hazards regressionrandom survival forest |
| spellingShingle | Tao Shi Jianping Yang Jianping Yang Yanji Zhou Sirui Yang Fazhi Yang Xinuo Ma Yujuan Peng Jinfang Pu Hong Wei Lixing Chen Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learning Frontiers in Medicine acute ST-segment elevation myocardial infarction LASSO regression explainable machine learning Cox proportional hazards regression random survival forest |
| title | Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learning |
| title_full | Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learning |
| title_fullStr | Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learning |
| title_full_unstemmed | Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learning |
| title_short | Survival prediction modelling in patients with acute ST-segment elevation myocardial infarction with LASSO regression and explainable machine learning |
| title_sort | survival prediction modelling in patients with acute st segment elevation myocardial infarction with lasso regression and explainable machine learning |
| topic | acute ST-segment elevation myocardial infarction LASSO regression explainable machine learning Cox proportional hazards regression random survival forest |
| url | https://www.frontiersin.org/articles/10.3389/fmed.2025.1594273/full |
| work_keys_str_mv | AT taoshi survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT jianpingyang survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT jianpingyang survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT yanjizhou survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT siruiyang survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT fazhiyang survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT xinuoma survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT yujuanpeng survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT jinfangpu survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT hongwei survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning AT lixingchen survivalpredictionmodellinginpatientswithacutestsegmentelevationmyocardialinfarctionwithlassoregressionandexplainablemachinelearning |