International Peer-Reviewed JournalOpen AccessISSN 2456-8880
irejournals@gmail.com+91-7433024337

Home / Current Issue / Paper 1719791

1719791 Vol 8 · Issue 4 Download Paper

Feature-Optimised Heart Disease Prediction Using Machine Learning Techniques

Sunita Sharma Tanishka Rahul Aviral Jain

Subject area: Biological & Medical Sciences  ·  Area of research: Machine Learning in Cardiology

DOI: https://doi.org/10.64388/IREV8I4-1719791

Abstract

Heart disease is a major cause of mortality worldwide, making early and accurate prediction essential for effective clinical intervention. This paper presents a feature-optimized heart disease prediction framework using machine learning techniques to improve diagnostic performance. The proposed approach incorporates data preprocessing, feature optimization, and classification to identify the most relevant clinical attributes while reducing redundancy. Several machine learning algorithms, including Logistic Regression, Decision Tree, Random Forest, Support Vector Machine (SVM), K-Nearest Neighbors (KNN), and XGBoost, are evaluated using standard performance metrics such as accuracy, precision, recall, F1-score, and ROC-AUC. Experimental results demonstrate that feature optimization enhances prediction accuracy and model efficiency, with ensemble-based methods achieving superior performance. The proposed framework offers an effective decision-support tool for the early detection of heart disease and has the potential to assist healthcare professionals in improving diagnostic accuracy and patient care.

Keywords

Heart Disease Prediction, Machine Learning, Feature Optimization, Feature Selection, Classification, Random Forest, Support Vector Machine (SVM), Xgboost, Clinical Decision Support.

References

[1] World Health Organization, Cardiovascular Diseases (CVDs). Geneva, Switzerland: WHO, 2023.

[2] D. Dua and C. Graff, "UCI Machine Learning Repository," University of California, Irvine, School of Information and Computer Sciences, 2019.

[3] R. Bharti, A. Khamparia, M. Shabaz, G. Dhiman, S. Pande, and P. Singh, "Prediction of Heart Disease Using a Combination of Machine Learning and Deep Learning," Computational Intelligence and Neuroscience, vol. 2021, Art. no. 8387680, 2021.

[4] S. Srinivasan, S. Gunasekaran, S. K. Mathivanan, B. A. M. Benjula, P. Jayagopal, and G. T. Dalu, "An Active Learning Machine Technique Based Prediction of Cardiovascular Heart Disease from UCI-Repository Database," Scientific Reports, vol. 13, no. 1, Art. no. 13588, 2023.

[5] C. Zhou, P. Dai, A. Hou, Z. Zhang, L. Liu, and F. Wang, "A Comprehensive Review of Deep Learning-Based Models for Heart Disease Prediction," Artificial Intelligence Review, vol. 57, Art. no. 263, 2024.

[6] P. Shinde, M. Sanghavi, and T. A. Tran, "A Survey on Machine Learning Techniques for Heart Disease Prediction," SN Computer Science, vol. 6, Art. no. 334, 2025.

[7] M. Padilla Rodriguez and M. Nafea, "Centralized and Federated Heart Disease Classification Models Using UCI Dataset and Their Shapley-Value Based Interpretability," arXiv preprint arXiv:2408.06183, 2024.

[8] A. Sekyiwaa Osei-Nkwantabisa and R. Ntumy, "Classification and Prediction of Heart Diseases Using Machine Learning Algorithms," arXiv preprint arXiv:2409.03697, 2024.

[9] M. M. Ahsan and Z. Siddique, "Machine Learning-Based Heart Disease Diagnosis: A Systematic Literature Review," arXiv preprint arXiv:2112.06459, 2021.

[10] U. Priyadharshini and R. V. Vijayan, "Structured-to-Text ClinicalBERT Embeddings with Random Forest for Heart Disease Prediction: A Proof-of-Concept Study on the UCI Statlog Dataset," Frontiers in Artificial Intelligence, vol. 9, 2026.

[11] H. A. Al-Alshaikh, P. Prabu, R. C. Poonia, A. K. J. Saudagar, M. Yadav, H. S. AlSagri, and A. A. AlSanad, "Comprehensive evaluation and performance analysis of machine learning in heart disease prediction," Scientific Reports, vol. 14, Art. no. 7819, Apr. 2024.

[12] A. A. Stonier, R. K. Gorantla, and K. Manoj, "Cardiac disease risk prediction using machine learning algorithms," Healthcare Technology Letters, vol. 11, no. 4, pp. 213–217, Aug. 2024.

[13] C. S. Chaithra, S. Siddesha, V. N. M. Aradhya, and S. K. Niranjan, "A review of machine learning techniques used in the prediction of heart disease," Revue d'Intelligence Artificielle, vol. 38, no. 1, pp. 201–212, 2024.

[14] C. Zhou, P. Dai, A. Hou, Z. Zhang, L. Liu, and F. Wang, "A comprehensive review of deep learning-based models for heart disease prediction," Artificial Intelligence Review, vol. 57, Art. no. 263, 2024.

[15] S. Kumar, K. K. Gola, N. Jee, and B. M. Singh, "Optimized feature fusion-based modified cascaded kernel extreme learning machine for heart disease prediction in e-healthcare," Computer Methods in Biomechanics and Biomedical Engineering, vol. 27, no. 8, pp. 980–993, 2024.

[16] L. Li, X. Chen, and S. Hu, "Application of an end-to-end model with self-attention mechanism in cardiac disease prediction," Frontiers in Physiology, vol. 14, Art. no. 1308774, 2024.

