Parametric-based feature selection via spherical harmonic coefficients for the left ventricle myocardial infarction screening

Abstract

Computer-aided diagnosis (CAD) of heart diseases using machine learning techniques has recently received much attention. In this study, we present a novel parametric-based feature selection method using the three-dimensional spherical harmonic (SHs) shape descriptors of the left ventricle (LV) for intelligent myocardial infarction (MI) classification. The main hypothesis is that the SH coefficients of the parameterized endocardial shapes in MI patients are recognizable and distinguishable from healthy subjects. The SH parameterization, expansion, and registration of the LV endocardial shapes were performed, then parametric-based features were extracted. The proposed method performance was investigated by varying considered phases (i.e., the end-systole (ES) or the end-diastole (ED) frames), the spatial alignment procedures based on three modes (i.e., the center of the apical (CoA), the center of mass (CoM), and the center of the basal (CoB)), and considered orders of SH coefficients. After applying principal component analysis (PCA) on the feature vectors, support vector machine (SVM), K-nearest neighbors (K-NN), and random forest (RF) were trained and tested using the leave-one-out cross-validation (LOOCV). The proposed method validation was performed via a dataset containing healthy and MI subjects selected from the automated cardiac diagnosis challenge (ACDC) database. The promising results show the effectiveness of the proposed classification model. SVM reached the best performance with accuracy, sensitivity, specificity, and F-score of 97.50%, 95.00%, 100.00%, and 97.56%, respectively, using the introduced optimum feature set. This study demonstrates the robustness of combining the SH coefficients and machine learning techniques. We also quantify and notably highlight the contribution of different parameters in the classification and finally introduce an optimal feature set with maximum discriminant strength for the MI classification task. Moreover, the obtained results confirm that the proposed method performs more accurately than conventional point-based methods and also the current start-of-the-art, i.e., clinical measures. We showed our method’s generalizability using employing it in dilated cardiomyopathy (DCM) detection and achieving promising results too. Graphical abstract: Parametric-based feature selection via spherical harmonics coefficients for the left ventricle myocardial infarction screening [Figure not available: see fulltext.].