Utilizing Machine Learning for the Early Detection of Coronary Heart Disease
Received: 18 June 2024 | Revised: 15 July 2024 | Accepted: 20 July 2024 | Online: 9 October 2024
Corresponding author: Zaid Ameen Abduljabbar
Abstract
Coronary Heart Disease (CHD) is a persistent health issue, and risk prognosis is very important because it creates opportunities for doctors to provide early solutions. Despite such promising results, this type of analysis runs into several problems, such as accurately handling high-dimensional data because of the abundance of extracted information that hampers the prediction process. This paper presents a new approach that integrates Principal Component Analysis (PCA) and feature selection techniques to improve the prediction performance of CHD models, especially in light of dimensionality consideration. Feature selection is identified as one of the contributors to enhance model performance. Reducing the input space and identifying important attributes related to heart disease offers a refined approach to CHD prediction. Then four classifiers were used, namely PCA, Random Forest (RF), Decision Trees (DT), and AdaBoost, and an accuracy of approximately 96% was achieved, which is quite satisfactory. The experimentations demonstrated the effectiveness of this approach, as the proposed model was more effective than the other traditional models including the RF and LR in aspects of precision, recall, and AUC values. This study proposes an approach to reduce data dimensionality and select important features, leading to improved CHD prediction and patient outcomes.
Keywords:
random forest, decision trees, Principal Component Analysis (PCA), Machine Learning (ML) classifiers, Coronary Heart Disease (CHD), hyperparameters, predictionDownloads
References
C. Bemando, E. Miranda, and M. Aryuni, "Machine-Learning-Based Prediction Models of Coronary Heart Disease Using Naïve Bayes and Random Forest Algorithms," in 2021 International Conference on Software Engineering & Computer Systems and 4th International Conference on Computational Science and Information Management (ICSECS-ICOCSIM), Pekan, Malaysia, Aug. 2021, pp. 232–237.
Y. Khourdifi and M. Bahaj, "Heart Disease Prediction and Classification Using Machine Learning Algorithms Optimized by Particle Swarm Optimization and Ant Colony Optimization," International Journal of Intelligent Engineering and Systems, vol. 12, no. 1, pp. 242–252, Feb. 2019.
H. Yang and J. M. Garibaldi, "A hybrid model for automatic identification of risk factors for heart disease," Journal of Biomedical Informatics, vol. 58, pp. S171–S182, Dec. 2015.
H. Kim, M. I. M. Ishag, M. Piao, T. Kwon, and K. H. Ryu, "A Data Mining Approach for Cardiovascular Disease Diagnosis Using Heart Rate Variability and Images of Carotid Arteries," Symmetry, vol. 8, no. 6, Jun. 2016, Art. no. 47.
A. K. Gárate-Escamila, A. Hajjam El Hassani, and E. Andrès, "Classification models for heart disease prediction using feature selection and PCA," Informatics in Medicine Unlocked, vol. 19, Jan. 2020, Art. no. 100330.
M. J. Jassim Ghrabat, G. Ma, and C. Cheng, "Towards Efficient for Learning Model Image Retrieval," in 2018 14th International Conference on Semantics, Knowledge and Grids (SKG), Guangzhou, China, Sep. 2018, pp. 92–99.
C. Janiesch, P. Zschech, and K. Heinrich, "Machine learning and deep learning," Electronic Markets, vol. 31, no. 3, pp. 685–695, Sep. 2021.
M. Rolínek, D. Zietlow, and G. Martius, "Variational Autoencoders Pursue PCA Directions (by Accident)," in 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), Long Beach, CA, USA, Jun. 2019, pp. 12398–12407.
K. Dissanayake and M. G. Md Johar, "Comparative Study on Heart Disease Prediction Using Feature Selection Techniques on Classification Algorithms," Applied Computational Intelligence and Soft Computing, vol. 2021, no. 1, 2021, Art. no. 5581806.
J. Abdollahi and B. Nouri-Moghaddam, "A hybrid method for heart disease diagnosis utilizing feature selection based ensemble classifier model generation," Iran Journal of Computer Science, vol. 5, no. 3, pp. 229–246, Sep. 2022.
K. M. Almustafa, "Prediction of heart disease and classifiers’ sensitivity analysis," BMC Bioinformatics, vol. 21, no. 1, Jul. 2020, Art. no. 278.
T. K. Sajja and H. K. Kalluri, "A Deep Learning Method for Prediction of Cardiovascular Disease Using Convolutional Neural Network," Revue d’Intelligence Artificielle, vol. 34, no. 5, pp. 601–606, Nov. 2020.
S. N. Pasha, D. Ramesh, S. Mohmmad, A. Harshavardhan, and Shabana, "Cardiovascular disease prediction using deep learning techniques," IOP Conference Series: Materials Science and Engineering, vol. 981, no. 2, Sep. 2020, Art. no. 022006.
C. Xiao, Y. Li, and Y. Jiang, "Heart Coronary Artery Segmentation and Disease Risk Warning Based on a Deep Learning Algorithm," IEEE Access, vol. 8, pp. 140108–140121, 2020.
K. Vayadande et al., "Heart Disease Prediction using Machine Learning and Deep Learning Algorithms," in 2022 International Conference on Computational Intelligence and Sustainable Engineering Solutions (CISES), Greater Noida, India, May 2022, pp. 393–401.
T. Amarbayasgalan, V. H. Pham, N. Theera-Umpon, Y. Piao, and K. H. Ryu, "An Efficient Prediction Method for Coronary Heart Disease Risk Based on Two Deep Neural Networks Trained on Well-Ordered Training Datasets," IEEE Access, vol. 9, pp. 135210–135223, 2021.
R. Bharti, A. Khamparia, M. Shabaz, G. Dhiman, S. Pande, and P. Singh, "Prediction of Heart Disease Using a Combination of Machine Learning and Deep Learning," Computational Intelligence and Neuroscience, vol. 2021, no. 1, 2021, Art. no. 8387680.
S. Prusty, S. Patnaik, and S. Kumar Dash, "Comparative analysis and prediction of coronary heart disease," Indonesian Journal of Electrical Engineering and Computer Science, vol. 27, no. 2, Aug. 2022, Art. no. 944.
V. Shorewala, "Early detection of coronary heart disease using ensemble techniques," Informatics in Medicine Unlocked, vol. 26, Jan. 2021, Art. no. 100655.
M. Sebastiani, C. Vacchi, A. Manfredi, and G. Cassone, "Personalized Medicine and Machine Learning: A Roadmap for the Future," Journal of Clinical Medicine, vol. 11, no. 14, Jan. 2022, Art. no. 4110.
F. Özbilgin, Ç. Kurnaz, and E. Aydın, "Prediction of Coronary Artery Disease Using Machine Learning Techniques with Iris Analysis," Diagnostics, vol. 13, no. 6, Jan. 2023, Art. no. 1081.
C. Eyupoglu and O. Karakuş, "Novel CAD Diagnosis Method Based on Search, PCA, and AdaBoostM1 Techniques," Journal of Clinical Medicine, vol. 13, no. 10, Jan. 2024, Art. no. 2868.
G. Deivendran, S. Vishal Balaji, B. Paramasivan, and S. Vimal, "Coronary Illness Prediction Using the AdaBoost Algorithm," in Sensor Data Analysis and Management, John Wiley & Sons, Ltd, 2021, pp. 161–172.
D. Lapp, "Heart Disease Dataset." Kaggle, [Online]. Available: https://www.kaggle.com/datasets/johnsmith88/heart-disease-dataset.
A. Janosi, W. Steinbrunn, M. Pfisterer, and R. Detrano, "Heart Disease." UCI Machine Learning Repository, 1989.
A. Dey, J. Singh, and N. Singh, "Analysis of Supervised Machine Learning Algorithms for Heart Disease Prediction with Reduced Number of Attributes using Principal Component Analysis," International Journal of Computer Applications, vol. 140, no. 2, pp. 27–31, Apr. 2016.
J. Tang, A. Henderson, and P. Gardner, "Exploring AdaBoost and Random Forests machine learning approaches for infrared pathology on unbalanced data sets," Analyst, vol. 146, no. 19, pp. 5880–5891, 2021.
P. Theerthagiri, "Predictive analysis of cardiovascular disease using gradient boosting based learning and recursive feature elimination technique," Intelligent Systems with Applications, vol. 16, Nov. 2022, Art. no. 200121.
R. Shafique, A. Mehmood, S. Ullah, and G. S. Choi, "Cardiovascular Disease Prediction System Using Extra Trees Classifier." Research Square, Sep. 16, 2019.
X. Zhang et al., "An accurate diagnosis of coronary heart disease by Catboost, with easily accessible data," Journal of Physics: Conference Series, vol. 1955, no. 1, Mar. 2021, Art. no. 012027.
M. J. J. Ghrabat et al., "Fully automated model on breast cancer classification using deep learning classifiers," Indonesian Journal of Electrical Engineering and Computer Science, vol. 28, no. 1, Oct. 2022, Art. no. 183.
J. P. Li, A. U. Haq, S. U. Din, J. Khan, A. Khan, and A. Saboor, "Heart Disease Identification Method Using Machine Learning Classification in E-Healthcare," IEEE Access, vol. 8, pp. 107562–107582, 2020.
R. Alizadehsani, M. J. Hosseini, Z. A. Sani, A. Ghandeharioun, and R. Boghrati, "Diagnosis of Coronary Artery Disease Using Cost-Sensitive Algorithms," in 2012 IEEE 12th International Conference on Data Mining Workshops, Brussels, Belgium, Dec. 2012, pp. 9–16.
M. J. Kraak, "Exploratory Visualization," in Encyclopedia of GIS, S. Shekhar and H. Xiong, Eds. Boston, MA, USA: Springer US, 2008, pp. 301–307.
S. Usha and S. Kanchana, "Predicting Heart Disease Using Feature Selection Techniques Based On Data Driven Approach," Webology, vol. 18, no. 4, pp. 97–108, 2021.
Y. Muhammad, M. Tahir, M. Hayat, and K. T. Chong, "Early and accurate detection and diagnosis of heart disease using intelligent computational model," Scientific Reports, vol. 10, no. 1, Nov. 2020, Art. no. 19747.
M. J. J. Ghrabat, G. Ma, Z. A. Abduljabbar, M. A. Al Sibahee, and S. J. Jassim, "Greedy Learning of Deep Boltzmann Machine (GDBM)’s Variance and Search Algorithm for Efficient Image Retrieval," IEEE Access, vol. 7, pp. 169142–169159, 2019.
M. J. Ghrabat, G. Ma, P. L. P. Avila, M. J. Jassim, and S. J. Jassim, "Content-based image retrieval of color, shape and texture by using novel multi-SVM classifier," International Journal of Machine Learning and Computing, vol. 9, no. 4, pp. 483–489, 2019.
S. M. Alanazi and G. S. M. Khamis, "Optimizing Machine Learning Classifiers for Enhanced Cardiovascular Disease Prediction," Engineering, Technology & Applied Science Research, vol. 14, no. 1, pp. 12911–12917, Feb. 2024.
Downloads
How to Cite
License
Copyright (c) 2024 Mudhafar jalil Jassim Ghrabat, Siamand Hassan Mohialdin, Luqman Qader Abdulrahman, Murthad Hussein Al-Yoonus, Zaid Ameen Abduljabbar, Dhafer G. Honi, Vincent Omollo Nyangaresi, Iman Qayes Abduljaleel, Husam A. Neamah
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain the copyright and grant the journal the right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) after its publication in ETASR with an acknowledgement of its initial publication in this journal.