[17] D. Dua and C. Graff, "UCI Machine Learning Repository," School of Information and Computer Sciences, University of California, Irvine, 2019.

[18] A. Géron, Hands-On Machine Learning with Scikit-Learn, Keras, and TensorFlow: Concepts, Tools, and Techniques to Build Intelligent Systems, 3rd ed. Sebastopol, CA, USA: O'Reilly Media, 2022.

[19] I. Guyon and A. Elisseeff, "An Introduction to Variable and Feature Selection," Journal of Machine Learning Research, vol. 3, pp. 1157–1182, Mar. 2003.

[20] T. Chen and C. Guestrin, "XGBoost: A Scalable Tree Boosting System," in Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD), San Francisco, CA, USA, 2016, pp. 785–794.

[21] M. Sokolova and G. Lapalme, "A Systematic Analysis of Performance Measures for Classification Tasks," Information Processing & Management, vol. 45, no. 4, pp. 427–437, 2009.

[22] S. Raschka and V. Mirjalili, Python Machine Learning: Machine Learning and Deep Learning with Python, scikit-learn, and TensorFlow, 4th ed. Birmingham, U.K.: Packt Publishing, 2025.

[23] D. Chicco and G. Jurman, "The advantages of the Matthews correlation coefficient over F1 score and accuracy in binary classification evaluation," BMC Genomics, vol. 21, no. 1, 2020.

[24] C. Zhou, P. Dai, A. Hou, Z. Zhang, L. Liu, and F. Wang, "A Comprehensive Review of Deep Learning-Based Models for Heart Disease Prediction," Artificial Intelligence Review, vol. 57, 2024.

[25] S. Kumar, K. K. Gola, N. Jee, and B. M. Singh, "Optimized Feature Fusion-Based Modified Cascaded Kernel Extreme Learning Machine for Heart Disease Prediction in e-Healthcare," Computer Methods in Biomechanics and Biomedical Engineering, vol. 27, no. 8, 2024.

[26] M. Padilla Rodriguez and M. Nafea, "Centralized and Federated Heart Disease Classification Models Using the UCI Dataset and Shapley-Value-Based Interpretability," arXiv, 2024.

[27] H. A. Al-Alshaikh, P. Prabu, R. C. Poonia, A. K. J. Saudagar, M. Yadav, H. S. AlSagri, and A. A. AlSanad, "Comprehensive evaluation and performance analysis of machine learning in heart disease prediction," Scientific Reports, vol. 14, Art. no. 7819, 2024.

[28] C. Zhou, P. Dai, A. Hou, Z. Zhang, L. Liu, and F. Wang, "A comprehensive review of deep learning-based models for heart disease prediction," Artificial Intelligence Review, vol. 57, Art. no. 263, 2024.

How to cite this paper

Sunita Sharma, Tanishka, Rahul, Aviral Jain "Feature-Optimised Heart Disease Prediction Using Machine Learning Techniques" Iconic Research And Engineering Journals Volume 8 Issue 4 2024 Page 975-981 https://doi.org/10.64388/IREV8I4-1719791
Sunita Sharma, Tanishka, Rahul, Aviral Jain "Feature-Optimised Heart Disease Prediction Using Machine Learning Techniques" Iconic Research And Engineering Journals, vol. 8, no. 4, Oct. 2024, doi: https://doi.org/10.64388/IREV8I4-1719791
Sunita Sharma, Tanishka, Rahul, Aviral Jain (2024). Feature-Optimised Heart Disease Prediction Using Machine Learning Techniques. Iconic Research And Engineering Journals, 8(4). doi: https://doi.org/10.64388/IREV8I4-1719791
Sunita Sharma, Tanishka, Rahul, Aviral Jain "Feature-Optimised Heart Disease Prediction Using Machine Learning Techniques" Iconic Research And Engineering Journals, vol. 8, no. 4, Oct. 2024. Crossref, https://doi.org/10.64388/IREV8I4-1719791
@article{1719791,
      author = {Sunita Sharma, Tanishka, Rahul, Aviral Jain},
      title = {Feature-Optimised Heart Disease Prediction Using Machine Learning Techniques},
      journal = {Iconic Research And Engineering Journals},
      year = {2024},
      volume = {8},
      number = {4},
      pages = {975-981},
      issn = {2456-8880},
      url = {https://www.irejournals.com/formatedpaper/1719791.pdf},
      abstract = {Heart disease is a major cause of mortality worldwide, making early and accurate prediction essential for effective clinical intervention. This paper presents a feature-optimized heart disease prediction framework using machine learning techniques to improve diagnostic performance. The proposed approach incorporates data preprocessing, feature optimization, and classification to identify the most relevant clinical attributes while reducing redundancy. Several machine learning algorithms, including Logistic Regression, Decision Tree, Random Forest, Support Vector Machine (SVM), K-Nearest Neighbors (KNN), and XGBoost, are evaluated using standard performance metrics such as accuracy, precision, recall, F1-score, and ROC-AUC. Experimental results demonstrate that feature optimization enhances prediction accuracy and model efficiency, with ensemble-based methods achieving superior performance. The proposed framework offers an effective decision-support tool for the early detection of heart disease and has the potential to assist healthcare professionals in improving diagnostic accuracy and patient care.},
      keywords = {Heart Disease Prediction, Machine Learning, Feature Optimization, Feature Selection, Classification, Random Forest, Support Vector Machine (SVM), Xgboost, Clinical Decision Support.},
      month = {October},
      doi = {https://doi.org/10.64388/IREV8I4-1719791}
  }