This study aimed to develop a machine learning model for predicting brain arteriovenous malformation (bAVM) rupture using a combination of traditional risk factors and radiomics features. This multicenter retrospective study enrolled 586 patients with unruptured bAVMs from 2010 to 2020. All patients were grouped into the hemorrhage (n = 368) and non-hemorrhage (n = 218) groups. The bAVM nidus were segmented on CT angiography images using Slicer software, and radiomic features were extracted using Pyradiomics. The dataset included a training set and an independent testing set. The machine learning model was developed on the training set and validated on the testing set by merging numerous base estimators and a final estimator based on the stacking method. The area under the receiver operating characteristic (ROC) curve, precision, and the f1 score were evaluated to determine the performance of the model. A total of 1790 radiomics features and 8 traditional risk factors were contained in the original dataset, and 241 features remained for model training after L1 regularization filtering. The base estimator of the ensemble model was Logistic Regression, whereas the final estimator was Random Forest. In the training set, the area under the ROC curve of the model was 0.982 (0.967-0.996) and 0.893 (0.826-0.960) in the testing set. This study indicated that radiomics features are a valuable addition to traditional risk factors for predicting bAVM rupture. In the meantime, ensemble learning can effectively improve the performance of a prediction model.
Keywords: Brain arteriovenous malformation; Ensemble learning method; Machine learning; Prediction model; Radiomics.
© 2023. The Author(s), under exclusive licence to Springer Science+Business Media, LLC, part of Springer Nature.