Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model.
<h4>Background</h4>Acute myocardial infarction (AMI) remains a leading cause of hospitalization and death in China. Accurate mortality prediction of inpatient is crucial for clinical decision-making of non-ST-segment elevation myocardial infarction (NSTEMI) patients.<h4>Methods<...
Saved in:
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2024-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://doi.org/10.1371/journal.pone.0312448 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1850281568990396416 |
|---|---|
| author | Li Wang Yu Zhang Feng Li Caiyun Li Hongzeng Xu |
| author_facet | Li Wang Yu Zhang Feng Li Caiyun Li Hongzeng Xu |
| author_sort | Li Wang |
| collection | DOAJ |
| description | <h4>Background</h4>Acute myocardial infarction (AMI) remains a leading cause of hospitalization and death in China. Accurate mortality prediction of inpatient is crucial for clinical decision-making of non-ST-segment elevation myocardial infarction (NSTEMI) patients.<h4>Methods</h4>In this study, a total of 3061 patients between January 1, 2017 and December 31, 2022 diagnosed with NSTEMI were enrolled in this study. A new method based on Stacking ensemble model is proposed to predict the in-hospital mortality risk of NSTEMI using clinical data. This method mainly consists of three parts. Firstly, oversampling technique was used to alleviate the class imbalance problem. Secondly, the feature selection method of Recursive Feature Elimination (RFE) was selected for effective feature selection. Finally, a unique double-layer stacking model is designed to improve the performance of the algorithm. Seven classical artificial intelligence methods of Logistic Regression (LR), Decision Tree (DT), Support Vector Machine (SVM), Random Forest (RF), Adaptive Boosting (ADB), Extra Tree (ET), and Gradient Boosting Decision Tree (GBDT) were selected as candidate models for the base model of the first layer of the model, and extreme gradient enhancement (XGBOOST) was selected as the meta-model for the second layer.<h4>Results</h4>Patient were divided into the surviving group and the death group, and a total of 57 clinical features showed statistically significant for the two groups and finally included in the subsequent model. The results show that the Area Under Curve (AUC) of the Stacking model proposed in this paper is 0.987, which is higher than that of LR (0.934), DT (0.946), SVM (0.942), RF (0.948), ADB (0.949), ET (0.938) and GBDT (0.920). At the same time, the proposed Stacking model has higher performance than each single model in terms of Accuracy, Precision, Recall and F1 evaluation indicators.<h4>Conclusions</h4>The Stacking model proposed in this paper can integrate the advantages of LR, DT, SVM, RF, ADB, ET and GBDT models to achieve better prediction performance. This model can provide valuable insights for physicians to identify high-risk patients more precisely and timely, thereby maximizing the potential for early clinical interventions to reduce the mortality rate. |
| format | Article |
| id | doaj-art-1880aa53992e4ba98d5cb5464a4249e7 |
| institution | OA Journals |
| issn | 1932-6203 |
| language | English |
| publishDate | 2024-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj-art-1880aa53992e4ba98d5cb5464a4249e72025-08-20T01:48:15ZengPublic Library of Science (PLoS)PLoS ONE1932-62032024-01-011912e031244810.1371/journal.pone.0312448Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model.Li WangYu ZhangFeng LiCaiyun LiHongzeng Xu<h4>Background</h4>Acute myocardial infarction (AMI) remains a leading cause of hospitalization and death in China. Accurate mortality prediction of inpatient is crucial for clinical decision-making of non-ST-segment elevation myocardial infarction (NSTEMI) patients.<h4>Methods</h4>In this study, a total of 3061 patients between January 1, 2017 and December 31, 2022 diagnosed with NSTEMI were enrolled in this study. A new method based on Stacking ensemble model is proposed to predict the in-hospital mortality risk of NSTEMI using clinical data. This method mainly consists of three parts. Firstly, oversampling technique was used to alleviate the class imbalance problem. Secondly, the feature selection method of Recursive Feature Elimination (RFE) was selected for effective feature selection. Finally, a unique double-layer stacking model is designed to improve the performance of the algorithm. Seven classical artificial intelligence methods of Logistic Regression (LR), Decision Tree (DT), Support Vector Machine (SVM), Random Forest (RF), Adaptive Boosting (ADB), Extra Tree (ET), and Gradient Boosting Decision Tree (GBDT) were selected as candidate models for the base model of the first layer of the model, and extreme gradient enhancement (XGBOOST) was selected as the meta-model for the second layer.<h4>Results</h4>Patient were divided into the surviving group and the death group, and a total of 57 clinical features showed statistically significant for the two groups and finally included in the subsequent model. The results show that the Area Under Curve (AUC) of the Stacking model proposed in this paper is 0.987, which is higher than that of LR (0.934), DT (0.946), SVM (0.942), RF (0.948), ADB (0.949), ET (0.938) and GBDT (0.920). At the same time, the proposed Stacking model has higher performance than each single model in terms of Accuracy, Precision, Recall and F1 evaluation indicators.<h4>Conclusions</h4>The Stacking model proposed in this paper can integrate the advantages of LR, DT, SVM, RF, ADB, ET and GBDT models to achieve better prediction performance. This model can provide valuable insights for physicians to identify high-risk patients more precisely and timely, thereby maximizing the potential for early clinical interventions to reduce the mortality rate.https://doi.org/10.1371/journal.pone.0312448 |
| spellingShingle | Li Wang Yu Zhang Feng Li Caiyun Li Hongzeng Xu Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model. PLoS ONE |
| title | Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model. |
| title_full | Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model. |
| title_fullStr | Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model. |
| title_full_unstemmed | Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model. |
| title_short | Mortality prediction of inpatients with NSTEMI in a premier hospital in China based on stacking model. |
| title_sort | mortality prediction of inpatients with nstemi in a premier hospital in china based on stacking model |
| url | https://doi.org/10.1371/journal.pone.0312448 |
| work_keys_str_mv | AT liwang mortalitypredictionofinpatientswithnstemiinapremierhospitalinchinabasedonstackingmodel AT yuzhang mortalitypredictionofinpatientswithnstemiinapremierhospitalinchinabasedonstackingmodel AT fengli mortalitypredictionofinpatientswithnstemiinapremierhospitalinchinabasedonstackingmodel AT caiyunli mortalitypredictionofinpatientswithnstemiinapremierhospitalinchinabasedonstackingmodel AT hongzengxu mortalitypredictionofinpatientswithnstemiinapremierhospitalinchinabasedonstackingmodel |