Cardiovascular disease serves as the leading cause of death worldwide. Calcification detection is considered an important factor in cardiovascular diseases. Currently, medical practitioners visually inspect the presence of calcification using intravascular ultrasound (IVUS) images. The study aims to detect the extent of calcification as belonging to class I, II as mild calcification, and class III, IV as dense calcification from IVUS images acquired at 40 MHz. To detect calcification, the features were extracted using improved AlexNet architecture and then were fed into machine learning classifiers. The experiments were carried out using 14 real IVUS pullbacks of 10 patients. Experimental results show that the combination of traditional machine learning with deep learning approaches significantly improves accuracy. The results show that support vector machines outperform all other classifiers. The proposed model is compared with two other pre-trained models GoogLeNet (98.8%), SqueezeNet (99.2%), and exhibits considerable improvement in classification accuracy (99.8%). In the future other models such as Vision Transformers could be explored with additional feature selection methods such as ReliefF, PSO, ACO, etc. to improve the overall accuracy of diagnosis.
Keywords: convolutional neural network; coronary artery calcification; intravascular ultrasound; pre-trained networks; support vector machine.