Residential College | false |
Status | 已發表Published |
Prediction analysis of TBI 24-h survival outcome based on machine learning | |
Yang, Yang1,2; Zhou, Liulei1; Luo, Jinhua3; Xue, Jianhua1; Liu, Jiajia1; Zhang, Jiajia4; Wang, Ziheng4,5,6,7,8; Gong, Peipei4; Chen, Tianxi9 | |
2024-05-15 | |
Source Publication | Heliyon |
ISSN | 2405-8440 |
Volume | 10Issue:9Pages:e30198 |
Abstract | Background: Traumatic brain injury (TBI) is the major reason for the death of young people and is well known for its high mortality and morbidity. This paper aim to predict the 24h survival of patients with TBI. Methods: A total of 1224 samples were involved in this analysis, and the clinical indicators involved included age, gender, blood pressure, MGAP and other fields, among which the target variable was “outcome”, which was a binary variable. The methods mainly involved in this paper include data visualization analysis, single factor analysis, feature engineering analysis, random forest model (RF), K-Nearst Neighbors (KNN) model, and so on. Logistic regression model (LR) and deep neural network model (DNN). We will oversample the training set using the SMOTE method because of the very unbalanced labeling of the sample itself. Results: Although the accuracy of all models is very high, the recall rate is relatively low. The DNN model with the best performance only reaches 0.17, and the corresponding AUC is 0.80. After resampling, we find that the recall rate of positive samples of all models has increased a lot, but the AUC of some models has decreased. Finally, the optimal model is LR, whose positive sample recall rate is 0.67 and AUC is 0.82. Conclusion: Through resampling, we obtained that the best model is the RF model, whose recall rate and AUC are the best, and the AUC level is about 0.87, indicating that the accuracy performance of the model is still good. |
Keyword | Dnn Knn Lr Machine Learning Rf Survival Trauma |
DOI | 10.1016/j.heliyon.2024.e30198 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Science & Technology - Other Topics |
WOS Subject | Multidisciplinary Sciences |
WOS ID | WOS:001236365600001 |
Publisher | ELSEVIER SCI LTD, THE BOULEVARD, LANGFORD LANE, KIDLINGTON, OXFORD OX5 1GB, OXON, ENGLAND |
Scopus ID | 2-s2.0-85191311074 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Health Sciences Centre for Precision Medicine Research and Training |
Co-First Author | Yang, Yang |
Corresponding Author | Wang, Ziheng; Gong, Peipei; Chen, Tianxi |
Affiliation | 1.Department of Trauma Center, Affiliated Hospital of Nantong University, Nantong City, No.20 Xisi Road, Chongchuan District, Jiangsu Province, 226001, China 2.Department of Chemistry, School of Science, China Pharmaceutical University, Nanjing, 211198, China 3.Department of Anesthesia Surgery, Affiliated Hospital of Nantong University, Nantong City, No.20 Xisi Road, Chongchuan District, Jiangsu Province, 226001, China 4.Department of Neurosurgery, Affiliated Hospital of Nantong University, Nantong City, No.20 Xisi Road, Chongchuan District, Jiangsu Province, 226001, China 5.Clinical and Translational Research Center, Affiliated Hospital of Nantong University, Nantong City, No.20 Xisi Road, Chongchuan District, Jiangsu Province, 226001, China 6.Suzhou Industrial Park Monash Research Institute of Science and Technology, Suzhou, China 7.The School of Public Health and Preventive Medicine, Monash University, Melbourne, Australia 8.Centre for Precision Medicine Research and Training, Faculty of Health Sciences, University of Macau, Macau, China 9.Department of Emergency Medicine, Affiliated Hospital of Nantong University, Nantong City, No.20 Xisi Road, Chongchuan District, Jiangsu Province, 226001, China |
Corresponding Author Affilication | Faculty of Health Sciences |
Recommended Citation GB/T 7714 | Yang, Yang,Zhou, Liulei,Luo, Jinhua,et al. Prediction analysis of TBI 24-h survival outcome based on machine learning[J]. Heliyon, 2024, 10(9), e30198. |
APA | Yang, Yang., Zhou, Liulei., Luo, Jinhua., Xue, Jianhua., Liu, Jiajia., Zhang, Jiajia., Wang, Ziheng., Gong, Peipei., & Chen, Tianxi (2024). Prediction analysis of TBI 24-h survival outcome based on machine learning. Heliyon, 10(9), e30198. |
MLA | Yang, Yang,et al."Prediction analysis of TBI 24-h survival outcome based on machine learning".Heliyon 10.9(2024):e30198. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment