自监督模型USF-MAE在心脏超声视图分类上优于MoCo v3。
Benchmarking Self-Supervised Models for Cardiac Ultrasound View Classification
- 用自监督学习从无标签数据中提取特征,提升分类能力
- USF-MAE平均AUC达99.99%,准确率99.33%,显著更高
- 适合医疗影像自动化分析研究者参考
可靠解读心脏超声图像对临床诊断至关重要。自监督学习通过利用大量未标注数据学习有意义的表征,在医学影像领域展现出潜力。本研究在新提出的CACTUS数据集(37,736张图像)上评估并比较了两种自监督学习框架:我们团队开发的USF-MAE与MoCo v3,用于自动模拟心脏视图(A4C、PL、PSAV、PSMV、Random、SC)分类。两模型均采用5折交叉验证,确保泛化性能的稳健评估。数据集包含专家标注的多样化心脏超声图像。采用相同的训练协议进行公平对比,学习率设为0.0001,权重衰减为0.01。每折记录ROC-AUC、准确率、F1分数和召回率。结果表明,USF-MAE在各项指标上均持续优于MoCo v3。USF-MAE平均测试AUC为99.99%(±0.01%,95%置信区间),高于MoCo v3的99.97%(±0.01%)。其平均测试准确率为99.33%(±0.18%),高于MoCo v3的98.99%(±0.28%)。F1分数与召回率趋势一致,差异在各折间统计显著(配对t检验,p=0.0048<0.01)。该概念验证分析表明,相较于MoCo v3,USF-MAE在该数据集上能学习更具判别性的特征,具备提升心脏超声自动分类的潜力。
原文摘要 · Abstract (English)
Reliable interpretation of cardiac ultrasound images is essential for accurate clinical diagnosis and assessment. Self-supervised learning has shown promise in medical imaging by leveraging large unlabelled datasets to learn meaningful representations. In this study, we evaluate and compare two self-supervised learning frameworks, USF-MAE, developed by our team, and MoCo v3, on the recently introduced CACTUS dataset (37,736 images) for automated simulated cardiac view (A4C, PL, PSAV, PSMV, Random, and SC) classification. Both models used 5-fold cross-validation, enabling robust assessment of generalization performance across multiple random splits. The CACTUS dataset provides expert-annotated cardiac ultrasound images with diverse views. We adopt an identical training protocol for both models to ensure a fair comparison. Both models are configured with a learning rate of 0.0001 and a weight decay of 0.01. For each fold, we record performance metrics including ROC-AUC, accuracy, F1-score, and recall. Our results indicate that USF-MAE consistently outperforms MoCo v3 across metrics. The average testing AUC for USF-MAE is 99.99% (+/-0.01% 95% CI), compared to 99.97% (+/-0.01%) for MoCo v3. USF-MAE achieves a mean testing accuracy of 99.33% (+/-0.18%), higher than the 98.99% (+/-0.28%) reported for MoCo v3. Similar trends are observed for the F1-score and recall, with improvements statistically significant across folds (paired t-test, p=0.0048 < 0.01). This proof-of-concept analysis suggests that USF-MAE learns more discriminative features for cardiac view classification than MoCo v3 when applied to this dataset. The enhanced performance across multiple metrics highlights the potential of USF-MAE for improving automated cardiac ultrasound classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。