Predicting leptomeningeal disease spread after resection of brain metastases using machine learning

Abstract

OBJECTIVE The incidence of leptomeningeal disease (LMD) has increased as treatments for brain metastases (BMs) have improved and patients with metastatic disease are living longer. Sample sizes of individual studies investigating LMD after surgery for BMs and its risk factors have been limited, ranging from 200 to 400 patients at risk for LMD, which only allows the use of conventional biostatistics. Here, the authors used machine learning techniques to enhance LMD prediction in a cohort of surgically treated BMs. METHODS A conditional survival forest, a Cox proportional hazards model, an extreme gradient boosting (XGBoost) classifier, an extra trees classifier, and logistic regression were trained. A synthetic minority oversampling technique (SMOTE) was used to train the models and handle the inherent class imbalance. Patients were divided into an 80:20 training and test set. Fivefold cross-validation was used on the training set for hyperparameter optimization. Patients eligible for study inclusion were adults who had consecutively undergone neurosurgical BM treatment, had been admitted to Brigham and Women’s Hospital from January 2007 through December 2019, and had a minimum of 1 month of follow-up after neurosurgical treatment. RESULTS A total of 1054 surgically treated BM patients were included in this analysis. LMD occurred in 168 patients (15.9%) at a median of 7.05 months after BM diagnosis. The discrimination of LMD occurrence was optimal using an XGboost algorithm (area under the curve = 0.83), and the time to LMD was prognosticated evenly by the random forest algorithm and the Cox proportional hazards model (C-index = 0.76). The most important feature for both LMD classification and regression was the BM proximity to the CSF space, followed by a cerebellar BM location. Lymph node metastasis of the primary tumor at BM diagnosis and a cerebellar BM location were the strongest risk factors for both LMD occurrence and time to LMD. CONCLUSIONS The outcomes of LMD patients in the BM population are predictable using SMOTE and machine learning. Lymph node metastasis of the primary tumor at BM diagnosis and a cerebellar BM location were the strongest LMD risk factors.