Prediction of heart disease using graph neural networks

Tong Chen

doi:10.1117/12.2673160

14 April 2023 Prediction of heart disease using graph neural networks

Tong Chen

Proceedings Volume 12612, International Conference on Artificial Intelligence and Industrial Design (AIID 2022); 1261216 (2023) https://doi.org/10.1117/12.2673160
Event: International Conference on Artificial Intelligence and Industrial Design (AIID 2022), 2022, Zhuhai, China

Abstract

Cardiovascular Diseases (CVDs) have become increasingly crucial in recent years and have been regarded as the leading cause of death worldwide. Although it is necessary to detect and treat CVDs in their early stages, only 67% of heart diseases could be predicted by medical professionals. Motivated by recent advances in Graph Neural Networks (GNNs) that have ramified in a variety of industries, in this paper, we utilize three novel GNN models, including Graph Convolutional Network (GCN), GraphSAGE, and Graph Attention Network (GAT), to conduct the heart disease prediction task, comprised of two stages: table-to-graph transformation and Graph Neural Networks prediction. Experimental results show significant improvements with the utilization of GNNs compared with three novel Machine Learning models: Logistic Regression (LR), Naïve Bayes (NB), and Multi-Layer Perceptron (MLP), and GAT performs optimally among the three GNN models. To the best of our knowledge, this is the first work that predicts heart disease using GNNs.

Citation Download Citation

Tong Chen "Prediction of heart disease using graph neural networks", Proc. SPIE 12612, International Conference on Artificial Intelligence and Industrial Design (AIID 2022), 1261216 (14 April 2023); https://doi.org/10.1117/12.2673160

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available