Enhancing Heart Disease Detection Using Convolutional Neural Networks and Classic Machine Learning Methods
DOI:
https://doi.org/10.52435/complete.v4i2.394Keywords:
Convolutional Neural Networks , Diagnostic Accuracy, Healthcare Applications, Heart Disease Detection, Machine Learning AlgorithmsAbstract
This study addresses the problem of heart disease detection, a critical concern in public health. The research aims to compare the performance of Convolutional Neural Networks (CNN) with conventional machine learning algorithms in diagnosing heart disease using a dataset comprising 14 features. The primary objective is to determine whether CNNs can provide more accurate and reliable results than traditional techniques. The research employs rigorous preprocessing, normalizing relevant features, and splits the dataset into an 80-20 training-testing split. The model is trained for 300 epochs with a batch size of 64, and performance evaluation is conducted using confusion matrices and classification reports. The results reveal that the CNN model achieved a remarkable accuracy of 100%, demonstrating its potential to outperform conventional machine learning algorithms. These findings emphasize the significance of deep learning techniques in improving heart disease diagnostics, although further research is needed to optimize CNN models and address interpretability concerns for practical implementation in healthcare settings.
References
L. KN, N. R, N. K, R. Kumari, S. N, and V. K, “Heart Disease Detection using Machine Learning Technique,” in 2021 Second International Conference on Electronics and Sustainable Communication Systems (ICESC), Aug. 2021, pp. 1738–1743. doi: 10.1109/ICESC51422.2021.9532705.
C. Yohannes, I. Nurtanio, and K. C. Halim, “Potential of Heart Disease Detection Based on Iridology,” IOP Conf. Ser. Mater. Sci. Eng., vol. 875, no. 1, p. 012034, Jun. 2020, doi: 10.1088/1757-899X/875/1/012034.
R. Buettner and M. Schunter, “Efficient machine learning based detection of heart disease,” in 2019 IEEE International Conference on E-health Networking, Application & Services (HealthCom), Oct. 2019, pp. 1–6. doi: 10.1109/HealthCom46333.2019.9009429.
A. Rath, D. Mishra, G. Panda, and S. C. Satapathy, “Heartdisease detection using deep learning methods from imbalanced ECG samples,” Biomed. Signal Process. Control, vol. 68, p. 102820, Jul. 2021, doi: 10.1016/j.bspc.2021.102820.
F. F. Firdaus, H. A. Nugroho, and I. Soesanti, “Deep Neural Network with Hyperparameter Tuning for Detection of Heart Disease,” in 2021 IEEE Asia Pacific Conference on Wireless and Mobile (APWiMob), Apr. 2021, pp. 59–65. doi: 10.1109/APWiMob51111.2021.9435250.
A. Rath, D. Mishra, G. Panda, and S. C. Satapathy, “Heartdisease detection using deep learning methods from imbalanced ECG samples,” Biomed. Signal Process. Control, vol. 68, p. 102820, Jul. 2021, doi: 10.1016/j.bspc.2021.102820.
F. F. Firdaus, H. A. Nugroho, and I. Soesanti, “Deep Neural Network with Hyperparameter Tuning for Detection of Heart Disease,” in 2021 IEEE Asia Pacific Conference on Wireless and Mobile (APWiMob), Apr. 2021, pp. 59–65. doi 10.1109/APWiMob51111.2021.9435250.
A. Rath, D. Mishra, G. Panda, and S. C. Satapathy, “Heartdisease detection using deep learning methods from imbalanced ECG samples,” Biomed. Signal Process. Control, vol. 68, p. 102820, Jul. 2021, doi: 10.1016/j.bspc.2021.102820.
S. Shinde and J. C. Martinez-Ovando, “Heart Disease Detection with Deep Learning Using a Combination of Multiple Input Sources,” in 2021 IEEE Fifth Ecuador Technical Chapters Meeting (ETCM), Oct. 2021, pp. 1–3. doi: 10.1109/ETCM53643.2021.9590672.
A. Fares and Z. Masetic, “Comparison of Feature Extraction Methods for Heart Disease Classification,” in Advanced Technologies, Systems, and Applications VI, N. Ademovic, E. Mujcic, Z. Aksamija, J. Kevric, S. Avdakovic, and I. Volic, Eds., Cham: Springer International Publishing, 2022, pp. 221–236.
A. A. Abdullah, N. A. Alhadi, and W. Khairunizam, “Diagnosis of Heart Disease Using Machine Learning Methods,” in Intelligent Manufacturing and Mechatronics, M. S. Bahari, A. Harun, Z. Zainal Abidin, R. Hamidon, and S. Zakaria, Eds., Singapore: Springer Singapore, 2021, pp. 77–89.
S. B. Garg, P. Rani, and J. Garg, “Performance Analysis of Classification Methods in the Diagnosis of Heart Disease,” in Mobile Radio Communications and 5G Networks, N. Marriwala, C. C. Tripathi, D. Kumar, and S. Jain, Eds., Singapore: Springer Singapore, 2021, pp. 717–728.
A. Sharma, R. Kumar, and V. Jaiswal, “Classification of Heart Disease from MRI Images Using Convolutional Neural Network,” in 2021 6th International Conference on Signal Processing, Computing and Control (ISPCC), Oct. 2021, pp. 358–363. doi: 10.1109/ISPCC53510.2021.9609408.
M. A. F. Toledo, D. M. Lima, J. E. Krieger, and M. A. Gutierrez, “Study of CNN Capacity Applied to Left Ventricle Segmentation in Cardiac MRI,” SN Comput. Sci., vol. 2, no. 6, p. 480, Oct. 2021, doi: 10.1007/s42979-021-00897-x.
M. Fradi, L. Khriji, and M. Machhout, “Real-time arrhythmia heart disease detection system using CNN architecture based various optimizers-networks,” Multimed. Tools Appl., vol. 81, no. 29, pp. 41711–41732, Dec. 2022, doi: 10.1007/s11042-021-11268-2.
H. Yu, L. T. Yang, Q. Zhang, D. Armstrong, and M. J. Deen, “Convolutional neural networks for medical image analysis: State-of-the-art, comparisons, improvement and perspectives,” Neurocomputing, vol. 444, pp. 92–110, Jul. 2021, doi: 10.1016/j.neucom.2020.04.157.
M. Fradi, L. Khriji, and M. Machhout, “Real-time arrhythmia heart disease detection system using CNN architecture based various optimizers-networks,” Multimed. Tools Appl., vol. 81, no. 29, pp. 41711–41732, Dec. 2022, doi: 10.1007/s11042-021-11268-2.
L. KN, N. R, N. K, R. Kumari, S. N, and V. K, “Heart Disease Detection using Machine Learning Technique,” in 2021 Second International Conference on Electronics and Sustainable Communication Systems (ICESC), Aug. 2021, pp. 1738–1743. doi: 10.1109/ICESC51422.2021.9532705.
K. Battula, R. Durgadinesh, K. Suryapratap, and G. Vinaykumar, “Use of Machine Learning Techniques in the Prediction of Heart Disease,” in 2021 International Conference on Electrical, Computer, Communications and Mechatronics Engineering (ICECCME), Oct. 2021, pp. 1–5. doi: 10.1109/ICECCME52200.2021.9591026.
K. Uma Pavan Kumar, O. Gandhi, M. Venkata Reddy, and S. V. N. Srinivasu, “Usage of KNN, Decision Tree and Random Forest Algorithms in Machine Learning and Performance Analysis with a Comparative Measure,” in Machine Intelligence and Soft Computing, D. Bhattacharyya and N. Thirupathi Rao, Eds., Singapore: Springer Singapore, 2021, pp. 473–479.
C. Sothe et al., “Comparative performance of convolutional neural network, weighted and conventional support vector machine and random forest for classifying tree species using hyperspectral and photogrammetric data,” GIScience Remote Sens., vol. 57, no. 3, pp. 369–394, Apr. 2020, doi: 10.1080/15481603.2020.1712102.
A. S. Prapty and T. T. Shitu, “An Efficient Decision Tree Establishment and Performance Analysis with Different Machine Learning Approaches on Polycystic Ovary Syndrome,” in 2020 23rd International Conference on Computer and Information Technology (ICCIT), Dec. 2020, pp. 1–5. doi: 10.1109/ICCIT51783.2020.9392666.
A.-S. Khaled, “How to Stepping up Characters Recognition using CNN Algorithm?,” Int. J.Inform. Comput., vol. 4, no. 2, p. 36, Dec. 2022, doi: 10.35842/ijicom.v4i2.53.
K. Ang, N. Me, J. Evans E., H. Bha.,S. Jothi, “A Hybrid Approach for Heart Disease Prediction,” 2021 EAI., Jul. 2021, doi: 10.4108/eai.7-6-2021.2308784.
M. Saw, T. Saxena, S. Kaithwas, R. Yadav, and N. Lal, “Estimation of Prediction for Getting Heart Disease Using Logistic Regression Model of Machine Learning,” in 2020 International Conference on Computer Communication and Informatics (ICCCI), Jan. 2020, pp. 1–6. doi: 10.1109/ICCCI48352.2020.9104210.
F.A. Latifah, I. Slamet and Sugiyanto, “Comparison of heart disease classification with logistic regression algorithm and random forest algorithm,” AIP Conference Proceedings 2296, 020021 (2020), Nov.2020, Volume 2296, Issue 1, doi: 10.1063/5.0030579.
Y. Zhang, L. Diao, and L. Ma, “Logistic Regression Models in Predicting Heart Disease,” J. Phys. Conf. Ser., vol. 1769, no. 1, p. 012024, Jan. 2021, doi: 10.1088/1742-6596/1769/1/012024.
A. Dhankhar, S. Juneja, A. Juneja and Vikram Bali, “Kernel Parameter Tuning to Tweak the Performance of Classifiers for Identification of Heart Diseases,” International Journal of E-Health and Medical Communications (IJEHMC), 2021, Vol. 12, Issue 4, doi:10.4018/IJEHMC.20210701.oa1.
L. KN, N. R, N. K, R. Kumari, S. N, and V. K, “Heart Disease Detection using Machine Learning Technique,” in 2021 Second International Conference on Electronics and Sustainable Communication Systems (ICESC), Aug. 2021, pp. 1738–1743. doi: 10.1109/ICESC51422.2021.9532705.
L. Ashish, S. K. V, and S. Yeligeti, “WITHDRAWN: Ischemic heart disease detection using support vector Machine and extreme gradient boosting method,” Mater. Today Proc., Feb. 2021, doi: 10.1016/j.matpr.2021.01.715.
T. -N. Nguyen, T. -H. Nguyen, D. -D. Vo, and T. -D. Nguyen, “Multi-class Support Vector Machine Algorithm for Heart Disease Classification,” in 2020 5th International Conference on Green Technology and Sustainable Development (GTSD), Nov. 2020, pp. 137–140. doi: 10.1109/GTSD50082.2020.9303081.
P. Mahalingam, D. Kalpana, and T. Thyagarajan, “Overfit Analysis on Decision Tree Classifier for Fault Classification in DAMADICS,” in 2021 IEEE Madras Section Conference (MASCON), Aug. 2021, pp. 1–4. doi: 10.1109/MASCON51689.2021.9563557.
A. Ishak, A. Ginting, K. Siregar, and C. Junika, “Clasiffication of Heart Disease using Decision Tree Algorithm,” IOP Conf. Ser. Mater. Sci. Eng., vol. 1003, no. 1, p. 012119, Dec. 2020, doi: 10.1088/1757-899X/1003/1/012119.
Y. Wang et al., “A Machine learning-based prediction model for the heart diseases from chance factors through two-variable decision tree classifier,” J. Intell. Fuzzy Syst., vol. 41, no. 6, pp. 5985–6002, 2021, doi: 10.3233/JIFS-202226.
K. Eva Mamabolo and M. Mosia, “Heart Disease Risk Level Prediction: Knitting Machine Learning Classifiers,” in 2020 2nd International Multidisciplinary Information Technology and Engineering Conference (IMITEC), Nov. 2020, pp. 1–5. doi: 10.1109/IMITEC50163.2020.9334141.
N. Wijaya, J. Aryanto, K. Kasmawaru, and A. F. Rachman, “Implementation of KNN Algorithm for Occupancy Classification of Rehabilitation Houses,” Int. J. Inform. Comput., vol. 4, no. 2, p. 7, Dec. 2022, doi: 10.35842/ijicom.v4i2.36.
C. Gong, Z. Su, P. Wang, Q. Wang, and Y. You, “Evidential instance selection for K-nearest neighbor classification of big data,” Int. J. Approx. Reason., vol. 138, pp. 123–144, Nov. 2021, doi: 10.1016/j.ijar.2021.08.006.
Anggoro, D. A. ., & Aziz, N. C. (2021). Implementation of K-Nearest Neighbors Algorithm for Predicting Heart Disease Using Python Flask. Iraqi Journal of Science, 62(9), 3196–3219. https://doi.org/10.24996/ijs.2021.62.9.33
K. Nguyen et al., “Heart Disease Classification using Novel Heterogeneous Ensemble,” in 2021 IEEE EMBS International Conference on Biomedical and Health Informatics (BHI), Jul. 2021, pp. 1–4. doi: 10.1109/BHI50953.2021.9508516.
M. G. El-Shafiey, A. Hagag, E. -S. A. El-Dahshan, and M. A. Ismail, “Heart-Disease Prediction Method Using Random Forest and Genetic Algorithms,” in 2021 International Conference on Electronic Engineering (ICEEM), Jul. 2021, pp. 1–6. doi: 10.1109/ICEEM52022.2021.9480625.
S. Asadi, S. Roshan, and M. W. Kattan, “Random forest swarm optimization-based for heart diseases diagnosis,” J. Biomed. Inform., vol. 115, p. 103690, Mar. 2021, doi: 10.1016/j.jbi.2021.103690.
J. Hansen, “Diabetic risk prognosis with tree ensembles integrating feature attribution methods,” Evol. Intell., Sep. 2021, doi: 10.1007/s12065-021-00663-1.
G. S. Sajja, M. Mustafa, K. Phasinam, K. Kaliyaperumal, R. J. M. Ventayen, and T. Kassanuk, “Towards Application of Machine Learning in Classification and Prediction of Heart Disease,” in 2021 Second International Conference on Electronics and Sustainablee Communication Systems (ICESC), Aug. 2021, pp. 1664–1669. doi: 10.1109/ICESC51422.2021.9532940.
S. Faiayaz Waris and S. Koteeswaran, “WITHDRAWN: Heart disease early prediction using a novel machine learning method called improved K-means neighbor classifier in python,” Mater. Today Proc., Mar. 2021, doi: 10.1016/j.matpr.2021.01.570.
H. Kumar Thakkar, H. Shukla, and S. Patil, “A Comparative Analysis of Machine Learning Classifiers for Robust Heart Disease Prediction,” in 2020 IEEE 17th India Council International Conference (INDICON), Dec. 2020, pp. 1–6. doi: 10.1109/INDICON49873.2020.9342444.
S. Mishra, P. K. Mallick, H. K. Tripathy, A. K. Bhoi, and A. González-Briones, “Performance Evaluation of a Proposed Machine Learning Model for Chronic Disease Datasets Using an Integrated Attribute Evaluator and an Improved Decision Tree Classifier,” Appl. Sci., vol. 10, no. 22, 2020, doi 10.3390/app10228137.
S. Ketu and P. K. Mishra, “Empirical Analysis of Machine Learning Algorithms on Imbalance Electrocardiogram Based Arrhythmia Dataset for Heart Disease Detection,” Arab. J. Sci. Eng., vol. 47, no. 2, pp. 1447–1469, Feb. 2022, doi: 10.1007/s13369-021-05972-2.
S. Kusuma and K. R. Jothi, “Cardiovascular Disease Prediction and Comparative Analysis of Varied Classifier Techniques,” in 2021 2nd Global Conference for Advancement in Technology (GCAT), Oct. 2021, pp. 1–7. doi: 10.1109/GCAT52182.2021.9587734.
J. Piri and P. Mohapatra, “Imbalanced Cardiotocography Data Classification Using Re-sampling Techniques,” in Proceedings of International Conference on Machine Intelligence and Data Science Applications, M. Prateek, T. P. Singh, T. Choudhury, H. M. Pandey, and N. Gia Nhu, Eds., Singapore: Springer Singapore, 2021, pp. 681–692.
F. Tasnim and S. U. Habiba, “A Comparative Study on Heart Disease Prediction Using Data Mining Techniques and Feature Selection,” in 2021 2nd International Conference on Robotics, Electrical and Signal Processing Techniques (ICREST), Jan. 2021, pp. 338–341. doi: 10.1109/ICREST51555.2021.9331158.
49 A. A. Ali, H. S. Hassan, and E. M. Anwar, “Improve the Accuracy of Heart Disease Predictions Using Machine Learning and Feature Selection Techniques,” in Machine Learning, Image Processing, Network Security and Data Sciences, A. Bhattacharjee, S. Kr. Borgohain, B. Soni, G. Verma, and X.-Z. Gao, Eds., Singapore: Springer Singapore, 2020, pp. 214–228.
I. M. I’tisyam, R. N. Wijaya, and R. Pradila, “Implementation of Deep Learning for Classification of Mushroom Using the CNN Algorithm”.
K. Fathima and E. R. Vimina, “Heart Disease Prediction Using Deep Neural Networks: A Novel Approach,” in Intelligent Sustainable Systems, J. S. Raj, R. Palanisamy, I. Perikos, and Y. Shi, Eds., Singapore: Springer Singapore, 2022, pp. 725–736.
A. Sharma, R. Kumar, and V. Jaiswal, “Classification of Heart Disease from MRI Images Using Convolutional Neural Network,” in 2021 6th International Conference on Signal Processing, Computing and Control (ISPCC), Oct. 2021, pp. 358–363. doi: 10.1109/ISPCC53510.2021.9609408.
L. KN, N. R, N. K, R. Kumari, S. N, and V. K, “Heart Disease Detection using Machine Learning Technique,” in 2021 Second International Conference on Electronics and Sustainable Communication Systems (ICESC), Aug. 2021, pp. 1738–1743. doi: 10.1109/ICESC51422.2021.9532705.
M. Fradi, L. Khriji, and M. Machhout, “Real-time arrhythmia heart disease detection system using CNN architecture based various optimizers-networks,” Multimed. Tools Appl., vol. 81, no. 29, pp. 41711–41732, Dec. 2022, doi: 10.1007/s11042-021-11268-2.
R. Banerjee, A. Ghose, and K. Muthana Mandana, “A Hybrid CNN-LSTM Architecture for Detection of Coronary Artery Disease from ECG,” in 2020 International Joint Conference on Neural Networks (IJCNN), Jul. 2020, pp. 1–8. doi: 10.1109/IJCNN48605.2020.9207044.
N. Fayyazifar, “An Accurate CNN Architecture For Atrial Fibrillation Detection Using Neural Architecture Search,” in 2020 28th European Signal Processing Conference (EUSIPCO), Jan. 2021, pp. 1135–1139. doi 10.23919/Eusipco47968.2020.9287496.
K. Battula, R. Durgadinesh, K. Suryapratap, and G. Vinaykumar, “Use of Machine Learning Techniques in the Prediction of Heart Disease,” in 2021 International Conference on Electrical, Computer, Communications and Mechatronics Engineering (ICECCME), Oct. 2021, pp. 1–5. doi: 10.1109/ICECCME52200.2021.9591026.
R. Anand, S. Fazlul Kareem, R. Mohamed Arshad Mubeen, S. Ramesh, and B. Vignesh, “Analysis Of Heart Risk Detection In Machine Learning Using Blockchain,” in 2021 6th International Conference on Signal Processing, Computing and Control (ISPCC), Oct. 2021, pp. 685–689. doi: 10.1109/ISPCC53510.2021.9609353.
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Sri Hasta Mulyani, Nurhadi Wijaya, Fike Trinidya
This work is licensed under a Creative Commons Attribution 4.0 International License.
