Diagnosis of heart disease using oversampling methods and decision tree classifier in cardiology

Abstract

Purpose Heart disease is one of the most prevalent and critical diseases that endangers the lives of human beings. In addition to clinical diagnosis, machine learning and deep learning-based approaches are vital in the diagnosis of heart disease. Method This paper proposes a balanced and optimized machine-learning algorithm for heart disease detection. This technique combines oversampling techniques, attribute pruning, CART decision tree classifier, and rule pruning through hyper-parameter tuning to identify the presence of heart disease. It further identifies the key attributes that contribute to the occurrence of heart malfunctioning. Results Experimental results show that SMOTE sampled dataset exhibits effective performance when implemented using a balanced and optimized machine learning algorithm, with an improvement of 11%, 75%, 62%, and 71% in accuracy, precision, recall, and f1 scores when compared with the dataset that was not subjected to sampling. The algorithm works effectively when the imbalance ratio is high for a dataset. Conclusion The algorithm can be used to predict the presence of heart disease even in highly imbalanced datasets and identify critical features in the malfunctioning of the heart.

Leveraging Sampling Schemes on Skewed Class Distribution to Enhance Male Fertility Detection with Ensemble AI Learners

“…Hence, the process of generating new data is very much typical in healthcare applications. Albert et al [12] recently used smote and adasyn approach to detect heart disease using ML algorithms. The result has shown that oversampling improved model performance by 11% compared to the original dataset.…”

Deteksi Penyakit Jantung Menggunakan Metode Klasifikasi Decision Tree dan Regresi Logistik

“…Penyakit jantung merupakan salah satu penyakit paling umum dan kritis yang membahayakan kehidupan manusia. Selain diagnosis klinis, pembelajaran mesin dan pendekatan berbasis pembelajaran mendalam sangat penting dalam diagnosis penyakit jantung [7].…”