Multi-Label Active Learning-Based Machine Learning Model for Heart Disease Prediction

Sensors (Basel). 2022 Feb 4;22(3):1184. doi: 10.3390/s22031184.

Abstract

The rapid growth and adaptation of medical information to identify significant health trends and help with timely preventive care have been recent hallmarks of the modern healthcare data system. Heart disease is the deadliest condition in the developed world. Cardiovascular disease and its complications, including dementia, can be averted with early detection. Further research in this area is needed to prevent strokes and heart attacks. An optimal machine learning model can help achieve this goal with a wealth of healthcare data on heart disease. Heart disease can be predicted and diagnosed using machine-learning-based systems. Active learning (AL) methods improve classification quality by incorporating user-expert feedback with sparsely labelled data. In this paper, five (MMC, Random, Adaptive, QUIRE, and AUDI) selection strategies for multi-label active learning were applied and used for reducing labelling costs by iteratively selecting the most relevant data to query their labels. The selection methods with a label ranking classifier have hyperparameters optimized by a grid search to implement predictive modelling in each scenario for the heart disease dataset. Experimental evaluation includes accuracy and F-score with/without hyperparameter optimization. Results show that the generalization of the learning model beyond the existing data for the optimized label ranking model uses the selection method versus others due to accuracy. However, the selection method was highlighted in regards to the F-score using optimized settings.

Keywords: active learning; chronic diseases; data mining; heart disease; machine learning; multi-label classification.

MeSH terms

  • Cardiovascular Diseases*
  • Delivery of Health Care
  • Heart Diseases* / diagnosis
  • Humans
  • Machine Learning
  • Supervised Machine Learning