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

实验结果



相关推荐
晨航11 分钟前
AI Agent拐点已至,2B+2C星辰大海——行业深度报告
人工智能·ai·aigc
果冻人工智能17 分钟前
只让 AI 写点坏代码,它却学坏了整颗心
人工智能
zy_destiny27 分钟前
【工业场景】用YOLOv12实现饮料类别识别
人工智能·python·深度学习·yolo·机器学习·计算机视觉·目标跟踪
姚瑞南29 分钟前
从模糊感知到量化评估:构建一个Prompt打分工具
人工智能·自然语言处理·chatgpt·prompt·aigc
机器之心34 分钟前
ICLR 2025 Spotlight | 参数高效微调新范式!上海交大联合上海AI Lab推出参数冗余微调算法
人工智能
机器之心44 分钟前
OpenAI的AI复现论文新基准,Claude拿了第一名
人工智能
骑猪兜风2331 小时前
没有人知道“他妈的” 智能体到底是什么
人工智能·openai·ai编程
www_pp_1 小时前
# 实时人脸识别系统:基于 OpenCV 和 Python 的实现
人工智能·python·opencv
果冻人工智能1 小时前
MCP:让 AI 应用更聪明,只需几分钟
人工智能
人工智能培训咨询叶梓1 小时前
LLAMAFACTORY:一键优化大型语言模型微调的利器
人工智能·语言模型·自然语言处理·性能优化·调优·大模型微调·llama factory