Background: Coronary heart disease (CHD) is one of the deadliest diseases and a risk prediction model for cardiovascular conditions is needed. Due to the huge number of features that lead to heart problems, it is often difficult for an expert to evaluate these huge features into account. So, there is a need of appropriate feature selection for the given CHD dataset. For early CHD detection, deep learning modes (DL) show promising results in the existing studies.
Objective: This study aimed to develop a deep convolution neural network (CNN) model for classification with a selected number of efficient features using the LASSO (least absolute shrinkage and selection operator) technique. Also, aims to compare the model with similar studies and analyze the performance of the proposed model using accuracy measures.
Methods: The CHD dataset of NHANES (National Health and Nutritional Examination Survey) was examined with 49 features using LASSO technique. This research work is an attempt to apply an improved CNN model for the classification of the CHD dataset with huge features CNN model with feature extractor consists of a fully connected layer with two convolution 1D layers, and classifier part consists of two fully connected layers with SoftMax function was trained on this dataset. Metrics like accuracy recall, specificity, and ROC were used for the evaluation of the proposed model.
Results: The feature selection was performed by applying the LASSO model. The proposed CNN model achieved 99.36% accuracy, while previous studies model achieved over 80 to 92% accuracy.
Conclusion: The application of the proposed CNN with the LASSO model for the classification of CHD can speed up the diagnosis of CHD and appears to be effective in predicting cardiovascular disease based on risk features.
Keywords: Coronary heart disease; LASSO; convolutional neural network; deep learning; machine learning.