Artificial intelligence model for prediction of cardiovascular disease: An empirical study

Abstract

Cardiovascular disease (CVD) is a disease related to the heart and blood vessels. Prediction of CVD is essential for early detection and diagnosis, which is however compounded by the complex interplay between medical history, physical examination outcomes, and imaging results. While the existing automated systems are fraught with the usage of irrelevant and redundant attributes, artificial intelligence (AI) helps in the identification of potential CVD populations by prediction models. This work aims at developing an AI model for predicting CVD using different classifications of machine learning techniques. The CVD dataset was obtained from the UCI repository containing about 76 cardiac attributes for training in various machine learning models, which include a hybrid of artificial neural network-genetic algorithm (ANN-GA), artificial neural network, support vector machine (SVM), K-means, K-nearest neighbor (KNN), and decision tree (DT). The performance of the models was measured in terms of accuracy, means square error, sensitivity, specificity, and precision. The results showed that the hybrid model of ANN-GA performs better with an accuracy of 86.4%, compared to the SVM, K-means, KNN, and DT measured at 84.0%, 59.6%, 79.0%, and 77.8%, respectively. It was observed that the system performs better as the number of datasets increases in the database, with a fewer selection of attributes using genetic algorithm for selection. Thus, the ANN-GA model is recommended for CVD prediction and diagnosis.