Towards Robust Cardiac Segmentation using Graph Convolutional Networks

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Van De Vyver, Gilles, Thomas, Sarina, Ben-Yosef, Guy, Olaisen, Sindre Hellum, Dalen, Håvard, Løvstakken, Lasse, Smistad, Erik
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929560025563136
author Van De Vyver, Gilles
Thomas, Sarina
Ben-Yosef, Guy
Olaisen, Sindre Hellum
Dalen, Håvard
Løvstakken, Lasse
Smistad, Erik
author_facet Van De Vyver, Gilles
Thomas, Sarina
Ben-Yosef, Guy
Olaisen, Sindre Hellum
Dalen, Håvard
Løvstakken, Lasse
Smistad, Erik
contents 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 and show that this eliminates anatomical incorrect multi-structure segmentations on the publicly available CAMUS dataset. Additionally, this work contributes with an ablation study on the graph convolutional architecture and an evaluation of clinical measurements on the clinical HUNT4 dataset. Finally, 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. We show this predictor can detect out-of-distribution and unsuitable input images in real-time. Source code is available online: https://github.com/gillesvntnu/GCN_multistructure
format Preprint
id arxiv_https___arxiv_org_abs_2310_01210
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Towards Robust Cardiac Segmentation using Graph Convolutional Networks
Van De Vyver, Gilles
Thomas, Sarina
Ben-Yosef, Guy
Olaisen, Sindre Hellum
Dalen, Håvard
Løvstakken, Lasse
Smistad, Erik
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
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 and show that this eliminates anatomical incorrect multi-structure segmentations on the publicly available CAMUS dataset. Additionally, this work contributes with an ablation study on the graph convolutional architecture and an evaluation of clinical measurements on the clinical HUNT4 dataset. Finally, 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. We show this predictor can detect out-of-distribution and unsuitable input images in real-time. Source code is available online: https://github.com/gillesvntnu/GCN_multistructure
title Towards Robust Cardiac Segmentation using Graph Convolutional Networks
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2310.01210