CDIPEM: Cardiovascular disease incidence probability estimation CNN model

Abstract

Heart disease affects the majority of the world’s population, with a high rate of morbidity and mortality. The prediction of heart disease at an early stage is viewed as one of the imperative issues in clinical trials. Since there is a huge amount of medical data and it keeps on increasing, it becomes very difficult to analyze and process such data. Hence, machine learning algorithms become an obvious choice to handle such data. This paper presents a classification framework for clinical decision-making that uses an optimized convolutional neural network (CNN). The pre-processed clinical dataset is used for the training and testing of the classifier. A publicly available UCI dataset is used to examine how the system performs. The statistical features are extracted at the beginning and then subsequent data minimization is carried out. Therein-after, the prediction of heart diseases is done with the help of CNN. The identification of optimal weights is done to enhance the performance of CNN which in turn gives accurate disease prediction results. In this paper, we propose a hybrid Particle Swarm Optimization (PSO) and Gray wolf Optimization (GWO) technique namely PS-GW technique to identify the optimal weight parameters. Finally, performance analysis is done and the experimental results are compared with the existing approaches which show improved classification accuracy over conventional optimization techniques and some well-known classifiers such as k-nearest neighbour, decision tree, logistic regression, naive bayes and random forest.