Using a Machine Learning Approach to Predict Outcomes after Radiosurgery for Cerebral Arteriovenous Malformations

Abstract

Predictions of patient outcomes after a given therapy are fundamental to medical practice. We employ a machine learning approach towards predicting the outcomes after stereotactic radiosurgery for cerebral arteriovenous malformations (AVMs). Using three prospective databases, a machine learning approach of feature engineering and model optimization was implemented to create the most accurate predictor of AVM outcomes. Existing prognostic systems were scored for purposes of comparison. The final predictor was secondarily validated on an independent site’s dataset not utilized for initial construction. Out of 1,810 patients, 1,674 to 1,291 patients depending upon time threshold, with 23 features were included for analysis and divided into training and validation sets. The best predictor had an average area under the curve (AUC) of 0.71 compared to existing clinical systems of 0.63 across all time points. On the heldout dataset, the predictor had an accuracy of around 0.74 at across all time thresholds with a specificity and sensitivity of 62% and 85% respectively. This machine learning approach was able to provide the best possible predictions of AVM radiosurgery outcomes of any method to date, identify a novel radiobiological feature (3D surface dose), and demonstrate a paradigm for further development of prognostic tools in medical care.

A systematic review on machine learning in neurosurgery: the future of decision making in patient care.

“…Three of them were epilepsy surgery radiosurgery for brain metastases (10,29,43). Oermann et al used ML to predict patient outcomes of AVM radiosurgery (43). Another study implemented ML methods to predict the risk of hydrocephalus following gamma knife radiosurgery for intracranial schwannoma (30).…”

IMRT QA using machine learning: A multi‐institutional validation

“…Machine learning is a subfield of data science that focuses on designing algorithms that can learn from and make predictions on data. Machine learning applications in radiotherapy have emerged increasingly in recent years, with applications including predictive modeling of treatment outcome in radiation oncology,1, 2, 3, 4, 5, 6, 7 treatment optimization,8, 9, 10, 11 error detection and prevention,12, 13, 14, 15 and treatment machine quality assurance (QA) 16, 17, 18, 19. These machine learning techniques have provided physicians and physicists information for more effective and accurate treatment delivery as well as the ability to achieve personalized treatment.…”

DW-MRI and DCE-MRI are of complementary value in predicting pathologic response to neoadjuvant chemoradiotherapy for esophageal cancer

“…To further optimize the prediction model, the addition of other modalities could be examined, as well as the use of other models (e.g. machine-learning methods) [32,33].…”