A deep multi-stream model for robust prediction of left ventricular ejection fraction in 2D echocardiography
Artikel i vetenskaplig tidskrift, 2024
We 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 R2= 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 .
Författare
Jennifer Alvén
Göteborgs universitet
Chalmers, Elektroteknik, Signalbehandling och medicinsk teknik
Eva Hagberg
Sahlgrenska universitetssjukhuset
Göteborgs universitet
David Hagerman Olzon
Chalmers, Elektroteknik, Signalbehandling och medicinsk teknik
Sahlgrenska universitetssjukhuset
Göteborgs universitet
Richard Petersen
Sahlgrenska universitetssjukhuset
Göteborgs universitet
Ola Hjelmgren
Sahlgrenska universitetssjukhuset
Göteborgs universitet
Scientific Reports
2045-2322 (ISSN) 20452322 (eISSN)
Vol. 14 1 2104Ämneskategorier
Kardiologi
DOI
10.1038/s41598-024-52480-y