Heart Disease Prediction using Graph Neural Network

Authors

  • Rakhi Wajgi Department of Computer Technology, Yeshwantrao Chavan College of Engineering, Nagpur, Maharashtra (India)
  • Tushar Champaneria Department of Computer Engineering, Governemnt Engineering College, Modasa, Gujarat,(India)
  • Dipak Wajgi Department of Computer Engineering, St. VincntPallotti College of Engineering and Technology, Maharashtra (India)
  • Yogesh Suryawanshi Department of Electronics Engineering, YeshwantraoChavan College of Engineering , Nagpur, INDIA
  • Dinesh Bhoyar Department of Electronics and Telecommunication Engineering, Yeshwantrao Chavan College of Engineering, Nagpur, INDIA
  • Ajinkya Nilawar Department of Electronics and Communication Engineering, Shri Ramdeobaba College of Engineering and Management, Maharashtra (India)

Keywords:

Cardiovascular disease, Graph Neural Network, Optimizer

Abstract

Heart is an important organ playing vital role in the life of living organisms. Heart and circulatory disease encompasses a range of conditions affecting the heart and blood vessels, including coronary artery disease, arrhythmias, and heart failure mechanism. Early detection of malfunctioning before failure of heart is necessary. This paper deals with the model built using Graph Neural Network (GNN) to predict heart disease so that mortality rate caused due to sudden heart failure can be reduced. In order to improve the accuracy of GNN-based model, different optimizers are used. They are help to optimize or improve the model's performance by iteratively updating its parameters to reach the optimal values that minimize the difference between predicted and actual outputs. The proposed model is applied on a real dataset from kaggle containing 14 features. Out of all optimizers, RMSprop outperforms otherswith accuracy of 91% and MSE of 48%.

Downloads

Download data is not yet available.

References

Bei W. , Xiaoqing L. , Jingwei Q. , Haowen S., Zehua P., Zhi T., GNDD: A Graph Neural Network-Based Method for Drug-Disease Association Prediction, IEEE International conference on Bioinformatics and Biomedicine (BIBM) (2019)

Tzu-An S., Samadrita C., Fan Y. Heidi J., Georges F., Quanzheng L., Keith J., Joyita D. Graph Convolutional neural network for Alzheimer’s disease classification. IEEE 16th international symposium on biomedical imaging (ISBI 2019)

Sicen L., Tao L., Haoyang D., Buzhou T., Xiaolong W., Qingcai C., Jun Y., Yi Z., A hybrid method of recurrent neural network and graph neural network for next-period prescription prediction, International journal of machine learning and cybernetics (2020).

Amol K. Xue B., Luyang L., Bryan P., Matt B., Martin B., Shawn O., Examining COVID-19 Forecasting using spatio-temporal Graph Neural Network, arXiv:2007.03113v1 (2020)

Tao L., Weihua P., Qingcai C., Xioaolong W., Buzhou T., KeoG: a knowledge-aware edge-oriented graph neural network for documnet-level relation extraction, IEEE international conference on Bioinformatics and biomedicine (2020)

Haifeng L., Hongfei L., Chen S., Liang Y., Yuan L., Bo X., Zhihao Y., Jian W., Yuanyuan S., Drug Repositioning for SARS-CoV-2 Based on Graph neural network, IEEE International conference on Bioinformatics and Biomedicine (2020).

Waigi, R.; Choudhary, S.; Fulzele, P.; Mishra, G. Predicting the risk of heart disease using advanced machine learning approach. Eur. J. Mol. Clin. Med. 2020, 7, 1638–1645.

Breiman, L. Random forests. Mach. Learn. 2001, 45, 5–32.

Chen, T.; Guestrin, C. XGBoost: A Scalable Tree Boosting System. In Proceedings of the KDD ’16: 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, San Francisco, CA, USA, 13–17 August 2016; Association for Computing Machinery: New York, NY, USA, 2016; pp. 785–794.

Gietzelt, M.; Wolf, K.-H.; Marschollek, M.; Haux, R. Performance comparison of accelerometer calibration algorithms based on 3D-ellipsoid fitting methods. Comput. Methods Programs Biomed. 2013, 111, 62–71.

K, V.; Singaraju, J. Decision Support System for Congenital Heart Disease Diagnosis based on Signs and Symptoms using Neural Networks. Int. J. Comput. Appl. 2011, 19, 6–12.

Nagavelli U, Samanta D, Chakraborty P. Machine Learning Technology-Based Heart Disease Detection Models. J Healthcare Eng. 2022 Feb 27;2022:7351061. doi: 10.1155/2022/7351061. PMID: 35265303; PMCID: PMC8898839.

Sogancioglu E., Murphy K., Calli E., Scholten E. T., Schalekamp S., Van Ginneken B. Cardiomegaly detection on chest radiographs: segmentation versus classification. IEEE Access . 2020;8 doi: 10.1109/access.2020.2995567.94631

Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C., & Philip, S. Y. (2020). A comprehensive survey on graph neural networks. IEEE transactions on neural networks and learning systems, 32(1), 4-24.

https://blogs.nvidia.com/blog/what-are-graph-neural-networks/ accessed on 10th Dec. 2023

Kriege, N. M., Johansson, F. D., & Morris, C. (2020). A survey on graph kernels. Applied Network Science, 5(1), 1-42.

Downloads

Published

12.01.2024

How to Cite

Wajgi, R. ., Champaneria, T. ., Wajgi, D. ., Suryawanshi, Y. ., Bhoyar, D. ., & Nilawar, A. . (2024). Heart Disease Prediction using Graph Neural Network. International Journal of Intelligent Systems and Applications in Engineering, 12(12s), 280–287. Retrieved from https://ijisae.org/index.php/IJISAE/article/view/4514

Issue

Vol. 12 No. 12s (2024)

Section

Research Article

License

Creative Commons License

This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.

All papers should be submitted electronically. All submitted manuscripts must be original work that is not under submission at another journal or under consideration for publication in another form, such as a monograph or chapter of a book. Authors of submitted papers are obligated not to submit their paper for publication elsewhere until an editorial decision is rendered on their submission. Further, authors of accepted papers are prohibited from publishing the results in other publications that appear before the paper is published in the Journal unless they receive approval for doing so from the Editor-In-Chief.

IJISAE open access articles are licensed under a Creative Commons Attribution-ShareAlike 4.0 International License. This license lets the audience to give appropriate credit, provide a link to the license, and indicate if changes were made and if they remix, transform, or build upon the material, they must distribute contributions under the same license as the original.