English

Graph Neural Networks for Topological Feature Extraction in ECG Classification

Signal Processing 2023-11-09 v1 Computation and Language Machine Learning

Abstract

The electrocardiogram (ECG) is a dependable instrument for assessing the function of the cardiovascular system. There has recently been much emphasis on precisely classifying ECGs. While ECG situations have numerous similarities, little attention has been paid to categorizing ECGs using graph neural networks. In this study, we offer three distinct techniques for classifying heartbeats using deep graph neural networks to classify the ECG signals accurately. We suggest using different methods to extract topological features from the ECG signal and then using a branch of the graph neural network named graph isomorphism network for classifying the ECGs. On the PTB Diagnostics data set, we tested the three proposed techniques. According to the findings, the three proposed techniques are capable of making arrhythmia classification predictions with the accuracy of 99.38, 98.76, and 91.93 percent, respectively.

Keywords

Cite

@article{arxiv.2311.04228,
  title  = {Graph Neural Networks for Topological Feature Extraction in ECG Classification},
  author = {Kamyar Zeinalipour and Marco Gori},
  journal= {arXiv preprint arXiv:2311.04228},
  year   = {2023}
}

Comments

Accepted for WIRN 2022

R2 v1 2026-06-28T13:14:24.906Z