LLM2D
基于案例的可解释深度学习:通过对比学习进行多标签心电图分类
ProtoECGNet: Case-Based Interpretable Deep Learning for Multi-Label ECG Classification with Contrastive Learning
作者: Sahil Sethi, David Chen, Thomas Statchen, Michael C. Burkhart, Nipun Bhandari, Bashar Ramadan, Brett Beaulieu-Jones
发布日期: 4/14/2025
arXiv ID: oai:arXiv.org:2504.08713v1

摘要

arXiv:2504.08713v1 交叉公告类型:交叉 摘要:基于深度学习的心电图(ECG)分类展现了令人印象深刻的表现,但由于缺乏透明和忠实的解释,其临床应用受到了限制。事后方法如显著性图可能无法反映模型的真实决策过程。基于原型的推理提供了一种更透明的替代方案,它将决策与学习的现实ECG片段表示的相似性联系起来,从而实现忠实的、基于案例的解释。我们引入了ProtoECGNet,这是一种用于可解释、多标签ECG分类的基于原型的深度学习模型。ProtoECGNet 使用一个结构化、多分支的体系结构,反映了临床解释工作流程:它结合了一个1D CNN和全局原型进行心律分类,一个2D CNN和时间局部原型进行基于形态推理,以及一个2D CNN和全局原型进行弥散异常分类。每个分支都是在设计用于多标签学习的原型损失下进行训练的,该损失结合了聚类、分离、多样性,并且包含一种新颖的对比损失,该损失鼓励对同一类中的原型进行适当的分离,同时允许对频繁共存的诊断进行聚类。我们对来自PTB-XL数据集的所有71个诊断标签进行了ProtoECGNet的评估,证明其相对于最先进的黑盒模型具有竞争力的表现,同时提供结构化的、基于案例的解释。为了评估原型质量,我们对最终模型的投影原型进行了结构化的临床审查,发现它们被评价为具有代表性且清晰。ProtoECGNet表明,原型学习可以有效地应用于复杂的多标签时间序列分类,为临床决策支持中的透明和可信赖的深度学习模型提供了一条实用的道路。