用时间提示对齐提升胎儿先心病超声视频分类准确率与可靠性
TPA: Temporal Prompt Alignment for Fetal Congenital Heart Defect Classification
- 基于图文模型和时序对比学习,捕捉心脏运动特征并对齐文本提示
- 在私有数据集上达85.40%宏平均F1,校准误差降低5.38%
- 适合需要高可靠性与不确定性评估的医学影像智能诊断场景
胎儿先心病(CHD)的超声视频检测受图像噪声和探头位置变化影响。尽管自动化方法可减少操作依赖,现有机器学习方法常忽略时间信息,局限于二分类且缺乏预测校准。本文提出时间提示对齐(TPA)方法,利用基础图文模型与提示感知对比学习,对心脏超声视频进行胎儿CHD分类。TPA通过图像编码器提取视频子片段每帧特征,用可训练时序提取器聚合以捕捉心脏运动,并通过边缘-铰链对比损失将视频表示与类别特定文本提示对齐。为提升临床可靠性,引入条件变分自编码器风格调制(CVAESM)模块,学习潜在风格向量调节嵌入并量化分类不确定性。在私有CHD检测数据集和大型公开数据集EchoNet-Dynamic(用于收缩功能障碍)上评估,TPA在CHD诊断中取得85.40%的最优宏平均F1,预期校准误差降低5.38%,自适应校准误差降低6.8%;在EchoNet-Dynamic三分类任务中,宏平均F1提升4.73%(从53.89%增至58.62%)。
原文摘要 · Abstract (English)
Congenital heart defect (CHD) detection in ultrasound videos is hindered by image noise and probe positioning variability. While automated methods can reduce operator dependence, current machine learning approaches often neglect temporal information, limit themselves to binary classification, and do not account for prediction calibration. We propose Temporal Prompt Alignment (TPA), a method leveraging foundation image-text model and prompt-aware contrastive learning to classify fetal CHD on cardiac ultrasound videos. TPA extracts features from each frame of video subclips using an image encoder, aggregates them with a trainable temporal extractor to capture heart motion, and aligns the video representation with class-specific text prompts via a margin-hinge contrastive loss. To enhance calibration for clinical reliability, we introduce a Conditional Variational Autoencoder Style Modulation (CVAESM) module, which learns a latent style vector to modulate embeddings and quantifies classification uncertainty. Evaluated on a private dataset for CHD detection and on a large public dataset, EchoNet-Dynamic, for systolic dysfunction, TPA achieves state-of-the-art macro F1 scores of 85.40% for CHD diagnosis, while also reducing expected calibration error by 5.38% and adaptive ECE by 6.8%. On EchoNet-Dynamic's three-class task, it boosts macro F1 by 4.73% (from 53.89% to 58.62%). Temporal Prompt Alignment (TPA) is a framework for fetal congenital heart defect (CHD) classification in ultrasound videos that integrates temporal modeling, prompt-aware contrastive learning, and uncertainty quantification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。