Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury

ObjectiveThis study aimed to establish a prediction model for the short-term prognosis of children with traumatic brain injury (TBI) using machine learning algorithms.MethodsThe clinical data of children with TBI who were treated in the First Affiliated Hospital of Zhengzhou University were retrospe...

Full description

Saved in:
Bibliographic Details
Main Authors: Yongwei Wei, Jiandong Wang, Yu Su, Fan Zhou, Huaili Wang
Format: Article
Language:English
Published: Frontiers Media S.A. 2025-05-01
Series:Frontiers in Pediatrics
Subjects:
Online Access:https://www.frontiersin.org/articles/10.3389/fped.2025.1581945/full
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1850271591851622400
author Yongwei Wei
Jiandong Wang
Yu Su
Fan Zhou
Huaili Wang
author_facet Yongwei Wei
Jiandong Wang
Yu Su
Fan Zhou
Huaili Wang
author_sort Yongwei Wei
collection DOAJ
description ObjectiveThis study aimed to establish a prediction model for the short-term prognosis of children with traumatic brain injury (TBI) using machine learning algorithms.MethodsThe clinical data of children with TBI who were treated in the First Affiliated Hospital of Zhengzhou University were retrospectively analyzed. All children were divided into a modeling group and a validation group. In the laboratory indicators of the modeling group, the least absolute shrinkage and selection operator (LASSO) and multivariate Logistic regression analysis were used to screen out the independent influencing factors of poor prognosis in TBI, and a laboratory indicator model (LIM) was established. The risk scores of all patients were calculated. Then, the risk scores and other indicators were used to construct an extended prediction model through the extreme gradient boosting (XGBoost) algorithm. The discrimination, calibration, and clinical utility of the model were evaluated, and the extended model was explained using SHAP analysis. Finally, a subgroup analysis was performed using the risk scores to assess the robustness of the laboratory indicator model.ResultsAmong the laboratory indicators, lactate dehydrogenase (LDH), N-terminal pro-B-type natriuretic peptide (NT-proBNP), hydrogen ion concentration index (pH), hemoglobin (Hb), serum albumin (Alb), and C-reactive protein to albumin ratio (CRP/Alb) were the independent influencing factors for the prognosis of children with brain injury. The extended model demonstrated excellent predictive performance in both the modeling and validation populations. SHAP analysis showed the contribution values of the Glasgow Coma Scale (GCS), the laboratory indicator model, the location of the head hematoma, the pupillary light reflex, and the injury severity score in the prediction of the overall patient prognosis. The subgroup analysis showed that there were differences in the risk scores of children with different GCS scores, pupillary light reflexes, and head hematoma locations, and there were also differences in the prognosis between the high-risk score group and the low-risk score group within them.ConclusionThe extended model can accurately predict the prognosis of TBI patients and has strong clinical utility. The core model has good stratification ability and provides an effective risk stratification and personalized patient management tool for clinicians.
format Article
id doaj-art-18fa617df83643c4abd084a5ca33f49a
institution OA Journals
issn 2296-2360
language English
publishDate 2025-05-01
publisher Frontiers Media S.A.
record_format Article
series Frontiers in Pediatrics
spelling doaj-art-18fa617df83643c4abd084a5ca33f49a2025-08-20T01:52:11ZengFrontiers Media S.A.Frontiers in Pediatrics2296-23602025-05-011310.3389/fped.2025.15819451581945Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injuryYongwei WeiJiandong WangYu SuFan ZhouHuaili WangObjectiveThis study aimed to establish a prediction model for the short-term prognosis of children with traumatic brain injury (TBI) using machine learning algorithms.MethodsThe clinical data of children with TBI who were treated in the First Affiliated Hospital of Zhengzhou University were retrospectively analyzed. All children were divided into a modeling group and a validation group. In the laboratory indicators of the modeling group, the least absolute shrinkage and selection operator (LASSO) and multivariate Logistic regression analysis were used to screen out the independent influencing factors of poor prognosis in TBI, and a laboratory indicator model (LIM) was established. The risk scores of all patients were calculated. Then, the risk scores and other indicators were used to construct an extended prediction model through the extreme gradient boosting (XGBoost) algorithm. The discrimination, calibration, and clinical utility of the model were evaluated, and the extended model was explained using SHAP analysis. Finally, a subgroup analysis was performed using the risk scores to assess the robustness of the laboratory indicator model.ResultsAmong the laboratory indicators, lactate dehydrogenase (LDH), N-terminal pro-B-type natriuretic peptide (NT-proBNP), hydrogen ion concentration index (pH), hemoglobin (Hb), serum albumin (Alb), and C-reactive protein to albumin ratio (CRP/Alb) were the independent influencing factors for the prognosis of children with brain injury. The extended model demonstrated excellent predictive performance in both the modeling and validation populations. SHAP analysis showed the contribution values of the Glasgow Coma Scale (GCS), the laboratory indicator model, the location of the head hematoma, the pupillary light reflex, and the injury severity score in the prediction of the overall patient prognosis. The subgroup analysis showed that there were differences in the risk scores of children with different GCS scores, pupillary light reflexes, and head hematoma locations, and there were also differences in the prognosis between the high-risk score group and the low-risk score group within them.ConclusionThe extended model can accurately predict the prognosis of TBI patients and has strong clinical utility. The core model has good stratification ability and provides an effective risk stratification and personalized patient management tool for clinicians.https://www.frontiersin.org/articles/10.3389/fped.2025.1581945/fulltraumatic brain injurymachine learningprediction modelsubgroup analysischildren
spellingShingle Yongwei Wei
Jiandong Wang
Yu Su
Fan Zhou
Huaili Wang
Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury
Frontiers in Pediatrics
traumatic brain injury
machine learning
prediction model
subgroup analysis
children
title Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury
title_full Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury
title_fullStr Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury
title_full_unstemmed Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury
title_short Construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury
title_sort construction and validation of a machine learning based prognostic prediction model for children with traumatic brain injury
topic traumatic brain injury
machine learning
prediction model
subgroup analysis
children
url https://www.frontiersin.org/articles/10.3389/fped.2025.1581945/full
work_keys_str_mv AT yongweiwei constructionandvalidationofamachinelearningbasedprognosticpredictionmodelforchildrenwithtraumaticbraininjury
AT jiandongwang constructionandvalidationofamachinelearningbasedprognosticpredictionmodelforchildrenwithtraumaticbraininjury
AT yusu constructionandvalidationofamachinelearningbasedprognosticpredictionmodelforchildrenwithtraumaticbraininjury
AT fanzhou constructionandvalidationofamachinelearningbasedprognosticpredictionmodelforchildrenwithtraumaticbraininjury
AT huailiwang constructionandvalidationofamachinelearningbasedprognosticpredictionmodelforchildrenwithtraumaticbraininjury