Artificial neural networks predict the need for permanent cerebrospinal fluid diversion following posterior fossa tumor resection

Neurooncol Adv. 2022 Sep 13;4(1):vdac145. doi: 10.1093/noajnl/vdac145. eCollection 2022 Jan-Dec.

Abstract

Background: Resection of posterior fossa tumors (PFTs) can result in hydrocephalus that requires permanent cerebrospinal fluid (CSF) diversion. Our goal was to prospectively validate a machine-learning model to predict postoperative hydrocephalus after PFT surgery requiring permanent CSF diversion.

Methods: We collected preoperative and postoperative variables on 518 patients that underwent PFT surgery at our center in a retrospective fashion to train several statistical classifiers to predict the need for permanent CSF diversion as a binary class. A total of 62 classifiers relevant to our data structure were surveyed, including regression models, decision trees, Bayesian models, and multilayer perceptron artificial neural networks (ANN). Models were trained using the (N = 518) retrospective data using 10-fold cross-validation to obtain accuracy metrics. Given the low incidence of our positive outcome (12%), we used the positive predictive value along with the area under the receiver operating characteristic curve (AUC) to compare models. The best performing model was then prospectively validated on a set of 90 patients.

Results: Twelve percent of patients required permanent CSF diversion after PFT surgery. Of the trained models, 8 classifiers had an AUC greater than 0.5 on prospective testing. ANNs demonstrated the highest AUC of 0.902 with a positive predictive value of 83.3%. Despite comparable AUC, the remaining classifiers had a true positive rate below 35% (compared to ANN, P < .0001). The negative predictive value of the ANN model was 98.8%.

Conclusions: ANN-based models can reliably predict the need for ventriculoperitoneal shunt after PFT surgery.

Keywords: artificial neural networks; machine learning; metastasis; posterior fossa tumor; ventriculoperitoneal shunt.