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

实验结果



相关推荐
lili-felicity9 小时前
CANN异步推理实战:从Stream管理到流水线优化
大数据·人工智能
做人不要太理性9 小时前
CANN Runtime 运行时组件深度解析:任务下沉执行、异构内存规划与全栈维测诊断机制
人工智能·神经网络·魔珐星云
不爱学英文的码字机器9 小时前
破壁者:CANN ops-nn 仓库与昇腾 AI 算子优化的工程哲学
人工智能
晚霞的不甘9 小时前
CANN 编译器深度解析:TBE 自定义算子开发实战
人工智能·架构·开源·音视频
愚公搬代码9 小时前
【愚公系列】《AI短视频创作一本通》016-AI短视频的生成(AI短视频运镜方法)
人工智能·音视频
哈__9 小时前
CANN内存管理与资源优化
人工智能·pytorch
极新9 小时前
智启新篇,智创未来,“2026智造新IP:AI驱动品牌增长新周期”峰会暨北京电子商务协会第五届第三次会员代表大会成功举办
人工智能·网络协议·tcp/ip
island13149 小时前
CANN GE(图引擎)深度解析:计算图优化管线、内存静态规划与异构任务的 Stream 调度机制
开发语言·人工智能·深度学习·神经网络
艾莉丝努力练剑9 小时前
深度学习视觉任务:如何基于ops-cv定制图像预处理流程
人工智能·深度学习
禁默9 小时前
大模型推理的“氮气加速系统”:全景解读 Ascend Transformer Boost (ATB)
人工智能·深度学习·transformer·cann