Toward Robust Cardiac Segmentation Using Graph Convolutional Networks

Abstract

Fully automatic cardiac segmentation can be a fast and reproducible method to extract clinical measurements from an echocardiography examination. The U-Net architecture is the current state-of-the-art deep learning architecture for medical segmentation and can segment cardiac structures in real-time with average errors comparable to inter-observer variability. However, this architecture still generates large outliers that are often anatomically incorrect. This work uses the concept of graph convolutional neural networks that predict the contour points of the structures of interest instead of labeling each pixel. We propose a graph architecture that uses two convolutional rings based on cardiac anatomy. While this architecture does not improve performance on classical measures like Dice score and Hausdorff distance, it does eliminate anatomical incorrect segmentations. Additionally, we propose to use the inter-model agreement of the U-Net and the graph network as a predictor of both the input and segmentation quality in real-time. The results show that from the 100 high agreement samples, 93 were in distribution, while from the 100 low agreement samples, only 7 were in distribution. Finally, this work contributes with an ablation study of the graph convolutional architecture on the publicly available CAMUS dataset and an evaluation of clinical measurements on the clinical HUNT4 dataset. Source code is available online: https://github.com/gillesvntnu/GCN_multistructure

Keywords

• Cardiac segmentation
• graph convectional network
• robust segmentation
• ultrasound