arXiv:2502.16255eess.SPcs.AI2025-02被引 3

用自注意力融合心电图与患者数据,提升心律失常诊断准确率。

rECGnition_v2.0: Self-Attentive Canonical Fusion of ECG and Patient Data using deep learning for effective Cardiac Diagnostics

  • 通过自注意力机制融合心电图与患者特征,提升模型表现。
  • 在MIT-BIH数据集上达到98.07%准确率,仅需82.7M FLOPs/样本。
  • 模型轻量高效,适合临床部署且具备可解释性。

个体患者特征导致心电图读数差异,制约自动化分析在临床中的应用。本文提出SACC(自注意力典型相关)特征融合方法,结合双路径网络与深度可分离卷积,构建名为rECGnition_v2.0的端到端心律失常分类模型。在MIT-BIH、INCARTDB和EDB数据集上评估其性能。通过消融实验对比简单拼接、CCA与SACC,验证全局与局部心电特征的重要性。模型在参数量、计算量、内存占用和推理时间方面优于当前主流CNN模型。在MIT-BIH数据集上,对10类心律失常分类实现98.07%准确率与98.05% F1-score,FLOPs仅为82.7M/样本;在INCARTDB和EDB数据集上,分别取得98.01%与96.21%的AAMI分类F1-score。该模型结构紧凑,训练参数少,计算开销低,兼具可解释性与可扩展性。

原文摘要 · Abstract (English)

The variability in ECG readings influenced by individual patient characteristics has posed a considerable challenge to adopting automated ECG analysis in clinical settings. A novel feature fusion technique termed SACC (Self Attentive Canonical Correlation) was proposed to address this. This technique is combined with DPN (Dual Pathway Network) and depth-wise separable convolution to create a robust, interpretable, and fast end-to-end arrhythmia classification model named rECGnition_v2.0 (robust ECG abnormality detection). This study uses MIT-BIH, INCARTDB and EDB dataset to evaluate the efficiency of rECGnition_v2.0 for various classes of arrhythmias. To investigate the influence of constituting model components, various ablation studies were performed, i.e. simple concatenation, CCA and proposed SACC were compared, while the importance of global and local ECG features were tested using DPN rECGnition_v2.0 model and vice versa. It was also benchmarked with state-of-the-art CNN models for overall accuracy vs model parameters, FLOPs, memory requirements, and prediction time. Furthermore, the inner working of the model was interpreted by comparing the activation locations in ECG before and after the SACC layer. rECGnition_v2.0 showed a remarkable accuracy of 98.07% and an F1-score of 98.05% for classifying ten distinct classes of arrhythmia with just 82.7M FLOPs per sample, thereby going beyond the performance metrics of current state-of-the-art (SOTA) models by utilizing MIT-BIH Arrhythmia dataset. Similarly, on INCARTDB and EDB datasets, excellent F1-scores of 98.01% and 96.21% respectively was achieved for AAMI classification. The compact architectural footprint of the rECGnition_v2.0, characterized by its lesser trainable parameters and diminished computational demands, unfurled several advantages including interpretability and scalability.

心电图深度学习心律失常模型轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。