Author:
Alvén Jennifer,Hagberg Eva,Hagerman David,Petersen Richard,Hjelmgren Ola
Abstract
AbstractWe propose a deep multi-stream model for left ventricular ejection fraction (LVEF) prediction in 2D echocardiographic (2DE) examinations. We use four standard 2DE views as model input, which are automatically selected from the full 2DE examination. The LVEF prediction model processes eight streams of data (images + optical flow) and consists of convolutional neural networks terminated with transformer layers. The model is made robust to missing, misclassified and duplicate views via pre-training, sampling strategies and parameter sharing. The model is trained and evaluated on an existing clinical dataset (12,648 unique examinations) with varying properties in terms of quality, examining physician, and ultrasound system. We report $$R^2 = 0.84$$
R
2
=
0.84
and mean absolute error = 4.0% points for the test set. When evaluated on two public benchmarks, the model performs on par or better than all previous attempts on fully automatic LVEF prediction. Code and trained models are available on a public project repository.
Funder
Chalmers Tekniska Högskola
Sahlgrenska Universitetssjukhuset
Analytic Imaging Diagnostics Arena
Göteborgs Läkaresällskap
Chalmers University of Technology
Publisher
Springer Science and Business Media LLC