KNN Algorithm Used for Heart Attack Detection

Ibrahima Bah

doi:10.52981/fjes.v11i1.758

Ibrahima Bah Nanjing University of Information Science and Technology

DOI: https://doi.org/10.52981/fjes.v11i1.758

الملخص

Machine Learning, a branch of artificial intelligence, has become more accurate than human medical professionals in predicting the incidence of heart attack or death in patients at risk of coronary artery disease. In this paper, we attempt to employ Artificial Intelligence (AI) to predict heart attack. For this purpose, we employed the popular classification technique named the K-Nearest Neighbor (KNN) algorithm to predict the probability of having the Heart Attack (HA). The dataset used is the cardiovascular dataset available publicly on Kaggle, knowing that someone suffering from cardiovascular disease is likely to succumb to a heart attack. In this work, the research was conducted using two approaches. We use the KNN classifier for the first time, aided by using a correlation matrix to select the best features manually and faster computation, and then optimize the parameters with the K-fold cross-validation technique. This improvement led us to have an accuracy of 72.37% on the test set.

المراجع

1. Jabbar MA (2017) Prediction of heart disease using k-nearest neighbor and particle swarm optimization. Biomed Res 28:4154–4158
2. Heart attack - Symptoms and causes - Mayo Clinic. https://www.mayoclinic.org/diseases-conditions/heart-attack/symptoms-causes/syc-20373106
3. Kadhum N (2018) Improving The Accuracy Of KNN Classifier For Heart Attack Using Genetic Algorithm. 116–125
4. Intro to types of classification algorithms in Machine .... https://medium.com/sifium/machine-learning-types-of-classification-9497bd4f2e14
5. T. Obasi and M. Omair Shafiq, "Towards comparing and using Machine Learning tech-niques for detecting and predicting Heart Attack and Diseases," 2019 IEEE International Conference on Big Data (Big Data), 2019, pp. 2393-2402, doi: 10.1109/BigData47090.2019.9005488.
6. Salman I (2019) Heart attack mortality prediction: An application of machine learning methods. Turkish J Electr Eng Comput Sci 27:4378–4389. https://doi.org/10.3906/ELK-1811-4
7. M. Wang, X. Yao and Y. Chen, "An Imbalanced-Data Processing Algorithm for the Predic-tion of Heart Attack in Stroke Patients," in IEEE Access, vol. 9, pp. 25394-25404, 2021, doi: 10.1109/ACCESS.2021.3057693.
8. Guo G, Wang H, Bell D, et al (2003) KNN model-based approach in classification. Lect Notes Comput Sci (including Subser Lect Notes Artif Intell Lect Notes Bioinformatics) 2888:986–996. https://doi.org/10.1007/978-3-540-39964-3_62
9. Perumal, P., and P. T. Priyanka. "SUPERVISED HEART ATTACK PREDICTION USING SVM WITH PCA." Journal of Critical Reviews 7.19 (2020): 8089-8095.
10. Kevin Zakka, A Complete Guide to K-Nearest-Neighbors with Applications in Python and R, https://kevinzakka.github.io/2016/07/13/k-nearest-neighbor/, last accessed 2020/06/02.
11. Warning Signs of a Heart Attack | American Heart Association. https://www.heart.org/en/health-topics/heart-attack/warning-signs-of-a-heart-attack
12. Joonghyun, Ryu, et al. "Computation of molecular surface using Euclidean Voronoi Dia-gram." Computer-Aided Design and Applications 2.1-4 (2005): 439-448.
13. Khan SS, Ning H, Wilkins JT, Allen N, Carnethon M, Berry JD, Sweis RN, Lloyd-Jones DM. Association of Body Mass Index with Lifetime Risk of Cardiovascular Disease and Compression of Morbidity. JAMA Cardiol. 2018 Apr 1;3(4):280-287. doi: 10.1001/jamacardio.2018.0022. PMID: 29490333; PMCID: PMC5875319.
14. C. Thirumalai, A. Duba and R. Reddy, "Decision making system using machine learning and Pearson for heart attack," 2017 International conference of Electronics, Communication and Aerospace Technology (ICECA), 2017, pp. 206-210, doi: 10.1109/ICECA.2017.8212797.