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

实验结果



相关推荐
深度学习入门9 分钟前
学习深度学习是否要先学习机器学习?
人工智能·深度学习·神经网络·学习·机器学习·ai·深度学习入门
dog25014 分钟前
BBR 的 buffer 动力学观感
人工智能·算法
python15629 分钟前
使用Langfuse和RAGAS,搭建高可靠RAG应用
人工智能·windows·python
虾球xz1 小时前
游戏引擎学习第281天:在房间之间为摄像机添加动画效果
c++·人工智能·学习·游戏引擎
冷yan~1 小时前
GitHub文档加载器设计与实现
java·人工智能·spring·ai·github·ai编程
AI大模型系统化学习2 小时前
Excel MCP: 自动读取、提炼、分析Excel数据并生成可视化图表和分析报告
人工智能·ai·大模型·ai大模型·大模型学习·大模型入门·mcp
lboyj2 小时前
填孔即可靠:猎板PCB如何用树脂塞孔重构高速电路设计规则
人工智能·重构
Blossom.1182 小时前
从虚拟现实到混合现实:沉浸式体验的未来之路
人工智能·目标检测·机器学习·计算机视觉·语音识别·vr·mr
赵青临的辉2 小时前
简单神经网络(ANN)实现:从零开始构建第一个模型
人工智能·深度学习·神经网络
KALC2 小时前
告别“知识孤岛”:RAG赋能网络安全运营
人工智能·网络安全