A machine learning approach to cardiovascular disease prediction with advanced feature selection

Abdikadir Hussein Elmi, Abdijalil Abdullahi, Mohamed Ali Barre


Cardiovascular diseases (CVDs) pose a significant global public health challenge, necessitating precise risk assessment for proactive treatment and optimal utilization of healthcare resources. This study employs machine learning algorithms and sophisticated feature selection techniques to enhance the accuracy and comprehensibility of CVD prediction models. While traditional risk assessment tools are valuable, they frequently fail to consider the myriad intricate factors that contribute to the heightened risk of CVD. Our methodology employs machine learning algorithms to analyze diverse healthcare data sources and produce advanced predictive models. The salient feature of this research lies in the meticulous application of advanced feature selection techniques, enabling the identification of pivotal factors within heterogeneous datasets. Optimizing feature selection enhances the interpretability of the model, reduces dimensionality, and improves predictive accuracy. The area under the ROC curve (AUC-ROC) score of the wrapper method model significantly decreased from 95.1% to 75.1% after tuning, based on empirical tests that supported the suggested method. This showcases its capacity as a tool for assessing premature CVD susceptibility and developing tailored healthcare strategies. The study highlights the significance of integrating machine learning with feature selection due to the widespread influence of cardiovascular diseases. Integrating this system has the potential to enhance patient care and optimize the utilization of healthcare resources.


Cardiovascular diseases; Feature selection; Filter method; Machine learning; Wrapper method

Full Text:


DOI: http://doi.org/10.11591/ijeecs.v33.i2.pp1030-1041


  • There are currently no refbacks.

Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

The Indonesian Journal of Electrical Engineering and Computer Science (IJEECS)
p-ISSN: 2502-4752, e-ISSN: 2502-4760
This journal is published by the Institute of Advanced Engineering and Science (IAES) in collaboration with Intelektual Pustaka Media Utama (IPMU).

shopify stats IJEECS visitor statistics