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

实验结果



相关推荐
SHIPKING39330 分钟前
【机器学习&深度学习】制作数据集
人工智能
Ronin-Lotus6 小时前
深度学习篇---剪裁&缩放
图像处理·人工智能·缩放·剪裁
cpsvps6 小时前
3D芯片香港集成:技术突破与产业机遇全景分析
人工智能·3d
国科安芯7 小时前
抗辐照芯片在低轨卫星星座CAN总线通讯及供电系统的应用探讨
运维·网络·人工智能·单片机·自动化
AKAMAI7 小时前
利用DataStream和TrafficPeak实现大数据可观察性
人工智能·云原生·云计算
微光-沫年7 小时前
150-SWT-MCNN-BiGRU-Attention分类预测模型等!
机器学习·matlab·分类
Ai墨芯1117 小时前
深度学习水论文:特征提取
人工智能·深度学习
无名工程师7 小时前
神经网络知识讨论
人工智能·神经网络
nbsaas-boot7 小时前
AI时代,我们更需要自己的开发方式与平台
人工智能
SHIPKING3937 小时前
【机器学习&深度学习】LLamaFactory微调效果与vllm部署效果不一致如何解决
人工智能·深度学习·机器学习