用原型学习实现可解释的心电图多标签分类,让模型决策像医生看病例一样透明。
ProtoECGNet: Case-Based Interpretable Deep Learning for Multi-Label ECG Classification with Contrastive Learning
- 通过多分支结构融合不同特征,用原型匹配真实心电图片段进行推理。
- 在PTB-XL数据集上71个诊断标签中表现媲美黑箱模型,且能生成清晰解释。
- 适合临床场景,医生可评估原型是否代表典型病例,提升信任度。
基于深度学习的心电图(ECG)分类虽表现优异,但临床应用受限于缺乏透明可信的解释。事后解释方法如显著性图可能无法反映模型真实决策过程。原型学习通过将决策建立在真实心电图片段的相似性上,提供更透明的案例式解释。本文提出ProtoECGNet,一种用于可解释多标签心电图分类的原型学习模型。该模型采用多分支结构,对应临床分析流程:使用一维CNN与全局原型进行节律分类,二维CNN与时间定位原型用于形态学分析,二维CNN与全局原型识别弥漫性异常。各分支均采用为多标签设计的原型损失,结合聚类、分离、多样性及一种新型对比损失,促进无关类别原型间的分离,同时允许常共现诊断的原型聚集。在PTB-XL数据集全部71个诊断标签上评估,性能与当前最佳黑箱模型相当,并提供结构化案例式解释。通过结构化医生评审,最终模型的原型被评价为具有代表性且清晰。结果表明,原型学习可有效拓展至复杂多标签时序分类任务,为临床决策支持提供可信赖的透明模型路径。
原文摘要 · Abstract (English)
Deep learning-based electrocardiogram (ECG) classification has shown impressive performance but clinical adoption has been slowed by the lack of transparent and faithful explanations. Post hoc methods such as saliency maps may fail to reflect a model's true decision process. Prototype-based reasoning offers a more transparent alternative by grounding decisions in similarity to learned representations of real ECG segments, enabling faithful, case-based explanations. We introduce ProtoECGNet, a prototype-based deep learning model for interpretable, multi-label ECG classification. ProtoECGNet employs a structured, multi-branch architecture that reflects clinical interpretation workflows: it integrates a 1D CNN with global prototypes for rhythm classification, a 2D CNN with time-localized prototypes for morphology-based reasoning, and a 2D CNN with global prototypes for diffuse abnormalities. Each branch is trained with a prototype loss designed for multi-label learning, combining clustering, separation, diversity, and a novel contrastive loss that encourages appropriate separation between prototypes of unrelated classes while allowing clustering for frequently co-occurring diagnoses. We evaluate ProtoECGNet on all 71 diagnostic labels from the PTB-XL dataset, demonstrating competitive performance relative to state-of-the-art black-box models while providing structured, case-based explanations. To assess prototype quality, we conduct a structured clinician review of the final model's projected prototypes, finding that they are rated as representative and clear. ProtoECGNet shows that prototype learning can be effectively scaled to complex, multi-label time-series classification, offering a practical path toward transparent and trustworthy deep learning models for clinical decision support.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。