Conv-RGNN: An efficient Convolutional Residual Graph Neural Network for ECG classification

Abstract

Background and objective:

Electrocardiogram (ECG) analysis is crucial in diagnosing cardiovascular diseases (CVDs). It is important to consider both temporal and spatial features in ECG analysis to improve automated CVDs diagnosis. Significant progress has been made in automated CVDs diagnosis based on ECG with the continuous development of deep learning. Current most researches often treat 12-lead ECG signals as synchronous sequences in Euclidean space, focusing primarily on extracting temporal features while overlooking the spatial relationships among the 12-lead. However, the spatial distribution of 12-lead ECG electrodes can be more naturally represented using non-Euclidean data structures, which makes the relationships among leads more consistent with their intrinsic characteristics.

Methods:

This study proposes an innovative method, Convolutional Residual Graph Neural Network (Conv-RGNN), for ECG classification. The first step is to segment the 12-lead ECG into twelve single-lead ECG, which are then mapped to nodes in a graph that captures the relationships between the different leads through spatial connections, resulting in the 12-lead ECG graph. The graph is then used as input for Conv-RGNN. A convolutional neural network with a position attention mechanism is used to extract temporal sequence information and selectively integrate contextual information to enhance semantic features at different positions. The spatial features of the 12-lead ECG graph are extracted using the residual graph neural network.

Results:

The experimental results indicate that Conv-RGNN is highly competitive in two multi-label datasets and one single-label dataset, demonstrating exceptional parameter efficiency, inference speed, model performance, and robustness.

Conclusion:

The Conv-RGNN proposed in this paper offer a promising and feasible approach for intelligent diagnosis in resource-constrained environments.