Deakin University
Browse

File(s) not publicly available

Prognosis prediction in traumatic brain injury patients using machine learning algorithms

Version 2 2024-06-05, 05:54
Version 1 2023-02-16, 04:04
journal contribution
posted on 2024-06-05, 05:54 authored by H Khalili, M Rismani, MA Nematollahi, MS Masoudi, A Asadollahi, R Taheri, H Pourmontaseri, A Valibeygi, M Roshanzamir, Roohallah AlizadehsaniRoohallah Alizadehsani, A Niakan, A Andishgar, Shariful IslamShariful Islam, UR Acharya
AbstractPredicting treatment outcomes in traumatic brain injury (TBI) patients is challenging worldwide. The present study aimed to achieve the most accurate machine learning (ML) algorithms to predict the outcomes of TBI treatment by evaluating demographic features, laboratory data, imaging indices, and clinical features. We used data from 3347 patients admitted to a tertiary trauma centre in Iran from 2016 to 2021. After the exclusion of incomplete data, 1653 patients remained. We used ML algorithms such as random forest (RF) and decision tree (DT) with ten-fold cross-validation to develop the best prediction model. Our findings reveal that among different variables included in this study, the motor component of the Glasgow coma scale, the condition of pupils, and the condition of cisterns were the most reliable features for predicting in-hospital mortality, while the patients’ age takes the place of cisterns condition when considering the long-term survival of TBI patients. Also, we found that the RF algorithm is the best model to predict the short-term mortality of TBI patients. However, the generalized linear model (GLM) algorithm showed the best performance (with an accuracy rate of 82.03 ± 2.34) in predicting the long-term survival of patients. Our results showed that using appropriate markers and with further development, ML has the potential to predict TBI patients’ survival in the short- and long-term.

History

Journal

Scientific reports

Volume

13

Article number

960

Pagination

960-

Location

England

ISSN

2045-2322

eISSN

2045-2322

Language

en

Publication classification

C1 Refereed article in a scholarly journal

Issue

1

Publisher

Springer Science and Business Media LLC

Usage metrics

    Research Publications

    Categories

    No categories selected

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC