A Machine Learning-Based Approach to Predict Prognosis and Length of Hospital Stay in Adults and Children With Traumatic Brain Injury: Retrospective Cohort Study

被引:13
|
作者
Fang, Cheng [1 ]
Pan, Yifeng [2 ]
Zhao, Luotong [1 ]
Niu, Zhaoyi [1 ]
Guo, Qingguo [1 ]
Zhao, Bing [1 ]
机构
[1] Anhui Med Univ, Affiliated Hosp 2, Dept Neurosurg, 678 Furong Raod, Hefei 230601, Peoples R China
[2] Anhui Xinhua Univ, Sch Big Data & Artificial Intelligence, Hefei, Peoples R China
关键词
convolutional neural network; machine learning; neurosurgery; support vector machine; support vector regression; traumatic brain injury; ARTIFICIAL NEURAL-NETWORKS; CNN; NEUTROPHIL; RATIO; CARE;
D O I
10.2196/41819
中图分类号
R19 [保健组织与事业(卫生事业管理)];
学科分类号
摘要
Background: The treatment and care of adults and children with traumatic brain injury (TBI) constitute an intractable global health problem. Predicting the prognosis and length of hospital stay of patients with TBI may improve therapeutic effects and significantly reduce societal health care burden. Applying novel machine learning methods to the field of TBI may be valuable for determining the prognosis and cost-effectiveness of clinical treatment. Objective: We aimed to combine multiple machine learning approaches to build hybrid models for predicting the prognosis and length of hospital stay for adults and children with TBI. Methods: We collected relevant clinical information from patients treated at the Neurosurgery Center of the Second Affiliated Hospital of Anhui Medical University between May 2017 and May 2022, of which 80% was used for training the model and 20% for testing via screening and data splitting. We trained and tested the machine learning models using 5 cross-validations to avoid overfitting. In the machine learning models, 11 types of independent variables were used as input variables and Glasgow Outcome Scale score, used to evaluate patients' prognosis, and patient length of stay were used as output variables. Once the models were trained, we obtained and compared the errors of each machine learning model from 5 rounds of cross-validation to select the best predictive model. The model was then externally tested using clinical data of patients treated at the First Affiliated Hospital of Anhui Medical University from June 2021 to February 2022. Results: The final convolutional neural network-support vector machine (CNN-SVM) model predicted Glasgow Outcome Scale score with an accuracy of 93% and 93.69% in the test and external validation sets, respectively, and an area under the curve of 94.68% and 94.32% in the test and external validation sets, respectively. The mean absolute percentage error of the final built convolutional neural network-support vector regression (CNN-SVR) model predicting inpatient time in the test set and external validation set was 10.72% and 10.44%, respectively. The coefficient of determination (R2) was 0.93 and 0.92 in the test set and external validation set, respectively. Compared with back-propagation neural network, CNN, and SVM models built separately, our hybrid model was identified to be optimal and had high confidence. Conclusions: This study demonstrates the clinical utility of 2 hybrid models built by combining multiple machine learning approaches to accurately predict the prognosis and length of stay in hospital for adults and children with TBI. Application of these models may reduce the burden on physicians when assessing TBI and assist clinicians in the medical decision-making process.
引用
收藏
页数:13
相关论文
共 50 条
  • [21] PSYCHOSOCIAL FACTORS ASSOCIATED WITH PROLONGED LENGTH OF STAY IN ACQUIRED BRAIN INJURY REHABILITATION: A RETROSPECTIVE COHORT STUDY
    Mcdougall, Alexandre
    Mylabathula, Swapna
    Alavinia, Mohammad
    Kam, Alice
    Bayley, Mark
    Guo, Meiqi
    Tam, Alan
    JOURNAL OF REHABILITATION MEDICINE, 2023, 55
  • [22] Machine Learning-Based Prediction for 4-Year Risk of Metabolic Syndrome in Adults: A Retrospective Cohort Study
    Zhang, Hui
    Chen, Dandan
    Shao, Jing
    Zou, Ping
    Cui, Nianqi
    Tang, Leiwen
    Wang, Xiyi
    Wang, Dan
    Wu, Jingjie
    Ye, Zhihong
    RISK MANAGEMENT AND HEALTHCARE POLICY, 2021, 14 : 4361 - 4368
  • [23] Frailty Predicts in-Hospital Death in Traumatic Brain Injury Patients: A Retrospective Cohort Study
    Liu, Hua
    Wu, Wenxi
    Xu, Min
    Ling, Xiaoyang
    Lu, Wei
    Cheng, Feng
    Wang, Jian
    THERAPEUTICS AND CLINICAL RISK MANAGEMENT, 2024, 20 : 665 - 675
  • [24] The influence of post-acute rehabilitation length of stay on traumatic brain injury outcome: a retrospective exploratory study
    Ashley, Jessica G.
    Ashley, Mark J.
    Masel, Brent E.
    Randle, Kevin
    Kreber, Lisa A.
    Singh, Charan
    Harrington, David
    Griesbach, Grace S.
    BRAIN INJURY, 2018, 32 (05) : 600 - 607
  • [25] Ocular Biometric Components in Hyperopic Children and a Machine Learning-Based Model to Predict Axial Length
    Wang, Jingyun
    Jost, Reed M.
    Birch, Eileen E.
    TRANSLATIONAL VISION SCIENCE & TECHNOLOGY, 2024, 13 (05):
  • [26] Machine learning-based model for predicting inpatient mortality in adults with traumatic brain injury: a systematic review and meta-analysis
    Wu, Zhe
    Lai, Jinqing
    Huang, Qiaomei
    Lin, Long
    Lin, Shu
    Chen, Xiangrong
    Huang, Yinqiong
    FRONTIERS IN NEUROSCIENCE, 2023, 17
  • [27] Machine learning-based classification of chronic traumatic brain injury using hybrid diffusion imaging
    Muller, Jennifer J.
    Wang, Ruixuan
    Milddleton, Devon
    Alizadeh, Mahdi
    Kang, Ki Chang
    Hryczyk, Ryan
    Zabrecky, George
    Hriso, Chloe
    Navarreto, Emily
    Wintering, Nancy
    Bazzan, Anthony J.
    Wu, Chengyuan
    Monti, Daniel A.
    Jiao, Xun
    Wu, Qianhong
    Newberg, Andrew B.
    Mohamed, Feroze B.
    FRONTIERS IN NEUROSCIENCE, 2023, 17
  • [28] Pediatric severe traumatic brain injury mortality prediction determined with machine learning-based modeling
    Daley, Mark
    Cameron, Saoirse
    Ganesan, Saptharishi Lalgudi
    Patel, Maitray A.
    Stewart, Tanya Charyk
    Miller, MichaelR.
    Alharfi, Ibrahim
    Fraser, Douglas D.
    INJURY-INTERNATIONAL JOURNAL OF THE CARE OF THE INJURED, 2022, 53 (03): : 992 - 998
  • [29] A follow-up study of older adults with traumatic brain injury: Taking into account decreasing length of stay
    Frankel, JE
    Marwitz, JH
    Cifu, DX
    Kreutzer, JS
    Englander, J
    Rosenthal, M
    ARCHIVES OF PHYSICAL MEDICINE AND REHABILITATION, 2006, 87 (01): : 57 - 62
  • [30] Effect of Red Cell Distribution Width on the Prognosis of Patients with Traumatic Brain Injury: A Retrospective Cohort Study
    Lin, Bing
    Fu, Zhao-Yin
    Chen, Meng-Hua
    WORLD NEUROSURGERY, 2023, 170 : E744 - E754