arXiv:2601.18830cs.LGeess.SP2026-01被引 1

简单混合模型比深层循环结构更适合多标签心电图分类。

How Much Temporal Modeling is Enough? A Systematic Study of Hybrid CNN-RNN Architectures for Multi-Label ECG Classification

  • 用卷积网络提取波形特征,搭配单层双向LSTM建模时间依赖。
  • 单层BiLSTM在各项指标上优于深层循环结构,尤其在准确率和泛化性上。
  • 研究揭示复杂模型未必更好,适合临床部署的模型应匹配信号本质结构。

多标签心电图(ECG)分类因多种心脏疾病共存、类别严重不平衡及多导联信号中长程时间依赖而极具挑战。尽管近期研究多采用深度堆叠循环网络,但其架构复杂性是否必要尚未得到严格验证。本文系统评估了卷积神经网络(CNN)与多种循环结构(LSTM、GRU、双向LSTM及堆叠变体)组合在PTB-XL数据集(23个诊断类别)上的表现。其中CNN作为形态驱动基线,循环层逐步引入以评估其对时间建模与泛化能力的贡献。结果表明,仅使用单层双向LSTM的组合在性能与复杂度间达到最优平衡:相较更深循环结构,其在哈明损失(0.0338)、宏平均AUPRC(0.4715)、微平均F1(0.6979)和子集准确率(0.5723)上均更优。虽然堆叠结构对某些罕见类别的召回略有提升,但整体精度下降且易过拟合,显示增加循环深度带来的收益递减。结论表明,模型架构应与ECG信号内在时间结构相匹配,而非盲目堆叠复杂循环层,这对实现稳健且具临床意义的部署至关重要。

原文摘要 · Abstract (English)

Accurate multi-label classification of electrocardiogram (ECG) signals remains challenging due to the coexistence of multiple cardiac conditions, pronounced class imbalance, and long-range temporal dependencies in multi-lead recordings. Although recent studies increasingly rely on deep and stacked recurrent architectures, the necessity and clinical justification of such architectural complexity have not been rigorously examined. In this work, we perform a systematic comparative evaluation of convolutional neural networks (CNNs) combined with multiple recurrent configurations, including LSTM, GRU, Bidirectional LSTM (BiLSTM), and their stacked variants, for multi-label ECG classification on the PTB-XL dataset comprising 23 diagnostic categories. The CNN component serves as a morphology-driven baseline, while recurrent layers are progressively integrated to assess their contribution to temporal modeling and generalization performance. Experimental results indicate that a CNN integrated with a single BiLSTM layer achieves the most favorable trade-off between predictive performance and model complexity. This configuration attains superior Hamming loss (0.0338), macro-AUPRC (0.4715), micro-F1 score (0.6979), and subset accuracy (0.5723) compared with deeper recurrent combinations. Although stacked recurrent models occasionally improve recall for specific rare classes, our results provide empirical evidence that increasing recurrent depth yields diminishing returns and may degrade generalization due to reduced precision and overfitting. These findings suggest that architectural alignment with the intrinsic temporal structure of ECG signals, rather than increased recurrent depth, is a key determinant of robust performance and clinically relevant deployment.

心电图多标签分类循环网络模型简化

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