Stroke risk prediction by color Doppler ultrasound of carotid artery-based deep learning using Inception V3 and VGG-16.

Abstract

This study aims to automatically classify color Doppler images into two categories for stroke risk prediction based on the carotid plaque. The first category is high-risk carotid vulnerable plaque, and the second is stable carotid plaque. In this research study, we used a deep learning framework based on transfer learning to classify color Doppler images into two categories: one is high-risk carotid vulnerable plaque, and the other is stable carotid plaque. The data were collected from the Second Affiliated Hospital of Fujian Medical University, including stable and vulnerable cases. A total of 87 patients with risk factors for atherosclerosis in our hospital were selected. We used 230 color Doppler ultrasound images for each category and further divided those into the training set and test set in a ratio of 70 and 30%, respectively. We have implemented Inception V3 and VGG-16 pre-trained models for this classification task. Using the proposed framework, we implemented two transfer deep learning models: Inception V3 and VGG-16. We achieved the highest accuracy of 93.81% by using fine-tuned and adjusted hyperparameters according to our classification problem. In this research, we classified color Doppler ultrasound images into high-risk carotid vulnerable and stable carotid plaques. We fine-tuned pre-trained deep learning models to classify color Doppler ultrasound images according to our dataset. Our suggested framework helps prevent incorrect diagnoses caused by low image quality and individual experience, among other factors.