Enhanced Cardiovascular Disease Prediction Modelling using Machine Learning Techniques: A Focus on CardioVitalnet.

Journal: Network (Bristol, England)
Published Date:

Abstract

Aiming at early detection and accurate prediction of cardiovascular disease (CVD) to reduce mortality rates, this study focuses on the development of an intelligent predictive system to identify individuals at risk of CVD. The primary objective of the proposed system is to combine deep learning models with advanced data mining techniques to facilitate informed decision-making and precise CVD prediction. This approach involves several essential steps, including the preprocessing of acquired data, optimized feature selection, and disease classification, all aimed at enhancing the effectiveness of the system. The chosen optimal features are fed as input to the disease classification models and into some Machine Learning (ML) algorithms for improved performance in CVD classification. The experiment was simulated in the Python platform and the evaluation metrics such as accuracy, sensitivity, and F1_score were employed to assess the models' performances. The ML models (Extra Trees (ET), Random Forest (RF), AdaBoost, and XG-Boost) classifiers achieved high accuracies of 94.35%, 97.87%, 96.44%, and 99.00%, respectively, on the test set, while the proposed CardioVitalNet (CVN) achieved 87.45% accuracy. These results offer valuable insights into the process of selecting models for medical data analysis, ultimately enhancing the ability to make more accurate diagnoses and predictions.

Authors

  • Chukwuebuka Joseph Ejiyi
    College of Nuclear Technology and Automation Engineering, Sichuan Industrial Internet Intelligent Monitoring and Application Engineering Research Center, Chengdu University of Technology, Sichuan, Chengdu, China; Network and Data Security Key Laboratory of Sichuan Province, University of Electronic Science and Technology of China, Chengdu, Sichuan, China.
  • Zhen Qin
    College of Forestry, Southwest Forestry University, Kunming, Yunnan, China.
  • Grace Ugochi Nneji
  • Happy Nkanta Monday
  • Victor K Agbesi
    School of Computer Science and Engineering, University of Electronic Science and Technology of China, Chengdu, China.
  • Makuachukwu Bennedith Ejiyi
    Pharmacy Department, University of Nigeria Nsukka, Nsukka, Enugu State, Nigeria. Electronic address: makuachukwubennedith@gmail.com.
  • Thomas Ugochukwu Ejiyi
    Department of Pure and Industrial Chemistry, University of Nigeria Nsukka, Enugu, Nigeria.
  • Olusola O Bamisile
    College of Nuclear Technology and Automation Engineering, & Sichuan Industrial Internet Intelligent Monitoring and Application Engineering Research Center, Chengdu University of Technology, Sichuan, Chengdu, China.