An Integrated Model Combining Machine Learning and Deep Learning Algorithms for Classification of Rupture Status of IAs.

BACKGROUND

The rupture risk assessment of intracranial aneurysms (IAs) is clinically relevant. How to accurately assess the rupture risk of IAs remains a challenge in clinical decision-making.

PURPOSE

We aim to build an integrated model to improve the assessment of the rupture risk of IAs.

MATERIALS AND METHODS

A total of 148 (39 ruptured and 109 unruptured) IA subjects were retrospectively computed with computational fluid dynamics (CFDs), and the integrated models were proposed by combining machine learning (ML) and deep learning (DL) algorithms. ML algorithms that include random forest (RF), k-nearest neighbor (KNN), XGBoost (XGB), support vector machine (SVM), and LightGBM were, respectively, adopted to classify ruptured and unruptured IAs. A Pointnet DL algorithm was applied to extract hemodynamic cloud features from the hemodynamic clouds obtained from CFD. Morphological variables and hemodynamic parameters along with the extracted hemodynamic cloud features were acted as the inputs to the classification models. The classification results with and without hemodynamic cloud features are computed and compared.

RESULTS

Without consideration of hemodynamic cloud features, the classification accuracy of RF, KNN, XGB, SVM, and LightGBM was 0.824, 0.759, 0.839, 0.860, and 0.829, respectively, and the AUCs of them were 0.897, 0.584, 0.892, 0.925, and 0.890, respectively. With the consideration of hemodynamic cloud features, the accuracy successively increased to 0.908, 0.873, 0.900, 0.926, and 0.917. Meanwhile, the AUCs reached 0.952, 0.881, 0.950, 0.969, and 0.965 eventually. Adding consideration of hemodynamic cloud features, the SVM could perform best with the highest accuracy of 0.926 and AUC of 0.969, respectively.

CONCLUSION

The integrated model combining ML and DL algorithms could improve the classification of IAs. Adding consideration of hemodynamic cloud features could bring more accurate classification, and hemodynamic cloud features were important for the discrimination of ruptured IAs.