摘要
arXiv:2504.08713v2 通知类型: 交叉替换
摘要:基于深度学习的心电图(ECG)分类已经展现出了令人印象深刻的性能,但由于缺乏透明且忠实的解释,临床应用仍受到限制。事后方法如显著性图可能无法准确反映模型的真实决策过程。基于原型的推理提供了一种更透明的替代方案,通过将决策基于实际心电图段落的相似性,使忠实、案例为基础的解释成为可能。我们引入了ProtoECGNet,这是一种基于原型的深度学习模型,用于可解释的多标签心电图分类。ProtoECGNet 使用一种结构化、多分支架构,反映了临床解释工作流程:它结合了一个一维卷积神经网络(1D CNN)与全局原型进行节律分类,一个二维卷积神经网络(2D CNN)与时间局部原型进行基于形态的推理,以及另一个二维卷积神经网络(2D CNN)与全局原型进行广泛的异常处理。每个分支都使用一种为多标签学习设计的原型损失进行训练,结合了聚类、分离、多样性和一种新的对比损失,该损失鼓励不同类别原型之间的适当分离,同时允许聚类于频繁共存的诊断。我们在 PTB-XL 数据集上对所有 71 个诊断标签进行评估,显示出与最先进的黑盒模型相当的性能,同时提供了结构化的案例为基础的解释。为了评估原型质量,我们对最终模型投影后的原型进行了一种结构化的临床审查,发现这些原型被评定为具有代表性和清晰性。ProtoECGNet 表明原型学习可以有效地扩展到复杂的多标签时间序列分类,为临床决策支持提供了透明且可信任的深度学习模型的实用途径。