Benchmarking Self-Supervised Models for Cardiac Ultrasound View Classification
Authors: Youssef Megahed, Salma I. Megahed, Robin Ducharme, Inok Lee, Adrian D. C. Chan, Mark C. Walker, Steven Hawken
Organizations: Department of Systems and Computer Engineering, Carleton University, Ottawa, Ontario, Canada · Department of Methodological and Implementation Research, Ottawa Hospital Research Institute, Ottawa, Ontario, Canada · Department of Acute Care Research, Ottawa Hospital Research Institute, Ottawa, Ontario, Canada · Department of Obstetrics and Gynecology, University of Ottawa, Ottawa, Ontario, Canada · School of Epidemiology and Public Health, University of Ottawa, Ottawa, Ontario, Canada · Department of Obstetrics, Gynecology & Newborn Care, The Ottawa Hospital, Ottawa, Ontario, Canada · International and Global Health Office, University of Ottawa, Ottawa, Ontario, Canada
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.
Figures & tables
Category
Number of Images
Total Images
37,736
A4C
7,422
PL
6,102
PSAV
5,832
PSMV
6,014
Random
6,021
Table 1: Dataset distribution of cardiac view classes and data split used in stratified 5-fold cross-validation.
Figure 1: A) Original ultrasound frame with overlay annotations. B) Cropped and cleaned, processed and inpainted image used for analysis.
Component
MoCo v3 [ 12 ]
USF-MAE (ours) [ 2 ]
Backbone architecture
ViT-B/16
ViT-B/16
Pretraining paradigm
Contrastive SSL
Masked autoencoding (MAE)
Pretraining dataset
ImageNet-1K
OpenUS-46
Dataset domain
Natural images
Ultrasound images
Number of images
∼ 1.28M
∼ 370K
Pretraining objective
Instance discrimination via momentum contrast
Patch reconstruction (MSE over masked patches)
Table 2: Comparison of pre-training configurations for MoCo v3 and USF-MAE.
Figure 2: Comparison of MoCo v3 and USF-MAE Pipelines.
Fold
Accuracy (%)
Recall (%)
F1-score (%)
AUC (%)
MoCo v3
USF-MAE
MoCo v3
USF-MAE
MoCo v3
USF-MAE
MoCo v3
USF-MAE
1
99.18
99.35
99.18
99.35
99.18
99.35
99.98
99.99
2
99.01
99.35
99.01
99.35
99.01
99.35
99.97
99.99
3
98.91
99.42
98.91
99.42
98.91
99.42
99.97
99.98
4
98.97
99.22
98.97
99.22
98.97
99.22
99.98
99.99
5
98.90
99.31
98.90
99.31
98.90
99.31
99.98
99.98
Table 3: Per-fold performance across 5-fold cross-validation.
Figure 3: Classification performance for cardiac view classification: A) normalized confusion matrix and B) per-class ROC curves showing near perfect discrimination across all classes.