Benchmarking Self-Supervised Models for Cardiac Ultrasound View Classification
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
Abstract
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 |
| 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) |
| 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 |