Bioengineering (Jun 2024)
Integrating PointNet-Based Model and Machine Learning Algorithms for Classification of Rupture Status of IAs
Abstract
Background: The rupture of intracranial aneurysms (IAs) would result in subarachnoid hemorrhage with high mortality and disability. Predicting the risk of IAs rupture remains a challenge. Methods: This paper proposed an effective method for classifying IAs rupture status by integrating a PointNet-based model and machine learning algorithms. First, medical image segmentation and reconstruction algorithms were applied to 3D Digital Subtraction Angiography (DSA) imaging data to construct three-dimensional IAs geometric models. Geometrical parameters of IAs were then acquired using Geomagic, followed by the computation of hemodynamic clouds and hemodynamic parameters using Computational Fluid Dynamics (CFD). A PointNet-based model was developed to extract different dimensional hemodynamic cloud features. Finally, five types of machine learning algorithms were applied on geometrical parameters, hemodynamic parameters, and hemodynamic cloud features to classify and recognize IAs rupture status. The classification performance of different dimensional hemodynamic cloud features was also compared. Results: The 16-, 32-, 64-, and 1024-dimensional hemodynamic cloud features were extracted with the PointNet-based model, respectively, and the four types of cloud features in combination with the geometrical parameters and hemodynamic parameters were respectively applied to classify the rupture status of IAs. The best classification outcomes were achieved in the case of 16-dimensional hemodynamic cloud features, the accuracy of XGBoost, CatBoost, SVM, LightGBM, and LR algorithms was 0.887, 0.857, 0.854, 0.857, and 0.908, respectively, and the AUCs were 0.917, 0.934, 0.946, 0.920, and 0.944. In contrast, when only utilizing geometrical parameters and hemodynamic parameters, the accuracies were 0.836, 0.816, 0.826, 0.832, and 0.885, respectively, with AUC values of 0.908, 0.922, 0.930, 0.884, and 0.921. Conclusion: In this paper, classification models for IAs rupture status were constructed by integrating a PointNet-based model and machine learning algorithms. Experiments demonstrated that hemodynamic cloud features had a certain contribution weight to the classification of IAs rupture status. When 16-dimensional hemodynamic cloud features were added to the morphological and hemodynamic features, the models achieved the highest classification accuracies and AUCs. Our models and algorithms would provide valuable insights for the clinical diagnosis and treatment of IAs.
Keywords