Fusion of multi-scale feature extraction and adaptive multi-channel graph neural network for 12-lead ECG classification

Abstract

Background and objective:

The 12-lead electrocardiography (ECG) is a widely used diagnostic method in clinical practice for cardiovascular diseases. The potential correlation between interlead signals is an important reference for clinical diagnosis but is often overlooked by most deep learning methods. Although graph neural networks can capture the associations between leads through edge topology, the complex correlations inherent in 12-lead ECG may involve edge topology, node features, or their combination.

Methods:

In this study, we propose a multi-scale adaptive graph fusion network (MSAGFN) model, which fuses multi-scale feature extraction and adaptive multi-channel graph neural network (AMGNN) for 12-lead ECG classification. The proposed MSAGFN model first extracts multi-scale features individually from 12 leads and then utilizes these features as nodes to construct feature graphs and topology graphs. To efficiently capture the most correlated information from the feature graphs and topology graphs, AMGNN iteratively performs a series of graph operations to learn the final graph-level representations for prediction. Moreover, we incorporate consistency and disparity constraints into our model to further refine the learned features.

Results:

Our model was validated on the PTB-XL dataset, achieving an area under the receiver operating characteristic curve score of 0.937, mean accuracy of 0.894, and maximum F1 score of 0.815. These results surpass the corresponding metrics of state-of-the-art methods. Additionally, we conducted ablation studies to further demonstrate the effectiveness of our model.

Conclusions:

Our study demonstrates that, in 12-lead ECG classification, by constructing topology graphs based on physiological relationships and feature graphs based on lead feature relationships, and effectively integrating them, we can fully explore and utilize the complementary characteristics of the two graph structures. By combining these structures, we construct a comprehensive data view, significantly enhancing the feature representation and classification accuracy.