用双曲几何构建可解释的心脏超声左室充盈压分类模型
HypOProto: Hyperbolic Ordinal Prototypes for Left Ventricular Filling Pressure Classification

- 基于双曲空间原型构建有序分类框架,模拟临床诊断递进关系
- 在412例患者数据上达到93.6%准确率,超越现有方法
- 可视化突出关键病灶区域,适合临床医生验证与信任
超声心动图是评估心脏功能的常用影像手段,左室充盈压(LVFP)是心力衰竭等疾病的关键生理指标。传统分类依赖多普勒测得的$E/e'$比值,具有操作者依赖性且在资源有限地区难以获取,促使直接从B模式超声图像推断LVFP的方法发展。现有深度学习方法虽性能优异,但多为黑箱模型,缺乏临床可解释性。本文提出HypOProto,一种基于双曲序原型的可解释框架,使用冻结的可解释基础模型作为主干网络。该方法将原型沿生理$E/e'$尺度排列:边界病例位于双曲面根部,微小角度差异即可区分;正常与升高病例则分布在外部位置,体现诊断确定性增强。这种双曲结构编码了临床有意义的序数关系,提升可解释性。同时引入新颖的双曲原型角分离(HyperPAS)损失,强制类间原型在双曲空间中分离。HypOProto在412例患者数据上达到93.6%准确率,优于当前最优方法,并通过可视化揭示临床相关区域。本工作首次将原型框架应用于超声心动图中的LVFP分类。代码已开源:https://github.com/DeepRCL/HypOProto。
原文摘要 · Abstract (English)
Echocardiography (echo) is a widely used imaging modality for assessing cardiac function, with Left Ventricular Filling Pressure (LVFP) serving as a critical physiological marker for conditions such as heart failure. Standard LVFP classification into normal \emph{vs} elevated categories relies on the Doppler-derived $E/e'$ ratio, which is operator-dependent and often unavailable in resource-limited settings, motivating methods that infer LVFP directly from B-mode echo. Existing deep learning approaches achieve high performance but remain largely black-box, limiting clinical interpretability. We propose HypOProto, a hyperbolic, ordinal prototype-based framework for interpretable LVFP classification using a frozen, explainable foundation model backbone. HypOProto arranges prototypes along the physiological $E/e'$ scale, placing borderline cases near the hyperboloid root where small angular differences separate similar cases, while normal and elevated cases occupy outward positions reflecting increasing diagnostic certainty. This hyperbolic geometry encodes clinically meaningful ordinal relationships and improves interpretability. We also introduce a novel Hyperbolic Prototype Angular Separation (HyperPAS) loss, enforcing inter-class prototype separation in hyperbolic space. HypOProto achieves SOTA performance while maintaining transparency, and highlights clinically relevant regions in visualizations. This work represents the first prototype-based framework for LVFP classification in echo. Our code can be found at https://github.com/DeepRCL/HypOProto.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。