Machine Learning-Enabled Determination of Diffuseness of Brain Arteriovenous Malformations from Magnetic Resonance Angiography

Abstract

The diffuseness of brain arteriovenous malformations (bAVMs) is a significant factor in surgical outcome evaluation and hemorrhagic risk prediction. However, there are still predicaments in identifying diffuseness, such as the judging variety resulting from different experience and difficulties in quantification. The purpose of this study was to develop a machine learning (ML) model to automatically identify the diffuseness of bAVM niduses using three-dimensional (3D) time-of-flight magnetic resonance angiography (TOF-MRA) images. A total of 635 patients with bAVMs who underwent TOF-MRA imaging were enrolled. Three experienced neuroradiologists delineated the bAVM lesions and identified the diffuseness on TOF-MRA images, which were considered the ground-truth reference. The U-Net-based segmentation model was trained to segment lesion areas. Eight mainstream ML models were trained through the radiomic features of segmented lesions to identify diffuseness, based on which an integrated model was built and yielded the best performance. In the test set, the Dice score, F2 score, precision, and recall for the segmentation model were 0.80 [0.72–0.84], 0.80 [0.71–0.86], 0.84 [0.77–0.93], and 0.82 [0.69–0.89], respectively. For the diffuseness identification model, the ensemble-based model was applied with an area under the Receiver-operating characteristic curves (AUC) of 0.93 (95% CI 0.87–0.99) in the training set. The AUC, accuracy, precision, recall, and F1 score for the diffuseness identification model were 0.95, 0.90, 0.81, 0.84, and 0.83, respectively, in the test set. The ML models showed good performance in automatically detecting bAVM lesions and identifying diffuseness. The method may help to judge the diffuseness of bAVMs objectively, quantificationally, and efficiently.