Hyperparameter optimization

Operating system

Support vector machine

Data mining

Machine learning

Matthews correlation coefficient

Artificial intelligence

Correlation coefficient

Computer science

Process (computing)

IEEE Access

About half of the people who develop heart failure (HF) die within five years of diagnosis. Over the years, researchers have developed several machine learning-based models for the early prediction of HF and to help cardiologists to improve the diagnosis process. In this paper, we introduce an expert system that stacks two support vector machine (SVM) models for the effective prediction of HF. The first SVM model is linear and L <sub xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">1</sub> regularized. It has the capability to eliminate irrelevant features by shrinking their coefficients to zero. The second SVM model is L <sub xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">2</sub> regularized. It is used as a predictive model. To optimize the two models, we propose a hybrid grid search algorithm (HGSA) that is capable of optimizing the two models simultaneously. The effectiveness of the proposed method is evaluated using six different evaluation metrics: accuracy, sensitivity, specificity, the Matthews correlation coefficient (MCC), ROC charts, and area under the curve (AUC). The experimental results confirm that the proposed method improves the performance of a conventional SVM model by 3.3%. Moreover, the proposed method shows better performance compared to the ten previously proposed methods that achieved accuracies in the range of 57.85%-91.83%. In addition, the proposed method also shows better performance than the other state-of-the-art machine learning ensemble models.

An Optimized Stacked Support Vector Machines Based Expert System for the Effective Prediction of Heart Failure