The Performance of Different Machine Learning Algorithm and Regression Models in Predicting High-Grade Intracranial Meningioma

Abstract

Meningioma is the most common primary tumor of the central nervous system (CNS). Individualized treatment strategies should be formulated for the patients according to the WHO (World Health Organization) grade. Our aim was to investigate the effectiveness of various machine learning and traditional statistical models in predicting the WHO grade of preoperative patients with meningioma. Patients diagnosed with meningioma after surgery in West China Hospital and Shangjin Hospital of Sichuan University from 2009 to 2016 were included in the study cohort. As the training cohort (n = 1975), independent risk factors associated with high-grade meningioma were used to establish the Nomogram model. which was validated in a subsequent cohort (n = 1048) from 2017 to 2019 in our hospital. Logistic regression (LR), XGboost, Adaboost, Support Vector Machine (SVM), K-Nearest Neighbor (KNN), and Random Forest (RF) models were determined using F1 score, recall, accuracy, the area under the curve (ROC), calibration plot and decision curve analysis (DCA) were used to evaluate the different models. Logistic regression showed better predictive performance and interpretability than machine learning. Gender, recurrence history, T1 signal intensity, enhanced signal degree, peritumoral edema, tumor diameter, cystic, location, and NLR index were identified as independent risk factors and added to the nomogram. The AUC (Area Under Curve) value of RF was 0.812 in the training set, 0.807 in the internal validation set, and 0.842 in the external validation set. The calibration curve and DCA (Decision Curve Analysis) indicated that it had better prediction efficiency of LR than others. The Nomogram preoperative prediction model of meningioma of WHO II and III grades showed effective prediction ability. While machine learning exhibits strong fitting ability, it performs poorly in the validation set.

Keywords

• high-grade meningioma
• nomogram
• logistic regression
• machine learning