Predictive value of machine learning model for the prognosis of patients with hydrocephalus after craniocerebral injury treated by ventriculoperitoneal shunt
谷川弘美,臧迪,王哲,齐曾鑫,胡锦,虞剑,孙一睿,朱侗明,吴雪海
DOI: https://doi.org/10.3760/cma.j.cn112050-20210705-00323
2021-01-01
Abstract:Objective:To explore the value of machine learning model in predicting the prognosis of patients with hydrocephalus after craniocerebral injury treated by ventriculoperitoneal shunt(VPS).Methods:A retrospective analysis was conducted on the clinical data of 94 patients who underwent VPS due to hydrocephalus after craniocerebral injury admitted to the Department of Neurosurgery of Huashan Hospital, Fudan University from January 2013 to July 2019. The last Glasgow coma score (GCS) at discharge after shunting was compared to preoperative GCS to determine the progressive group and the non-progressive group. Traditional statistical methods were used to compare the age, gender, length of hospital stay, intracranial infection, posterior fossa lesion, ventricular hemorrhage, skull defect, ventricular pressure and types of hydrocephalus (low pressure hydrocephalus and normal pressure hydrocephalus) between the two groups. Python 3.6 and scikit-learn 0.23 toolkit were used for machine learning analysis, logistic regression algorithm was used as the basic algorithm for machine learning analysis, and randomly split training and testing were performed. In this series, 70% of the data (65 cases) were the training set, and the remaining 30% of the data (29 cases) were the test set. We then analyzed the predictive effect of the above risk factors on the prognosis of patients after shunt.Results:Traditional statistical results showed that compared with the patients in non-progressive group, those in progressive group were older and had a lower GCS score before VPS, and the differences were statistically significant (both
P <0.05); while there was no significant difference in the incidence of intracranial infection, posterior fossa lesion, intraventricular hemorrhage, skull defect, length of hospital stay, ventricular pressure, or presence of low-pressure hydrocephalus between the two groups (all
P> 0.05). The prediction results of the machine learning model showed that the accuracy rate of predicting clinical progress after VPS in the training set was 67.4±20.0%, and the AUC was 0.69±0.18, and those in the test sets were 62.1% and 0.71 respectively. Compared with clinical diagnosis, the overall accuracy rate of predicting the poor prognosis of hydrocephalus after brain injury based on machine learning risk factors was 68.1% (64/94), the sensitivity was 71.4% (30/42), the specificity was 65.4% (34/52), and AUC was 0.89(
χ2 = 12.600,
P<0.001). The weight analysis of the machine learning prediction model showed that low-pressure hydrocephalus accounted for 50.8%, pre-shunt GCS score accounted for 22.7%, intracranial infection accounted for 13.2%, and skull defect accounted for 9.7%.
Conclusions:Compared with traditional statistical methods, the prediction model based on machine learning can capture clinical indicators with smaller statistical effects, and has greater predictive value for the poor prognosis of hydrocephalus after brain injury.