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

文章目录

  • [ProtoASNet: Dynamic Prototypes for Inherently Interpretable and Uncertainty-Aware Aortic Stenosis Classification in Echocardiography](#ProtoASNet: Dynamic Prototypes for Inherently Interpretable and Uncertainty-Aware Aortic Stenosis Classification in Echocardiography)

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) 表示不确定性的原型(蓝色)可以捕捉数据分布中固有模糊性的区域(绿色和黄色区域的交集)。在实际应用中,这个区域包含视觉质量较差的视频。

实验结果



相关推荐
UQI-LIUWJ4 分钟前
模型运行速度笔记: s/epoch VS s/iter
人工智能·笔记·机器学习
墨绿色的摆渡人27 分钟前
用 Python 从零开始创建神经网络(三):添加层级(Adding Layers)
人工智能·python·深度学习·神经网络
nuc_baixu29 分钟前
opencv kdtree & pcl kdtree 效率对比
人工智能·opencv·计算机视觉
学步_技术1 小时前
自动驾驶系列—自动驾驶中的短距离感知:超声波雷达的核心技术与场景应用
人工智能·机器学习·自动驾驶·uss
凡人的AI工具箱1 小时前
15分钟学 Go 第 54 天 :项目总结与经验分享
开发语言·人工智能·后端·算法·golang
&永恒的星河&1 小时前
爆火的AI智能助手ChatGPT中自注意力机制的演化与优化
人工智能·chatgpt
WenGyyyL1 小时前
无人机检测车辆——多目标检测
人工智能·目标检测·无人机
goomind2 小时前
YOLOv11实战PCB电路板缺陷识别
人工智能·python·深度学习·yolo·目标检测·计算机视觉·缺陷检测
彭彭不吃虫子2 小时前
有效论证的四个方法:①真值表法 ② 逻辑推演法 ③等值演算法 ④ 主析取范式法
人工智能
新加坡内哥谈技术2 小时前
预测AI如何提升销售绩效管理:五大方式
人工智能