Machine learning based prediction of chronic shunt-dependent hydrocephalus after spontaneous subarachnoid hemorrhage.

Background: Chronic posthemorrhagic hydrocephalus often arises following spontaneous subarachnoid hemorrhage (SAH). Timely identification of patients predisposed to develop chronic shunt-dependent hydrocephalus may significantly enhance clinical outcomes.

Methods: We performed an analysis of 510 SAH-patients treated at our institution between 2013 and 2018. Clinical and radiological variables, including age, sex, Hunt & Hess grade, Fisher-Score, external ventricular drainage placement, central nervous system infection, aneurysm characteristics, and treatment modalities, were evaluated. Supervised machine learning models, trained and compared using Python and scikit-learn, were employed to predict chronic shunt-dependent hydrocephalus. Model performance was rigorously assessed through repeated cross-validation. To facilitate transparency and collaboration, we publicly released the dataset and code on GitHub (https://github.com/RISCSoftware/shuntclf) and developed an interactive web application (https://huggingface.co/spaces/risc42/shuntclf).

Results: Among the evaluated machine learning models, logistic regression exhibited superior performance, with an AUC-ROC of 0.819 and an AUC-PR of 0.482, along with the highest F1 score of 0.473. Although the balanced accuracy scores of the models were generally proximate, ranging from 0.735 to 0.764, logistic regression consistently outperform others in key metrics such as AUC-ROC and AUC-PR. Conversely, female gender and absence of aneurysm within the anterior communicating artery were associated with reduced shunt requirement likelihood.

Conclusion: Machine learning models, including logistic regression, demonstrate strong predictive capability for early chronic shunt-dependent hydrocephalus following spontaneous SAH, which may potentially contribute to more timely shunt placement interventions. This predictive capability is supported by our web interface, which simplifies the application of these models, aiding clinicians in efficiently determining the need for shunt placement.