UM  > Faculty of Health Sciences
Residential Collegefalse
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 PublicationHeliyon
ISSN2405-8440
Volume10Issue: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.

KeywordDnn Knn Lr Machine Learning Rf Survival Trauma
DOI10.1016/j.heliyon.2024.e30198
URLView the original
Indexed BySCIE
Language英語English
WOS Research AreaScience & Technology - Other Topics
WOS SubjectMultidisciplinary Sciences
WOS IDWOS:001236365600001
PublisherELSEVIER SCI LTD, THE BOULEVARD, LANGFORD LANE, KIDLINGTON, OXFORD OX5 1GB, OXON, ENGLAND
Scopus ID2-s2.0-85191311074
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionFaculty of Health Sciences
Centre for Precision Medicine Research and Training
Co-First AuthorYang, Yang
Corresponding AuthorWang, Ziheng; Gong, Peipei; Chen, Tianxi
Affiliation1.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 AffilicationFaculty 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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Yang, Yang]'s Articles
[Zhou, Liulei]'s Articles
[Luo, Jinhua]'s Articles
Baidu academic
Similar articles in Baidu academic
[Yang, Yang]'s Articles
[Zhou, Liulei]'s Articles
[Luo, Jinhua]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Yang, Yang]'s Articles
[Zhou, Liulei]'s Articles
[Luo, Jinhua]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.