Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

CoReEcho: Continuous Representation Learning for 2D+time Echocardiography Analysis

About

Deep learning (DL) models have been advancing automatic medical image analysis on various modalities, including echocardiography, by offering a comprehensive end-to-end training pipeline. This approach enables DL models to regress ejection fraction (EF) directly from 2D+time echocardiograms, resulting in superior performance. However, the end-to-end training pipeline makes the learned representations less explainable. The representations may also fail to capture the continuous relation among echocardiogram clips, indicating the existence of spurious correlations, which can negatively affect the generalization. To mitigate this issue, we propose CoReEcho, a novel training framework emphasizing continuous representations tailored for direct EF regression. Our extensive experiments demonstrate that CoReEcho: 1) outperforms the current state-of-the-art (SOTA) on the largest echocardiography dataset (EchoNet-Dynamic) with MAE of 3.90 & R2 of 82.44, and 2) provides robust and generalizable features that transfer more effectively in related downstream tasks. The code is publicly available at https://github.com/fadamsyah/CoReEcho.

Fadillah Adamsyah Maani, Numan Saeed, Aleksandr Matsun, Mohammad Yaqub• 2024

Related benchmarks

TaskDatasetResultRank
Ejection Fraction PredictionEchoNet-Dynamic (test)
R20.82
44
LVEF estimationEchoNet-Pediatric
MAE4.7
17
Myocardial Infarction classificationHMC-QU A4C view (stratified 5-fold CV)
Sensitivity93.05
9
LVEF estimationCAMUS (test)
MAE7.35
7
Ejection Fraction regressionCAMUS A4C view (10-fold CV)
G&M Qual Correlation0.807
6
Showing 5 of 5 rows

Other info

Code

Follow for update