Investigations on coronary artery plaque detection and subclassification using machine learning classifier

Abstract

Coronary artery diseases are one of the high-risk diseases, which occur due to the insufficient blood supply to the heart. The different types of plaques formed inside the artery leads to the blockage of the blood stream. Understanding the type of plaques along with the detection and classification of plaques supports in reducing the mortality of patients. The objective of this study is to present a novel clustering method of plaque segmentation followed by wavelet transform based feature extraction. The extracted features of all different kinds of calcified and sub calcified plaques are applied to first train and test three machine learning classifiers including support vector machine, random forest and decision tree classifiers. The bootstrap ensemble classifier then decides the best classification result through a voting method of three classifiers. A training dataset including 64 normal CTA images and 73 abnormal CTA images is used, while a testing dataset consists of 111 normal CTA images and 103 abnormal CTA images. The evaluation metrics shows better classification rate and accuracy of 97.7%. The Sensitivity and Specificity rates are 97.8% and 97.5%, respectively. As a result, our study results demonstrate the feasibility and advantages of developing and applying this new image processing and machine learning scheme to assist coronary artery plaque detection and classification.