Arrhythmia classification based on multi-input convolutional neural network with attention mechanism

基于多输入卷积神经网络和注意力机制的心律失常分类

阅读:1

Abstract

Arrhythmia is a prevalent cardiac disorder that can lead to severe complications such as stroke and cardiac arrest. While deep learning has advanced automated ECG analysis, challenges remain in accurately classifying arrhythmias due to signal variability, data imbalance, and feature representation limitations. In this work, we propose a novel arrhythmia classification algorithm based on a multi-input convolutional neural network (CNN) enhanced with a Squeeze-and-Excitation (SE) attention mechanism. Distinct from previous methods that rely on single-resolution features or unimodal inputs, our model integrates multi-scale time-frequency representations derived from Short-Time Fourier Transform (STFT) applied to ECG signals segmented into two temporal resolutions. The dual-branch CNN architecture enables complementary feature learning from both short and long segments, while SE blocks enhance inter-channel dependencies to prioritize critical features. The fusion strategy combines feature maps via bicubic interpolation and element-wise summation to maintain spatial integrity. Evaluated on MIT-BIH and SPH arrhythmia databases, the proposed model achieves high accuracy (99.13% and 95.84%, respectively) and Macro-F1 scores (94.46% and 95.91%), outperforming several state-of-the-art approaches. These results highlight the model's potential for robust and interpretable arrhythmia classification in clinical practice.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。