Light-weight Spatio-Temporal Graphs for Segmentation and Ejection Fraction Prediction in Cardiac Ultrasound

• Published in: Medical Image Computing and Computer Assisted Intervention – MICCAI 2022 pp. 380 - 390
• Main Authors: Thomas, Sarina, Gilbert, Andrew, Ben-Yosef, Guy
• Format: Book Chapter
• Language: English
• Published: Cham Springer Nature Switzerland
• Series: Lecture Notes in Computer Science
• Subjects: Echocardiography; Ejection fraction; Graph convolutional networks; Segmentation; Ultrasound
• ISBN: 9783031164392; 3031164393
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-031-16440-8_37

Accurate and consistent predictions of echocardiography parameters are important for cardiovascular diagnosis and treatment. In particular, segmentations of the left ventricle can be used to derive ventricular volume, ejection fraction (EF) and other relevant measurements. In this paper we propose a new automated method called EchoGraphs for predicting ejection fraction and segmenting the left ventricle by detecting anatomical keypoints. Models for direct coordinate regression based on Graph Convolutional Networks (GCNs) are used to detect the keypoints. GCNs can learn to represent the cardiac shape based on local appearance of each keypoint, as well as global spatial and temporal structures of all keypoints combined. We evaluate our EchoGraphs model on the EchoNet benchmark dataset. Compared to semantic segmentation, GCNs show accurate segmentation and improvements in robustness and inference run-time. EF is computed simultaneously to segmentations and our method also obtains state-of-the-art ejection fraction estimation. Source code is available online: https://github.com/guybenyosef/EchoGraphs.