ProtoASNet：用于超声心动图中主动脉瓣狭窄分类的动态原型网络，具备内在可解释性和不确定性意识

ProtoASNet: Dynamic Prototypes for Inherently Interpretable and Uncertainty-Aware Aortic Stenosis Classification in Echocardiography

摘要

主动脉瓣狭窄（Aortic Stenosis, AS）是一种常见的心脏瓣膜疾病，准确及时的诊断对于适当的治疗至关重要。目前大多数自动检测AS严重程度的方法依赖于黑箱模型，这些模型的信任度较低，阻碍了其在临床中的应用。为了解决这一问题，本文提出了一种名为ProtoASNet的原型网络，该网络直接从B模式超声心动图视频中检测AS，同时基于输入和学习到的时空原型之间的相似性进行可解释的预测。

核心贡献
可解释的预测：
ProtoASNet通过学习到的时空原型进行预测，这些原型通常突出如钙化和主动脉瓣叶片活动受限等标志性特征，从而提供临床相关的支持证据。
不确定性估计：
该网络利用弃权损失（abstention loss）来估计不确定性，通过定义一组捕捉观测数据中的歧义和信息不足的原型，提供了一个可以检测并解释其可能失败的可靠系统。
方法
原型网络架构：
ProtoASNet通过卷积神经网络提取输入视频的时空特征，然后将这些特征与学习到的原型进行比较，基于相似性进行分类。每个原型代表一种特定的特征模式，模型通过这些原型来解释和分类输入数据。
弃权损失：
在预测过程中，模型通过定义一组原型来捕捉数据中的不确定性区域。如果输入数据与这些原型的相似性较高，则标识为高不确定性区域，提示需要进一步的人工审核或更多数据支持。
代码地址

方法

图 1. ProtoASNet架构概览

(A) ProtoASNet架构概览。ProtoASNet从视频中提取时空特征向量，并将这些特征与学习到的原型进行比较。特征与原型之间的相似值被聚合以生成类别成员资格评分和不确定性评分。

(B) 表示不确定性的原型（蓝色）可以捕捉数据分布中固有模糊性的区域（绿色和黄色区域的交集）。在实际应用中，这个区域包含视觉质量较差的视频。

实验结果